68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
from sqlalchemy import func, distinct
|
|
from sqlalchemy.orm import aliased
|
|
|
|
from orm.topic import TopicFollower, Topic
|
|
from services.db import local_session
|
|
from orm.author import AuthorFollower, Author
|
|
from orm.shout import ShoutTopic, ShoutAuthor
|
|
|
|
|
|
def add_topic_stat_columns(q):
|
|
aliased_shout_authors = aliased(ShoutAuthor)
|
|
aliased_topic_followers = aliased(TopicFollower)
|
|
aliased_topic = aliased(Topic)
|
|
q = (
|
|
q.outerjoin(ShoutTopic, aliased_topic.id == ShoutTopic.c.topic)
|
|
.add_columns(func.count(distinct(ShoutTopic.c.shout)).label('shouts_stat'))
|
|
.outerjoin(aliased_shout_authors, ShoutTopic.c.shout == aliased_shout_authors.c.shout)
|
|
.add_columns(func.count(distinct(aliased_shout_authors.c.author)).label('authors_stat'))
|
|
.outerjoin(aliased_topic_followers, aliased_topic.c.id == aliased_topic_followers.c.topic)
|
|
.add_columns(func.count(distinct(aliased_topic_followers.c.follower)).label('followers_stat'))
|
|
)
|
|
|
|
q = q.group_by(aliased_topic.id)
|
|
|
|
return q
|
|
|
|
|
|
def add_author_stat_columns(q):
|
|
aliased_author_followers = aliased(AuthorFollower)
|
|
aliased_author_authors = aliased(AuthorFollower)
|
|
aliased_author = aliased(Author)
|
|
q = (
|
|
q.outerjoin(ShoutAuthor, aliased_author.id == ShoutAuthor.author)
|
|
.add_columns(func.count(distinct(ShoutAuthor.shout)).label('shouts_stat'))
|
|
.outerjoin(aliased_author_authors, AuthorFollower.follower == aliased_author.id)
|
|
.add_columns(func.count(distinct(aliased_author_authors.author)).label('authors_stat'))
|
|
.outerjoin(aliased_author_followers, AuthorFollower.author == aliased_author.id)
|
|
.add_columns(func.count(distinct(aliased_author_followers.follower)).label('followers_stat'))
|
|
)
|
|
|
|
q = q.group_by(aliased_author.id)
|
|
|
|
return q
|
|
|
|
|
|
def execute_with_ministat(q):
|
|
records = []
|
|
with local_session() as session:
|
|
for [entity, shouts_stat, authors_stat, followers_stat] in session.execute(q):
|
|
entity.stat = {
|
|
'shouts': shouts_stat,
|
|
'authors': authors_stat,
|
|
'followers': followers_stat
|
|
}
|
|
records.append(entity)
|
|
|
|
return records
|
|
|
|
|
|
def get_authors_with_stat(q):
|
|
q = add_author_stat_columns(q)
|
|
return execute_with_ministat(q)
|
|
|
|
|
|
def get_topics_with_stat(q):
|
|
q = add_topic_stat_columns(q)
|
|
return execute_with_ministat(q)
|