discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

virtual-cols-fix
All checks were successful
Deploy on push / deploy (push) Successful in 1m6s
Details

Browse Source
This commit is contained in:
Untone 2024-10-21 11:08:16 +03:00
parent 160f02e67f
commit 0cf963240e
1 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
orm/community.py
View File

@ -1,8 +1,7 @@
import time import time
from requests import Session
from sqlalchemy import Column, ForeignKey, Integer, String, func from sqlalchemy import Column, ForeignKey, Integer, String, func
from sqlalchemy.ext.hybrid import hybrid_method from sqlalchemy.ext.hybrid import hybrid_property
from sqlalchemy.orm import relationship from sqlalchemy.orm import relationship
from orm.author import Author from orm.author import Author
@ -12,7 +11,6 @@ from services.db import Base
class CommunityFollower(Base): class CommunityFollower(Base):
__tablename__ = "community_author" __tablename__ = "community_author"
id = None # type: ignore
author = Column(ForeignKey("author.id"), primary_key=True) author = Column(ForeignKey("author.id"), primary_key=True)
community = Column(ForeignKey("community.id"), primary_key=True) community = Column(ForeignKey("community.id"), primary_key=True)
joined_at = Column(Integer, nullable=False, default=lambda: int(time.time())) joined_at = Column(Integer, nullable=False, default=lambda: int(time.time()))
@ -30,16 +28,24 @@ class Community(Base):
authors = relationship(Author, secondary="community_author") authors = relationship(Author, secondary="community_author")
@hybrid_method @hybrid_property
def get_stats(self, session: Session): def stat(self):
from orm.shout import ShoutCommunity # Импорт здесь во избежание циклических зависимостей return CommunityStats(self)
shouts_count = (
session.query(func.count(ShoutCommunity.shout_id)).filter(ShoutCommunity.community_id == self.id).scalar()
)
followers_count = ( class CommunityStats:
session.query(func.count(CommunityFollower.author)).filter(CommunityFollower.community == self.id).scalar() def __init__(self, community):
) self.community = community
return {"shouts": shouts_count, "followers": followers_count} @property
def shouts(self):
from orm.shout import ShoutCommunity
return self.community.session.query(func.count(ShoutCommunity.shout_id))\
.filter(ShoutCommunity.community_id == self.community.id)\
.scalar()
@property
def followers(self):
return self.community.session.query(func.count(CommunityFollower.author))\
.filter(CommunityFollower.community == self.community.id)\
.scalar()