From 50d1eb6b8c8d4333fc50695521054443bc191cc0 Mon Sep 17 00:00:00 2001 From: jesopo Date: Mon, 7 Oct 2019 12:46:52 +0100 Subject: combine find_setting() and find_setting_prefix() on user/channel/server/bot --- src/IRCChannel.py | 19 +++++++++++-------- 1 file changed, 11 insertions(+), 8 deletions(-) (limited to 'src/IRCChannel.py') diff --git a/src/IRCChannel.py b/src/IRCChannel.py index 2404d7af..0449e135 100644 --- a/src/IRCChannel.py +++ b/src/IRCChannel.py @@ -162,14 +162,17 @@ class Channel(IRCObject.Object): else: return value - def find_settings(self, pattern: str, default: typing.Any=[] - ) -> typing.List[typing.Any]: - return self.bot.database.channel_settings.find(self.id, pattern, - default) - def find_settings_prefix(self, prefix: str, default: typing.Any=[] - ) -> typing.List[typing.Any]: - return self.bot.database.channel_settings.find_prefix(self.id, - prefix, default) + def find_settings(self, pattern: str=None, prefix: str=None, + default: typing.Any=[]) -> typing.List[typing.Any]: + if not pattern == None: + return self.bot.database.channel_settings.find(self.id, pattern, + default) + elif not prefix == None: + return self.bot.database.channel_settings.find_prefix(self.id, + prefix, default) + else: + raise ValueError("Please provide 'pattern' or 'prefix'") + def del_setting(self, setting: str): self.bot.database.channel_settings.delete(self.id, setting) -- cgit v1.3.1-10-gc9f91