from sqlalchemy import * from sqlalchemy.orm import relationship from files.__main__ import Base, app from os import environ from files.helpers.lazy import lazy from datetime import datetime site_name = environ.get("SITE_NAME").strip() class BadgeDef(Base): __tablename__ = "badge_defs" id = Column(BigInteger, primary_key=True) name = Column(String) description = Column(String) icon = Column(String) kind = Column(Integer, default=1) qualification_expr = Column(String) def __repr__(self): return f"" @property @lazy def path(self): return f"/assets/images/badges/{self.icon}.gif" @property @lazy def json_core(self): return { "name": self.name, "description": self.description, "icon": self.icon } class Badge(Base): __tablename__ = "badges" id = Column(Integer, primary_key=True) user_id = Column(Integer, ForeignKey('users.id')) badge_id = Column(Integer, ForeignKey("badge_defs.id")) description = Column(String) url = Column(String) badge = relationship("BadgeDef", viewonly=True) user = relationship("User", viewonly=True) def __repr__(self): return f"" @property @lazy def text(self): if self.name == "Agendaposter": ti = self.user.agendaposter_expires_utc if ti: return self.badge.description + " until " + datetime.utcfromtimestamp(ti).strftime('%Y-%m-%d %H:%M:%S') else: return self.badge.description + " permanently" elif self.description: return self.description else: return self.badge.description @property @lazy def type(self): return self.badge.id @property @lazy def name(self): return self.badge.name @property @lazy def path(self): return self.badge.path @property @lazy def json_core(self): return {'text': self.text, 'name': self.name, 'url': self.url, 'icon_url':f"https://{app.config['SERVER_NAME']}{self.path}" } @property @lazy def json(self): return self.json_core