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

aliasing
All checks were successful
Deploy to core / deploy (push) Successful in 1m25s
Details

Browse Source
This commit is contained in:
Untone 2024-02-23 03:28:46 +03:00
parent 8d97463c1d
commit ef7f2d7b92
2 changed files with 42 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
resolvers/author.py
View File

@ -2,7 +2,7 @@ import json
import time
from typing import List
from sqlalchemy import and_, desc, select, or_
from sqlalchemy import and_, desc, select, or_, distinct, func
from sqlalchemy.orm import aliased
from orm.author import Author, AuthorFollower, AuthorRating
@ -10,7 +10,7 @@ from orm.reaction import Reaction, ReactionKind
from orm.shout import Shout, ShoutAuthor, ShoutTopic
from orm.topic import Topic
from resolvers.follower import query_follows
from resolvers.stat import get_authors_with_stat
from resolvers.stat import get_authors_with_stat, unpack_stat
from services.auth import login_required
from services.db import local_session
from services.rediscache import redis
@ -268,11 +268,31 @@ def create_author(user_id: str, slug: str, name: str = ''):
@query.field('get_author_followers')
def get_author_followers(_, _info, slug) -> List[Author]:
q = select(Author)
def get_author_followers(_, _info, slug):
author_alias = aliased(Author)
author_follower_alias = aliased(AuthorFollower)
shout_author_alias = aliased(ShoutAuthor)
q = (
q.join(AuthorFollower, AuthorFollower.follower == Author.id)
.join(Author, Author.id == AuthorFollower.author)
.where(Author.slug == slug)
select(author_alias)
.join(author_follower_alias, author_follower_alias.author == author_alias.id)
.join(Author, Author.id == author_follower_alias.follower)
.filter(author_alias.slug == slug)
.add_columns(
func.count(distinct(shout_author_alias.shout)).label('shouts_stat'),
func.count(distinct(author_follower_alias.author)).label('authors_stat'),
func.count(distinct(author_follower_alias.follower)).label('followers_stat')
)
.outerjoin(shout_author_alias, author_alias.id == shout_author_alias.author)
.outerjoin(
aliased(AuthorFollower, name="author_follower_1"),
author_follower_alias.follower == author_alias.id
)
.outerjoin(
aliased(AuthorFollower, name="author_follower_2"),
author_follower_alias.author == author_alias.id
)
.group_by(author_alias.id)
)
return get_authors_with_stat(q)
return unpack_stat(q)

27
resolvers/stat.py
View File

@ -8,19 +8,19 @@ from orm.shout import ShoutTopic, ShoutAuthor
def add_topic_stat_columns(q):
aliased_shout_author = aliased(ShoutAuthor)
aliased_topic_follower = aliased(TopicFollower)
aliased_shout_authors = aliased(ShoutAuthor)
aliased_topic_followers = aliased(TopicFollower)
aliased_topic = aliased(Topic)
q = (
q.outerjoin(ShoutTopic, Topic.id == ShoutTopic.topic)
q.outerjoin(ShoutTopic, aliased_topic.id == ShoutTopic.topic)
.add_columns(func.count(distinct(ShoutTopic.shout)).label('shouts_stat'))
.outerjoin(aliased_shout_author, ShoutTopic.shout == aliased_shout_author.shout)
.add_columns(func.count(distinct(aliased_shout_author.author)).label('authors_stat'))
.outerjoin(aliased_topic_follower, aliased_topic_follower.topic == Topic.id)
.add_columns(func.count(distinct(aliased_topic_follower.follower)).label('followers_stat'))
.outerjoin(aliased_shout_authors, ShoutTopic.shout == aliased_shout_authors.shout)
.add_columns(func.count(distinct(aliased_shout_authors.author)).label('authors_stat'))
.outerjoin(aliased_topic_followers, aliased_topic_followers.topic == aliased_topic.id)
.add_columns(func.count(distinct(aliased_topic_followers.follower)).label('followers_stat'))
)
q = q.group_by(Topic.id)
q = q.group_by(aliased_topic.id)
return q
@ -28,16 +28,17 @@ def add_topic_stat_columns(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, Author.id == ShoutAuthor.author)
q.outerjoin(ShoutAuthor, aliased_author.id == ShoutAuthor.author)
.add_columns(func.count(distinct(ShoutAuthor.shout)).label('shouts_stat'))
.outerjoin(aliased_author_authors, AuthorFollower.follower == Author.id)
.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 == Author.id)
.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(Author.id)
q = q.group_by(aliased_author.id)
return q