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

postprocess-query-for-order-3
All checks were successful
Deploy to core / deploy (push) Successful in 2m9s
Details

Browse Source
This commit is contained in:
Untone 2024-02-14 12:05:19 +03:00
parent 9b174d94c6
commit 2e60fd2cc7
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
resolvers/author.py
View File

@ -2,7 +2,7 @@ import logging
import time import time
from typing import List from typing import List
from sqlalchemy import and_, distinct, func, select from sqlalchemy import and_, desc, distinct, func, select
from sqlalchemy.orm import aliased from sqlalchemy.orm import aliased
from orm.author import Author, AuthorFollower, AuthorRating from orm.author import Author, AuthorFollower, AuthorRating
@ -25,7 +25,7 @@ logger = logging.getLogger('\t[resolvers.author]\t')
logger.setLevel(logging.DEBUG) logger.setLevel(logging.DEBUG)
def add_author_stat_columns(q): def add_author_stat_columns(q, order=''):
shout_author_aliased = aliased(ShoutAuthor) shout_author_aliased = aliased(ShoutAuthor)
q = q.outerjoin(shout_author_aliased).add_columns( q = q.outerjoin(shout_author_aliased).add_columns(
func.count(distinct(shout_author_aliased.shout)).label('shouts_stat') func.count(distinct(shout_author_aliased.shout)).label('shouts_stat')
@ -42,6 +42,8 @@ def add_author_stat_columns(q):
) )
q = q.group_by(Author.id) q = q.group_by(Author.id)
if order == 'followers' or order == 'shouts':
q = q.order_By(desc(f'{order}_stat'))
return q return q
@ -207,7 +209,7 @@ async def get_author_id(_, _info, user: str):
@query.field('load_authors_by') @query.field('load_authors_by')
async def load_authors_by(_, _info, by, limit, offset): async def load_authors_by(_, _info, by, limit, offset):
q = select(Author) q = select(Author)
q = add_author_stat_columns(q) q = add_author_stat_columns(q, order=by.get('order'))
if by.get('slug'): if by.get('slug'):
q = q.filter(Author.slug.ilike(f"%{by['slug']}%")) q = q.filter(Author.slug.ilike(f"%{by['slug']}%"))
elif by.get('name'): elif by.get('name'):
@ -225,9 +227,6 @@ async def load_authors_by(_, _info, by, limit, offset):
q = q.limit(limit).offset(offset) q = q.limit(limit).offset(offset)
authors = await get_authors_from_query(q) authors = await get_authors_from_query(q)
order = by.get('order')
if order:
authors = sorted(authors, key=lambda a: a.stat.get(order, 0), reverse=True)
return authors return authors