4d22d9bce2
Ported in logic from upstream to use SQLAlchemy eager loading instead of repeated queries when building a submission_listing. Adjusted loaded relationships to include only those used on TheMotte. Using test data from seed_db, before and after: GET / |----------|--------|--------|--------|--------|--------|------------| | Database | SELECT | INSERT | UPDATE | DELETE | Totals | Duplicates | |----------|--------|--------|--------|--------|--------|------------| | default | 83 | 0 | 0 | 0 | 83 | 72 | |----------|--------|--------|--------|--------|--------|------------| Total queries: 83 in 0.031s GET / |----------|--------|--------|--------|--------|--------|------------| | Database | SELECT | INSERT | UPDATE | DELETE | Totals | Duplicates | |----------|--------|--------|--------|--------|--------|------------| | default | 14 | 0 | 0 | 0 | 14 | 0 | |----------|--------|--------|--------|--------|--------|------------| Total queries: 14 in 0.00718s
68 lines
1.7 KiB
Python
68 lines
1.7 KiB
Python
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 files.helpers.const import *
|
|
from files.helpers.assetcache import assetcache_path
|
|
from datetime import datetime
|
|
from json import loads
|
|
|
|
class BadgeDef(Base):
|
|
__tablename__ = "badge_defs"
|
|
__table_args__ = (
|
|
UniqueConstraint('name', name='badge_def_name_unique'),
|
|
)
|
|
|
|
id = Column(Integer, primary_key=True, autoincrement=True)
|
|
name = Column(String, nullable=False)
|
|
description = Column(String)
|
|
|
|
def __repr__(self):
|
|
return f"<BadgeDef(id={self.id})>"
|
|
|
|
class Badge(Base):
|
|
|
|
__tablename__ = "badges"
|
|
|
|
user_id = Column(Integer, ForeignKey('users.id'), primary_key=True)
|
|
badge_id = Column(Integer, ForeignKey('badge_defs.id'), primary_key=True)
|
|
description = Column(String)
|
|
url = Column(String)
|
|
|
|
Index('badges_badge_id_idx', badge_id)
|
|
|
|
user = relationship("User", viewonly=True)
|
|
badge = relationship("BadgeDef",
|
|
primaryjoin="foreign(Badge.badge_id) == remote(BadgeDef.id)",
|
|
lazy="joined", innerjoin=True, viewonly=True)
|
|
|
|
def __repr__(self):
|
|
return f"<Badge(user_id={self.user_id}, badge_id={self.badge_id})>"
|
|
|
|
@property
|
|
@lazy
|
|
def text(self):
|
|
if self.description: text = self.description
|
|
elif self.badge.description: text = self.badge.description
|
|
else: return self.name
|
|
return f'{self.name} - {text}'
|
|
|
|
@property
|
|
@lazy
|
|
def name(self):
|
|
return self.badge.name
|
|
|
|
@property
|
|
@lazy
|
|
def path(self):
|
|
return assetcache_path(f'images/badges/{self.badge_id}.webp')
|
|
|
|
@property
|
|
@lazy
|
|
def json(self):
|
|
return {'text': self.text,
|
|
'name': self.name,
|
|
'url': self.url,
|
|
'icon_url':self.path
|
|
}
|