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 import time
from typing import List 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 sqlalchemy.orm import aliased
from orm.author import Author, AuthorFollower, AuthorRating 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.shout import Shout, ShoutAuthor, ShoutTopic
from orm.topic import Topic from orm.topic import Topic
from resolvers.follower import query_follows 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.auth import login_required
from services.db import local_session from services.db import local_session
from services.rediscache import redis from services.rediscache import redis
@ -268,11 +268,31 @@ def create_author(user_id: str, slug: str, name: str = ''):
@query.field('get_author_followers') @query.field('get_author_followers')
def get_author_followers(_, _info, slug) -> List[Author]: def get_author_followers(_, _info, slug):
q = select(Author) author_alias = aliased(Author)
author_follower_alias = aliased(AuthorFollower)
shout_author_alias = aliased(ShoutAuthor)
q = ( q = (
q.join(AuthorFollower, AuthorFollower.follower == Author.id) select(author_alias)
.join(Author, Author.id == AuthorFollower.author) .join(author_follower_alias, author_follower_alias.author == author_alias.id)
.where(Author.slug == slug) .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): def add_topic_stat_columns(q):
aliased_shout_author = aliased(ShoutAuthor) aliased_shout_authors = aliased(ShoutAuthor)
aliased_topic_follower = aliased(TopicFollower) aliased_topic_followers = aliased(TopicFollower)
aliased_topic = aliased(Topic)
q = ( 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')) .add_columns(func.count(distinct(ShoutTopic.shout)).label('shouts_stat'))
.outerjoin(aliased_shout_author, ShoutTopic.shout == aliased_shout_author.shout) .outerjoin(aliased_shout_authors, ShoutTopic.shout == aliased_shout_authors.shout)
.add_columns(func.count(distinct(aliased_shout_author.author)).label('authors_stat')) .add_columns(func.count(distinct(aliased_shout_authors.author)).label('authors_stat'))
.outerjoin(aliased_topic_follower, aliased_topic_follower.topic == Topic.id) .outerjoin(aliased_topic_followers, aliased_topic_followers.topic == aliased_topic.id)
.add_columns(func.count(distinct(aliased_topic_follower.follower)).label('followers_stat')) .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 return q
@ -28,16 +28,17 @@ def add_topic_stat_columns(q):
def add_author_stat_columns(q): def add_author_stat_columns(q):
aliased_author_followers = aliased(AuthorFollower) aliased_author_followers = aliased(AuthorFollower)
aliased_author_authors = aliased(AuthorFollower) aliased_author_authors = aliased(AuthorFollower)
aliased_author = aliased(Author)
q = ( 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')) .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')) .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')) .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 return q