forked from rDrama/rDrama
1
0
Fork 0
master
Aevann1 2021-10-18 17:58:06 +02:00
parent f282b804bb
commit 8f0081749e
1 changed files with 9 additions and 4 deletions

View File

@ -3,6 +3,7 @@ from sqlalchemy.orm import relationship
from files.__main__ import Base, app from files.__main__ import Base, app
from os import environ from os import environ
from files.helpers.lazy import lazy from files.helpers.lazy import lazy
from datetime import datetime
site_name = environ.get("SITE_NAME").strip() site_name = environ.get("SITE_NAME").strip()
@ -48,7 +49,9 @@ class Badge(Base):
badge_id = Column(Integer, ForeignKey("badge_defs.id")) badge_id = Column(Integer, ForeignKey("badge_defs.id"))
description = Column(String) description = Column(String)
url = Column(String) url = Column(String)
badge = relationship("BadgeDef", viewonly=True) badge = relationship("BadgeDef", viewonly=True)
user = relationship("User", viewonly=True)
def __repr__(self): def __repr__(self):
@ -57,10 +60,12 @@ class Badge(Base):
@property @property
@lazy @lazy
def text(self): def text(self):
if self.description: if self.name == "Agendaposter":
return self.description ti = self.user.agendaposter_expires_utc
else: if ti: return self.badge.description + " until " + datetime.utcfromtimestamp(ti).strftime('%Y-%m-%d %H:%M:%S')
return self.badge.description else: return self.badge.description + " permenantly"
elif self.description: return self.description
else: return self.badge.description
@property @property
@lazy @lazy