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

Add some missing return type annotations in src/IRCServer.Server

Browse source
This commit is contained in:
jesopo 2018-11-21 20:05:21 +00:00
parent bac4ba89b7
commit 9a5ba99840
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/IRCServer.py
View file

@ -162,7 +162,7 @@ class Server(IRCObject.Object):
def set_own_nickname(self, nickname: str): def set_own_nickname(self, nickname: str):
self.nickname = nickname self.nickname = nickname
self.nickname_lower = utils.irc.lower(self.case_mapping, nickname) self.nickname_lower = utils.irc.lower(self.case_mapping, nickname)
def is_own_nickname(self, nickname: str): def is_own_nickname(self, nickname: str) -> bool:
return utils.irc.equals(self.case_mapping, nickname, self.nickname) return utils.irc.equals(self.case_mapping, nickname, self.nickname)
def add_own_mode(self, mode: str, arg: str=None): def add_own_mode(self, mode: str, arg: str=None):
@ -175,9 +175,9 @@ class Server(IRCObject.Object):
else: else:
self.add_own_mode(mode, arg) self.add_own_mode(mode, arg)
def has_user(self, nickname: str): def has_user(self, nickname: str) -> bool:
return utils.irc.lower(self.case_mapping, nickname) in self.users return utils.irc.lower(self.case_mapping, nickname) in self.users
def get_user(self, nickname: str, create: bool=True): def get_user(self, nickname: str, create: bool=True) -> IRCUser.User:
if not self.has_user(nickname) and create: if not self.has_user(nickname) and create:
user_id = self.get_user_id(nickname) user_id = self.get_user_id(nickname)
new_user = IRCUser.User(nickname, user_id, self, self.bot) new_user = IRCUser.User(nickname, user_id, self, self.bot)
@ -186,7 +186,7 @@ class Server(IRCObject.Object):
self.new_users.add(new_user) self.new_users.add(new_user)
return self.users.get(utils.irc.lower(self.case_mapping, nickname), return self.users.get(utils.irc.lower(self.case_mapping, nickname),
None) None)
def get_user_id(self, nickname: str): def get_user_id(self, nickname: str) -> int:
self.bot.database.users.add(self.id, nickname) self.bot.database.users.add(self.id, nickname)
return self.bot.database.users.get_id(self.id, nickname) return self.bot.database.users.get_id(self.id, nickname)
def remove_user(self, user: IRCUser.User): def remove_user(self, user: IRCUser.User):