Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

combine find_user_settings() and find_user_settings_prefix()

Browse source
This commit is contained in:
jesopo 2019-10-07 12:51:44 +01:00
parent 50d1eb6b8c
commit 257042ac2c
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
src/IRCChannel.py
View file

@ -187,14 +187,17 @@ class Channel(IRCObject.Object):
default: typing.Any=None) -> typing.Any:
return self.bot.database.user_channel_settings.get(user_id,
self.id, setting, default)
def find_user_settings(self, user_id: int, pattern: str,
default: typing.Any=[]) -> typing.List[typing.Any]:
return self.bot.database.user_channel_settings.find(user_id,
self.id, pattern, default)
def find_user_settings_prefix(self, user_id: int, prefix: str,
default: typing.Any=[]) -> typing.List[typing.Any]:
return self.bot.database.user_channel_settings.find_prefix(
user_id, self.id, prefix, default)
def find_user_settings(self, user_id: int, pattern: str=None,
prefix: str=None, default: typing.Any=[]
) -> typing.List[typing.Any]:
if not pattern == None:
return self.bot.database.user_channel_settings.find(user_id,
self.id, pattern, default)
elif not prefix == None:
return self.bot.database.user_channel_settings.find_prefix(user_id,
self.id, prefix, default)
else:
raise ValueError("Please provide 'pattern' or 'prefix'")
def del_user_setting(self, user_id: int, setting: str):
self.bot.database.user_channel_settings.delete(user_id, self.id,
setting)