ryg/royalnet
ryg/royalnet
1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
Code Activity

Refactor Telegram database table

Browse source
This commit is contained in:
Steffo 2019-04-11 18:49:16 +02:00
parent c625f3c279
commit e7b82dfa92
1 changed files with 8 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
royalnet/database/tables/telegram.py
View file

@ -13,10 +13,9 @@ class Telegram:
royal_id = Column(Integer, ForeignKey("royals.uid"))
tg_id = Column(BigInteger, primary_key=True)
tg_first_name = Column(String)
tg_last_name = Column(String)
tg_username = Column(String)
tg_avatar = Column(LargeBinary)
first_name = Column(String)
last_name = Column(String)
username = Column(String)
royal = relationship("Royal", backref="telegram")
@ -27,9 +26,9 @@ class Telegram:
return f"[c]telegram:{self.mention()}[/c]"
def mention(self) -> str:
if self.tg_username is not None:
return f"@{self.tg_username}"
elif self.tg_last_name is not None:
return f"{self.tg_first_name} {self.tg_last_name}"
if self.username is not None:
return f"@{self.username}"
elif self.last_name is not None:
return f"{self.first_name} {self.last_name}"
else:
return self.tg_first_name
return self.first_name