138 lines
4.0 KiB
Python
138 lines
4.0 KiB
Python
from sqlalchemy import func, distinct, select, join
|
|
from sqlalchemy.orm import aliased
|
|
|
|
from orm.topic import TopicFollower, Topic
|
|
from resolvers.rating import load_author_ratings
|
|
from services.db import local_session
|
|
from orm.author import AuthorFollower, Author
|
|
from orm.shout import ShoutTopic, ShoutAuthor
|
|
from services.logger import root_logger as logger
|
|
|
|
|
|
def add_topic_stat_columns(q):
|
|
aliased_shout_author = aliased(ShoutAuthor)
|
|
aliased_topic_follower = aliased(TopicFollower)
|
|
aliased_shout_topic = aliased(ShoutTopic)
|
|
|
|
q = (
|
|
q.outerjoin(aliased_shout_topic, aliased_shout_topic.topic == Topic.id)
|
|
.add_columns(
|
|
func.count(distinct(aliased_shout_topic.shout)).label('shouts_stat')
|
|
)
|
|
.outerjoin(
|
|
aliased_shout_author,
|
|
aliased_shout_topic.shout == aliased_shout_author.shout,
|
|
)
|
|
.add_columns(
|
|
func.count(distinct(aliased_shout_author.author)).label('authors_stat')
|
|
)
|
|
.outerjoin(aliased_topic_follower)
|
|
.add_columns(
|
|
func.count(distinct(aliased_topic_follower.follower)).label(
|
|
'followers_stat'
|
|
)
|
|
)
|
|
)
|
|
|
|
q = q.group_by(Topic.id)
|
|
|
|
return q
|
|
|
|
|
|
def add_author_stat_columns(q):
|
|
aliased_shout_author = aliased(ShoutAuthor)
|
|
aliased_author_authors = aliased(AuthorFollower)
|
|
aliased_author_followers = aliased(AuthorFollower)
|
|
|
|
q = (
|
|
q.outerjoin(aliased_shout_author, aliased_shout_author.author == Author.id)
|
|
.add_columns(
|
|
func.count(distinct(aliased_shout_author.shout)).label('shouts_stat')
|
|
)
|
|
.outerjoin(aliased_author_authors, aliased_author_authors.follower == Author.id)
|
|
.add_columns(
|
|
func.count(distinct(aliased_shout_author.author)).label('authors_stat')
|
|
)
|
|
.outerjoin(
|
|
aliased_author_followers, aliased_author_followers.author == Author.id
|
|
)
|
|
.add_columns(
|
|
func.count(distinct(aliased_author_followers.follower)).label(
|
|
'followers_stat'
|
|
)
|
|
)
|
|
)
|
|
|
|
q = q.group_by(Author.id)
|
|
|
|
return q
|
|
|
|
|
|
def execute_with_ministat(q):
|
|
records = []
|
|
logger.debug(f'execute with ministat: {q}')
|
|
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, ratings=False):
|
|
q = add_author_stat_columns(q)
|
|
authors = execute_with_ministat(q)
|
|
if ratings:
|
|
authors_with_ratings = []
|
|
for author in authors:
|
|
authors_with_ratings.append(load_author_ratings(author))
|
|
return authors_with_ratings
|
|
return authors
|
|
|
|
|
|
def get_topics_with_stat(q):
|
|
q = add_topic_stat_columns(q)
|
|
return execute_with_ministat(q)
|
|
|
|
|
|
def author_follows_authors(author_id: int):
|
|
af = aliased(AuthorFollower, name='af')
|
|
q = (
|
|
select(Author)
|
|
.select_from(join(Author, af, Author.id == af.author))
|
|
.where(af.follower == author_id)
|
|
)
|
|
q = add_author_stat_columns(q)
|
|
return execute_with_ministat(q)
|
|
|
|
|
|
def author_follows_topics(author_id: int):
|
|
q = (
|
|
select(Topic)
|
|
.select_from(join(Topic, TopicFollower, Topic.id == TopicFollower.topic))
|
|
.where(TopicFollower.follower == author_id)
|
|
)
|
|
|
|
q = add_topic_stat_columns(q)
|
|
return execute_with_ministat(q)
|
|
|
|
|
|
def query_follows(author_id: int):
|
|
try:
|
|
topics = author_follows_topics(author_id)
|
|
authors = author_follows_authors(author_id)
|
|
return {
|
|
'topics': topics,
|
|
'authors': authors,
|
|
'communities': [{'id': 1, 'name': 'Дискурс', 'slug': 'discours'}],
|
|
}
|
|
except Exception as e:
|
|
import traceback
|
|
|
|
logger.debug(traceback.format_exc())
|
|
raise Exception(e)
|