diff options
Diffstat (limited to 'src/IRCUser.py')
| -rw-r--r-- | src/IRCUser.py | 20 |
1 files changed, 12 insertions, 8 deletions
diff --git a/src/IRCUser.py b/src/IRCUser.py index 8266febd..7ba334c5 100644 --- a/src/IRCUser.py +++ b/src/IRCUser.py @@ -57,14 +57,18 @@ class User(IRCObject.Object): def get_setting(self, setting: str, default: typing.Any=None) -> typing.Any: return self.bot.database.user_settings.get(self.get_id(), setting, default) - def find_settings(self, pattern: str, default: typing.Any=[] - ) -> typing.List[typing.Any]: - return self.bot.database.user_settings.find(self.get_id(), pattern, - default) - def find_settings_prefix(self, prefix: str, default: typing.Any=[] - ) -> typing.List[typing.Any]: - return self.bot.database.user_settings.find_prefix(self.get_id(), - prefix, default) + + def find_setting(self, pattern: str=None, prefix: str=None, + default: typing.Any=[]) -> typing.List[typing.Any]: + if not pattern == None: + return self.bot.database.user_settings.find(self.get_id(), pattern, + default) + elif not prefix == None: + return self.bot.database.user_settings.find_prefix(self.get_id(), + prefix, default) + else: + raise ValueError("Please provide 'pattern' or 'prefix'") + def del_setting(self, setting): self.bot.database.user_settings.delete(self.get_id(), setting) def get_channel_settings_per_setting(self, setting: str, |
