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

refactored

Browse Source
This commit is contained in:
Untone
2024-01-23 03:06:48 +03:00
parent c6088c5705
commit 8f846b6f7a
2 changed files with 18 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
resolvers/reader.py
View File

@@ -13,18 +13,7 @@ from services.schema import query
from services.search import SearchService
from services.viewed import ViewedStorage
from resolvers.topic import get_random_topic
def add_stat_columns(q):
aliased_reaction = aliased(Reaction)
q = q.outerjoin(aliased_reaction).add_columns(
func.sum(case((aliased_reaction.kind == ReactionKind.COMMENT.value, 1), else_=0)).label("comments_stat"),
func.sum(case((aliased_reaction.kind == ReactionKind.LIKE.value, 1), else_=0)).label("likes_stat"),
func.sum(case((aliased_reaction.kind == ReactionKind.DISLIKE.value, 1), else_=0)).label("dislikes_stat"),
func.max(case((aliased_reaction.kind != ReactionKind.COMMENT.value, None),else_=aliased_reaction.created_at)).label("last_comment"),
)
return q
from resolvers.reaction import add_stat_columns
def apply_filters(q, filters, author_id=None): # noqa: C901
@@ -59,7 +48,7 @@ async def get_shout(_, _info, slug=None, shout_id=None):
joinedload(Shout.topics),
)
q = add_stat_columns(q)
q, _ar = add_stat_columns(q)
if slug is not None:
q = q.filter(Shout.slug == slug)
@@ -133,7 +122,7 @@ async def load_shouts_by(_, _info, options):
)
# stats
q = add_stat_columns(q)
q, _ar = add_stat_columns(q)
# filters
q = apply_filters(q, options.get("filters", {}))
@@ -242,7 +231,7 @@ async def load_shouts_feed(_, info, options):
.where(and_(Shout.published_at.is_not(None), Shout.deleted_at.is_(None), Shout.id.in_(subquery)))
)
q = add_stat_columns(q)
q, _ar = add_stat_columns(q)
q = apply_filters(q, options.get("filters", {}), reader.id)
order_by = options.get("order_by", Shout.published_at)
@@ -342,7 +331,7 @@ async def load_shouts_unrated(_, info, limit: int = 50, offset: int = 0):
# 3 or fewer votes is 0, 1, 2 or 3 votes (null, reaction id1, reaction id2, reaction id3)
q = q.having(func.count(distinct(Reaction.id)) <= 4)
q = add_stat_columns(q)
q, _ar = add_stat_columns(q)
q = q.group_by(Shout.id).order_by(func.random()).limit(limit).offset(offset)
user_id = info.context.get("user_id")
@@ -408,7 +397,7 @@ async def load_shouts_random_top(_, _info, options):
.where(Shout.id.in_(subquery))
)
q = add_stat_columns(q)
q, _ar = add_stat_columns(q)
limit = options.get("limit", 10)
q = q.group_by(Shout.id).order_by(func.random()).limit(limit)
@@ -431,7 +420,7 @@ async def load_shouts_random_topic(_, info, limit: int = 10):
.filter(and_(Shout.deleted_at.is_(None), Shout.visibility == "public", Shout.topics.any(slug=topic.slug)))
)
q = add_stat_columns(q)
q, _ar = add_stat_columns(q)
q = q.group_by(Shout.id).order_by(desc(Shout.created_at)).limit(limit)