User who has valid manual sub is premium

This commit is contained in:
Son NK 2020-02-23 16:31:55 +07:00
parent b2eb084720
commit fb62322794
1 changed files with 4 additions and 0 deletions

View File

@ -194,6 +194,10 @@ class User(db.Model, ModelMixin, UserMixin):
if sub:
return True
manual_sub: ManualSubscription = ManualSubscription.get_by(user_id=self.id)
if manual_sub and manual_sub.end_at > arrow.now():
return True
return False
def in_trial(self):