diff --git a/drama/classes/user.py b/drama/classes/user.py index a7784a0c6..d690d5c15 100644 --- a/drama/classes/user.py +++ b/drama/classes/user.py @@ -155,6 +155,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())) @@ -614,6 +619,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: