skorotkiewicz/rDrama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

sneed

Browse source
This commit is contained in:
fireworks88 2021-07-24 21:54:34 +02:00
parent 09d18ba4ed
commit 315c56189e
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
drama/classes/user.py
View file

@ -154,6 +154,11 @@ class User(Base, Stndrd, Age_times):
lazy="dynamic",
primaryjoin="User.id==SaveRelationship.user_id")
awards = relationship(
"AwardRelationship",
primaryjoin="User.id==AwardRelationship.user_id"
)
# properties defined as SQL server-side functions
referral_count = deferred(Column(Integer, server_default=FetchedValue()))
follower_count = deferred(Column(Integer, server_default=FetchedValue()))
@ -613,6 +618,9 @@ class User(Base, Stndrd, Age_times):
pic = random.randint(1, 50)
return f"/assets/images/defaultpictures/{pic}.png"
def has_award(self, kind):
return bool(len([x for x in self.awards if x.kind == kind]))
@property
def profile_url(self):
if self.has_profile and self.profileurl: