reader-loads-move

This commit is contained in:
Untone 2024-08-07 14:18:05 +03:00
parent a01ca30f5b
commit 572f63f12b
4 changed files with 251 additions and 170 deletions

View File

@ -23,8 +23,6 @@ from resolvers.reaction import (
create_reaction, create_reaction,
delete_reaction, delete_reaction,
load_reactions_by, load_reactions_by,
load_shouts_followed,
load_shouts_followed_by,
update_reaction, update_reaction,
load_shout_comments, load_shout_comments,
load_shout_ratings, load_shout_ratings,
@ -40,6 +38,8 @@ from resolvers.reader import (
load_shouts_unrated, load_shouts_unrated,
load_shouts_coauthored, load_shouts_coauthored,
load_shouts_discussed, load_shouts_discussed,
load_shouts_followed,
load_shouts_followed_by,
) )
from resolvers.topic import ( from resolvers.topic import (
get_topic, get_topic,

View File

@ -1,9 +1,7 @@
import time import time
from typing import List
from sqlalchemy import and_, asc, case, desc, func, select, text from sqlalchemy import and_, asc, case, desc, func, select
from sqlalchemy.orm import aliased, joinedload from sqlalchemy.orm import aliased
from sqlalchemy.sql import union
from orm.author import Author from orm.author import Author
from orm.rating import PROPOSAL_REACTIONS, RATING_REACTIONS, is_negative, is_positive from orm.rating import PROPOSAL_REACTIONS, RATING_REACTIONS, is_negative, is_positive
@ -17,14 +15,21 @@ from services.db import local_session
from utils.logger import root_logger as logger from utils.logger import root_logger as logger
from services.notify import notify_reaction from services.notify import notify_reaction
from services.schema import mutation, query from services.schema import mutation, query
from services.viewed import ViewedStorage
def add_reaction_stat_columns(q, aliased_reaction): def add_reaction_stat_columns(q, aliased_reaction):
# Adjust case statements to use positional arguments """
Добавляет статистические колонки к запросу реакций.
:param q: SQL-запрос для реакций.
:param aliased_reaction: Алиас для таблицы реакций.
:return: Запрос с добавленными колонками статистики.
"""
# Присоединение реакций и добавление статистических колонок
q = q.outerjoin(aliased_reaction, aliased_reaction.deleted_at.is_(None)).add_columns( q = q.outerjoin(aliased_reaction, aliased_reaction.deleted_at.is_(None)).add_columns(
# Подсчет комментариев
func.count(case((aliased_reaction.body.is_not(None), 1), else_=0)).label("comments_stat"), func.count(case((aliased_reaction.body.is_not(None), 1), else_=0)).label("comments_stat"),
# Calculate net likes as likes - dislikes # Вычисление рейтинга как разница между лайками и дизлайками
func.sum( func.sum(
case( case(
(aliased_reaction.kind == ReactionKind.LIKE.value, 1), (aliased_reaction.kind == ReactionKind.LIKE.value, 1),
@ -32,14 +37,19 @@ def add_reaction_stat_columns(q, aliased_reaction):
else_=0, else_=0,
) )
).label("rating_stat"), ).label("rating_stat"),
func.max(aliased_reaction.created_at).label("last_reacted_at"),
) )
return q return q
def is_featured_author(session, author_id): def is_featured_author(session, author_id) -> bool:
"""checks if author has at least one featured publication""" """
Проверяет, есть ли у автора хотя бы одна опубликованная статья.
:param session: Сессия базы данных.
:param author_id: Идентификатор автора.
:return: True, если у автора есть хотя бы одна опубликованная статья, иначе False.
"""
return ( return (
session.query(Shout) session.query(Shout)
.where(Shout.authors.any(id=author_id)) .where(Shout.authors.any(id=author_id))
@ -49,12 +59,19 @@ def is_featured_author(session, author_id):
) )
def check_to_feature(session, approver_id, reaction): def check_to_feature(session, approver_id, reaction) -> bool:
"""set shout to public if publicated approvers amount > 4""" """
Устанавливает публикацию в открытый доступ, если количество голосов превышает 4.
:param session: Сессия базы данных.
:param approver_id: Идентификатор утверждающего автора.
:param reaction: Объект реакции.
:return: True, если нужно установить публикацию в открытый доступ, иначе False.
"""
if not reaction.reply_to and is_positive(reaction.kind): if not reaction.reply_to and is_positive(reaction.kind):
if is_featured_author(session, approver_id): if is_featured_author(session, approver_id):
approvers = [approver_id] approvers = [approver_id]
# now count how many approvers are voted already # Подсчет количества голосующих утверждающих
reacted_readers = session.query(Reaction).where(Reaction.shout == reaction.shout).all() reacted_readers = session.query(Reaction).where(Reaction.shout == reaction.shout).all()
for reacted_reader in reacted_readers: for reacted_reader in reacted_readers:
if is_featured_author(session, reacted_reader.id): if is_featured_author(session, reacted_reader.id):
@ -64,8 +81,15 @@ def check_to_feature(session, approver_id, reaction):
return False return False
def check_to_unfeature(session, rejecter_id, reaction): def check_to_unfeature(session, rejecter_id, reaction) -> bool:
"""unfeature any shout if 20% of reactions are negative""" """
Убирает публикацию из открытого доступа, если 20% реакций негативные.
:param session: Сессия базы данных.
:param rejecter_id: Идентификатор отклоняющего автора.
:param reaction: Объект реакции.
:return: True, если нужно убрать публикацию из открытого доступа, иначе False.
"""
if not reaction.reply_to and is_negative(reaction.kind): if not reaction.reply_to and is_negative(reaction.kind):
if is_featured_author(session, rejecter_id): if is_featured_author(session, rejecter_id):
reactions = ( reactions = (
@ -90,6 +114,12 @@ def check_to_unfeature(session, rejecter_id, reaction):
async def set_featured(session, shout_id): async def set_featured(session, shout_id):
"""
Устанавливает публикацию в открытый доступ и обновляет роль автора.
:param session: Сессия базы данных.
:param shout_id: Идентификатор публикации.
"""
s = session.query(Shout).where(Shout.id == shout_id).first() s = session.query(Shout).where(Shout.id == shout_id).first()
s.featured_at = int(time.time()) s.featured_at = int(time.time())
Shout.update(s, {"featured_at": int(time.time())}) Shout.update(s, {"featured_at": int(time.time())})
@ -101,59 +131,85 @@ async def set_featured(session, shout_id):
def set_unfeatured(session, shout_id): def set_unfeatured(session, shout_id):
"""
Убирает публикацию из открытого доступа.
:param session: Сессия базы данных.
:param shout_id: Идентификатор публикации.
"""
s = session.query(Shout).where(Shout.id == shout_id).first() s = session.query(Shout).where(Shout.id == shout_id).first()
Shout.update(s, {"featured_at": None}) Shout.update(s, {"featured_at": None})
session.add(s) session.add(s)
session.commit() session.commit()
async def _create_reaction(session, info, shout, author_id: int, reaction): async def _create_reaction(session, info, shout, author_id: int, reaction) -> dict:
"""
Создает новую реакцию и выполняет связанные с этим действия, такие как обновление счетчиков и уведомление.
:param session: Сессия базы данных.
:param info: Информация о контексте GraphQL.
:param shout: Объект публикации.
:param author_id: Идентификатор автора.
:param reaction: Словарь с данными реакции.
:return: Словарь с данными о созданной реакции.
"""
r = Reaction(**reaction) r = Reaction(**reaction)
session.add(r) session.add(r)
session.commit() session.commit()
rdict = r.dict() rdict = r.dict()
# пересчет счетчика комментариев # Пересчет счетчика комментариев
if str(r.kind) == ReactionKind.COMMENT.value: if str(r.kind) == ReactionKind.COMMENT.value:
update_author_stat(author_id) update_author_stat(author_id)
# collaborative editing # Совместное редактирование
if rdict.get("reply_to") and r.kind in PROPOSAL_REACTIONS and author_id in shout.authors: if rdict.get("reply_to") and r.kind in PROPOSAL_REACTIONS and author_id in shout.authors:
handle_proposing(session, r, shout) handle_proposing(session, r, shout)
# рейтинг и саморегуляция # Рейтинг и саморегуляция
if r.kind in RATING_REACTIONS: if r.kind in RATING_REACTIONS:
# self-regultaion mechanics # Механизм саморегуляции
if check_to_unfeature(session, author_id, r): if check_to_unfeature(session, author_id, r):
set_unfeatured(session, shout.id) set_unfeatured(session, shout.id)
elif check_to_feature(session, author_id, r): elif check_to_feature(session, author_id, r):
await set_featured(session, shout.id) await set_featured(session, shout.id)
# follow if liked # Подписка, если понравилось
if r.kind == ReactionKind.LIKE.value: if r.kind == ReactionKind.LIKE.value:
try: try:
# reactions auto-following # Автоподписка при реакции
follow(None, info, "shout", shout.slug) follow(None, info, "shout", shout.slug)
except Exception: except Exception:
pass pass
# обновление счетчика комментариев в кеше # Обновление счетчика комментариев в кэше
if str(r.kind) == ReactionKind.COMMENT.value: if str(r.kind) == ReactionKind.COMMENT.value:
update_author_stat(author_id) update_author_stat(author_id)
rdict["shout"] = shout.dict() rdict["shout"] = shout.dict()
rdict["stat"] = {"commented": 0, "reacted": 0, "rating": 0} rdict["stat"] = {"commented": 0, "reacted": 0, "rating": 0}
# notifications call # Уведомление о создании
await notify_reaction(rdict, "create") await notify_reaction(rdict, "create")
return rdict return rdict
def prepare_new_rating(reaction: dict, shout_id: int, session, author_id: int): def prepare_new_rating(reaction: dict, shout_id: int, session, author_id: int):
"""
Проверяет возможность выставления новой оценки для публикации.
:param reaction: Словарь с данными реакции.
:param shout_id: Идентификатор публикации.
:param session: Сессия базы данных.
:param author_id: Идентификатор автора.
:return: Словарь с ошибкой или None.
"""
kind = reaction.get("kind") kind = reaction.get("kind")
opposite_kind = ReactionKind.DISLIKE.value if is_positive(kind) else ReactionKind.LIKE.value opposite_kind = ReactionKind.DISLIKE.value if is_positive(kind) else ReactionKind.LIKE.value
# Формирование запроса для проверки существующих оценок
q = select(Reaction).filter( q = select(Reaction).filter(
and_( and_(
Reaction.shout == shout_id, Reaction.shout == shout_id,
@ -166,6 +222,8 @@ def prepare_new_rating(reaction: dict, shout_id: int, session, author_id: int):
if reply_to and isinstance(reply_to, int): if reply_to and isinstance(reply_to, int):
q = q.filter(Reaction.reply_to == reply_to) q = q.filter(Reaction.reply_to == reply_to)
rating_reactions = session.execute(q).all() rating_reactions = session.execute(q).all()
# Проверка условий для выставления новой оценки
if rating_reactions: if rating_reactions:
same_rating = filter( same_rating = filter(
lambda r: r.created_by == author_id and r.kind == kind, lambda r: r.created_by == author_id and r.kind == kind,
@ -187,6 +245,13 @@ def prepare_new_rating(reaction: dict, shout_id: int, session, author_id: int):
@mutation.field("create_reaction") @mutation.field("create_reaction")
@login_required @login_required
async def create_reaction(_, info, reaction): async def create_reaction(_, info, reaction):
"""
Создает новую реакцию через GraphQL запрос.
:param info: Информация о контексте GraphQL.
:param reaction: Словарь с данными реакции.
:return: Словарь с информацией о созданной реакции или ошибкой.
"""
# logger.debug(f"{info.context} for {reaction}") # logger.debug(f"{info.context} for {reaction}")
info.context.get("user_id") info.context.get("user_id")
author_dict = info.context.get("author", {}) author_dict = info.context.get("author", {})
@ -232,6 +297,13 @@ async def create_reaction(_, info, reaction):
@mutation.field("update_reaction") @mutation.field("update_reaction")
@login_required @login_required
async def update_reaction(_, info, reaction): async def update_reaction(_, info, reaction):
"""
Обновляет существующую реакцию через GraphQL запрос.
:param info: Информация о контексте GraphQL.
:param reaction: Словарь с данными реакции.
:return: Словарь с информацией об обновленной реакции или ошибкой.
"""
logger.debug(f"{info.context} for {reaction}") logger.debug(f"{info.context} for {reaction}")
user_id = info.context.get("user_id") user_id = info.context.get("user_id")
roles = info.context.get("roles") roles = info.context.get("roles")
@ -247,12 +319,7 @@ async def update_reaction(_, info, reaction):
try: try:
result = session.execute(reaction_query).unique().first() result = session.execute(reaction_query).unique().first()
if result: if result:
[ [r, commented_stat, rating_stat] = result
r,
commented_stat,
rating_stat,
last_reacted_at,
] = result
if not r: if not r:
return {"error": "invalid reaction id"} return {"error": "invalid reaction id"}
@ -275,7 +342,6 @@ async def update_reaction(_, info, reaction):
session.commit() session.commit()
r.stat = { r.stat = {
# FIXME: "viewed": ViewedStorage.get_shout(r.shuot), sure, it is possible to collect reaction vews
"commented": commented_stat, "commented": commented_stat,
"rating": rating_stat, "rating": rating_stat,
} }
@ -295,6 +361,13 @@ async def update_reaction(_, info, reaction):
@mutation.field("delete_reaction") @mutation.field("delete_reaction")
@login_required @login_required
async def delete_reaction(_, info, reaction_id: int): async def delete_reaction(_, info, reaction_id: int):
"""
Удаляет существующую реакцию через GraphQL запрос.
:param info: Информация о контексте GraphQL.
:param reaction_id: Идентификатор удаляемой реакции.
:return: Словарь с информацией об удаленной реакции или ошибкой.
"""
logger.debug(f"{info.context} for {reaction_id}") logger.debug(f"{info.context} for {reaction_id}")
user_id = info.context.get("user_id") user_id = info.context.get("user_id")
author_id = info.context.get("author", {}).get("id") author_id = info.context.get("author", {}).get("id")
@ -312,7 +385,7 @@ async def delete_reaction(_, info, reaction_id: int):
session.delete(r) session.delete(r)
session.commit() session.commit()
# обновление счетчика комментариев в кеше # Обновление счетчика комментариев в кэше
if str(r.kind) == ReactionKind.COMMENT.value: if str(r.kind) == ReactionKind.COMMENT.value:
update_author_stat(author.id) update_author_stat(author.id)
await notify_reaction(reaction_dict, "delete") await notify_reaction(reaction_dict, "delete")
@ -324,6 +397,13 @@ async def delete_reaction(_, info, reaction_id: int):
def apply_reaction_filters(by, q): def apply_reaction_filters(by, q):
"""
Применяет фильтры к запросу реакций.
:param by: Словарь с параметрами фильтрации.
:param q: SQL-запрос.
:return: Запрос с примененными фильтрами.
"""
shout_slug = by.get("shout", None) shout_slug = by.get("shout", None)
if shout_slug: if shout_slug:
q = q.filter(Shout.slug == shout_slug) q = q.filter(Shout.slug == shout_slug)
@ -363,24 +443,25 @@ def apply_reaction_filters(by, q):
@query.field("load_reactions_by") @query.field("load_reactions_by")
async def load_reactions_by(_, info, by, limit=50, offset=0): async def load_reactions_by(_, info, by, limit=50, offset=0):
""" """
:param info: graphql meta Загружает реакции по указанным параметрам.
:param by: {
:shout - filter by slug
:shouts - filer by shout slug list
:created_by - to filter by author ID
:author - to filter by author slug
:topic - to filter by topic
:search - to search by reactions' body
:comment - true if body.length > 0
:rating - true if kind is rating related
:after - amount of time ago
:sort - a fieldname to sort desc by default
}
:param limit: int amount of reactions
:param offset: int offset in this order
:return: Reaction[]
"""
:param info: Информация о контексте GraphQL.
:param by: {
:shout - фильтрация по slug публикации
:shouts - фильтрация по списку slug публикаций
:created_by - фильтрация по идентификатору автора
:author - фильтрация по slug автора
:topic - фильтрация по теме
:search - поиск по тексту реакций
:comment - фильтрация комментариев
:rating - фильтрация реакций с рейтингом
:after - фильтрация по времени создания
:sort - поле для сортировки (по убыванию по умолчанию)
}
:param limit: Количество реакций для загрузки.
:param offset: Смещение для пагинации.
:return: Список реакций.
"""
q = ( q = (
select(Reaction, Author, Shout) select(Reaction, Author, Shout)
.select_from(Reaction) .select_from(Reaction)
@ -388,27 +469,27 @@ async def load_reactions_by(_, info, by, limit=50, offset=0):
.join(Shout, Reaction.shout == Shout.id) .join(Shout, Reaction.shout == Shout.id)
) )
# calculate counters # Подсчет статистики
aliased_reaction = aliased(Reaction) aliased_reaction = aliased(Reaction)
q = add_reaction_stat_columns(q, aliased_reaction) q = add_reaction_stat_columns(q, aliased_reaction)
# filter # Применение фильтров
q = apply_reaction_filters(by, q) q = apply_reaction_filters(by, q)
q = q.where(Reaction.deleted_at.is_(None)) q = q.where(Reaction.deleted_at.is_(None))
# group by # Группировка
q = q.group_by(Reaction.id, Author.id, Shout.id, aliased_reaction.id) q = q.group_by(Reaction.id, Author.id, Shout.id, aliased_reaction.id)
# order by # Сортировка
order_stat = by.get("sort", "").lower() # 'like' | 'dislike' | 'newest' | 'oldest' order_stat = by.get("sort", "").lower() # 'like' | 'dislike' | 'newest' | 'oldest'
order_by_stmt = desc(Reaction.created_at) order_by_stmt = desc(Reaction.created_at)
if order_stat == "oldest": if order_stat == "oldest":
order_by_stmt = asc(Reaction.created_at) order_by_stmt = asc(Reaction.created_at)
elif order_stat.endswith("like"): elif order_stat.endswith("like"):
order_by_stmt = desc(f"{order_stat}s_stat") order_by_stmt = desc("rating_stat")
q = q.order_by(order_by_stmt) q = q.order_by(order_by_stmt)
# pagination # Пагинация
q = q.limit(limit).offset(offset) q = q.limit(limit).offset(offset)
reactions = set() reactions = set()
@ -424,109 +505,23 @@ async def load_reactions_by(_, info, by, limit=50, offset=0):
] in result_rows: ] in result_rows:
reaction.created_by = author reaction.created_by = author
reaction.shout = shout reaction.shout = shout
reaction.stat = { reaction.stat = {"rating": rating_stat, "commented": commented_stat}
"rating": rating_stat,
"commented": commented_stat,
"last_reacted_at": last_reacted_at,
}
reactions.add(reaction) reactions.add(reaction)
return reactions return reactions
async def reacted_shouts_updates(follower_id: int, limit=50, offset=0) -> List[Shout]:
shouts: List[Shout] = []
with local_session() as session:
author = session.query(Author).filter(Author.id == follower_id).first()
if author:
# Shouts where follower is the author
q1 = (
select(Shout)
.outerjoin(
Reaction,
and_(Reaction.shout == Shout.id, Reaction.created_by == follower_id, Reaction.deleted_at.is_(None)),
)
.outerjoin(Author, Shout.authors.any(id=follower_id))
.options(joinedload(Shout.reactions), joinedload(Shout.authors))
)
q1 = add_reaction_stat_columns(q1, aliased(Reaction))
q1 = q1.filter(Author.id == follower_id).group_by(Shout.id)
# Shouts where follower reacted
q2 = (
select(Shout)
.join(Reaction, Reaction.shout == Shout.id)
.options(joinedload(Shout.reactions), joinedload(Shout.authors))
.filter(and_(Reaction.created_by == follower_id, Reaction.deleted_at.is_(None)))
.group_by(Shout.id)
)
q2 = add_reaction_stat_columns(q2, aliased(Reaction))
# Sort shouts by the `last_reacted_at` field
combined_query = union(q1, q2).order_by(desc(text("last_reacted_at"))).limit(limit).offset(offset)
results = session.execute(combined_query).scalars()
for [
shout,
reacted_stat,
commented_stat,
likes_stat,
dislikes_stat,
last_reacted_at,
] in results:
shout.stat = {
"viewed": await ViewedStorage.get_shout(shout.slug),
"rating": int(likes_stat or 0) - int(dislikes_stat or 0),
"reacted": reacted_stat,
"commented": commented_stat,
"last_reacted_at": last_reacted_at,
}
shouts.append(shout)
return shouts
@query.field("load_shouts_followed")
@login_required
async def load_shouts_followed(_, info, limit=50, offset=0) -> List[Shout]:
user_id = info.context["user_id"]
with local_session() as session:
author = session.query(Author).filter(Author.user == user_id).first()
if author:
try:
author_id: int = author.dict()["id"]
shouts = await reacted_shouts_updates(author_id, limit, offset)
return shouts
except Exception as error:
logger.debug(error)
return []
@query.field("load_shouts_followed_by")
async def load_shouts_followed_by(_, info, slug: str, limit=50, offset=0) -> List[Shout]:
with local_session() as session:
author = session.query(Author).filter(Author.slug == slug).first()
if author:
try:
author_id: int = author.dict()["id"]
shouts = await reacted_shouts_updates(author_id, limit, offset)
return shouts
except Exception as error:
logger.debug(error)
return []
@query.field("load_shout_ratings") @query.field("load_shout_ratings")
async def load_shout_ratings(_, info, shout: int, limit=100, offset=0): async def load_shout_ratings(_, info, shout: int, limit=100, offset=0):
""" """
get paginated reactions with no stats Получает оценки для указанной публикации с пагинацией.
:param info: graphql meta
:param shout: int shout id
:param limit: int amount of reactions
:param offset: int offset in this order
:return: Reaction[]
"""
:param info: Информация о контексте GraphQL.
:param shout: Идентификатор публикации.
:param limit: Количество реакций для загрузки.
:param offset: Смещение для пагинации.
:return: Список реакций.
"""
q = ( q = (
select(Reaction, Author, Shout) select(Reaction, Author, Shout)
.select_from(Reaction) .select_from(Reaction)
@ -534,7 +529,7 @@ async def load_shout_ratings(_, info, shout: int, limit=100, offset=0):
.join(Shout, Reaction.shout == Shout.id) .join(Shout, Reaction.shout == Shout.id)
) )
# filter, group, order, limit, offset # Фильтрация, группировка, сортировка, лимит, офсет
q = q.filter(and_(Reaction.deleted_at.is_(None), Reaction.shout == shout, Reaction.kind.in_(RATING_REACTIONS))) q = q.filter(and_(Reaction.deleted_at.is_(None), Reaction.shout == shout, Reaction.kind.in_(RATING_REACTIONS)))
q = q.group_by(Reaction.id) q = q.group_by(Reaction.id)
q = q.order_by(desc(Reaction.created_at)) q = q.order_by(desc(Reaction.created_at))
@ -558,12 +553,13 @@ async def load_shout_ratings(_, info, shout: int, limit=100, offset=0):
@query.field("load_shout_comments") @query.field("load_shout_comments")
async def load_shout_comments(_, info, shout: int, limit=50, offset=0): async def load_shout_comments(_, info, shout: int, limit=50, offset=0):
""" """
getting paginated comments with stats Получает комментарии для указанной публикации с пагинацией и статистикой.
:param info: graphql meta
:param shout: int shout id :param info: Информация о контексте GraphQL.
:param limit: int amount of reactions :param shout: Идентификатор публикации.
:param offset: int offset in this order :param limit: Количество комментариев для загрузки.
:return: Reaction[] :param offset: Смещение для пагинации.
:return: Список реакций.
""" """
aliased_reaction = aliased(Reaction) aliased_reaction = aliased(Reaction)
q = ( q = (
@ -583,7 +579,7 @@ async def load_shout_comments(_, info, shout: int, limit=50, offset=0):
.join(Shout, Reaction.shout == Shout.id) .join(Shout, Reaction.shout == Shout.id)
) )
# filter, group, order, limit, offset # Фильтрация, группировка, сортировка, лимит, офсет
q = q.filter(and_(Reaction.deleted_at.is_(None), Reaction.shout == shout, Reaction.body.is_not(None))) q = q.filter(and_(Reaction.deleted_at.is_(None), Reaction.shout == shout, Reaction.body.is_not(None)))
q = q.group_by(Reaction.id, Author.id, Shout.id) q = q.group_by(Reaction.id, Author.id, Shout.id)
q = q.order_by(desc(Reaction.created_at)) q = q.order_by(desc(Reaction.created_at))
@ -609,12 +605,13 @@ async def load_shout_comments(_, info, shout: int, limit=50, offset=0):
@query.field("load_comment_ratings") @query.field("load_comment_ratings")
async def load_comment_ratings(_, info, comment: int, limit=50, offset=0): async def load_comment_ratings(_, info, comment: int, limit=50, offset=0):
""" """
getting paginated comments with stats Получает оценки для указанного комментария с пагинацией и статистикой.
:param info: graphql meta
:param comment: int replied comment id :param info: Информация о контексте GraphQL.
:param limit: int amount of reactions :param comment: Идентификатор комментария.
:param offset: int offset in this order :param limit: Количество оценок для загрузки.
:return: Reaction[] :param offset: Смещение для пагинации.
:return: Список реакций.
""" """
aliased_reaction = aliased(Reaction) aliased_reaction = aliased(Reaction)
q = ( q = (
@ -634,7 +631,7 @@ async def load_comment_ratings(_, info, comment: int, limit=50, offset=0):
.join(Shout, Reaction.shout == Shout.id) .join(Shout, Reaction.shout == Shout.id)
) )
# filter, group, order, limit, offset # Фильтрация, группировка, сортировка, лимит, офсет
q = q.filter(and_(Reaction.deleted_at.is_(None), Reaction.reply_to == comment, Reaction.body.is_not(None))) q = q.filter(and_(Reaction.deleted_at.is_(None), Reaction.reply_to == comment, Reaction.body.is_not(None)))
q = q.group_by(Reaction.id, Author.id, Shout.id) q = q.group_by(Reaction.id, Author.id, Shout.id)
q = q.order_by(desc(Reaction.created_at)) q = q.order_by(desc(Reaction.created_at))

View File

@ -1,4 +1,6 @@
from sqlalchemy.orm import aliased, selectinload from typing import List
from sqlalchemy.orm import aliased, selectinload, joinedload
from sqlalchemy.sql import union
from sqlalchemy.sql.expression import ( from sqlalchemy.sql.expression import (
and_, and_,
asc, asc,
@ -504,3 +506,81 @@ async def load_shouts_discussed(_, info, limit=50, offset=0):
q, aliased_reaction = query_shouts() q, aliased_reaction = query_shouts()
q = q.filter(Shout.id.in_(reaction_subquery)) q = q.filter(Shout.id.in_(reaction_subquery))
return get_shouts_with_stats(q, limit, offset=offset) return get_shouts_with_stats(q, limit, offset=offset)
async def reacted_shouts_updates(follower_id: int, limit=50, offset=0) -> List[Shout]:
"""
Обновляет публикации, на которые подписан автор, с учетом реакций.
:param follower_id: Идентификатор подписчика.
:param limit: Количество публикаций для загрузки.
:param offset: Смещение для пагинации.
:return: Список публикаций.
"""
shouts: List[Shout] = []
with local_session() as session:
author = session.query(Author).filter(Author.id == follower_id).first()
if author:
# Публикации, где подписчик является автором
q1, aliased_reaction1 = query_shouts()
q1 = q1.filter(Shout.authors.any(id=follower_id))
# Публикации, на которые подписчик реагировал
q2, aliased_reaction2 = query_shouts()
q2 = q2.options(joinedload(Shout.reactions))
q2 = q2.filter(Reaction.created_by == follower_id)
# Сортировка публикаций по полю `last_reacted_at`
combined_query = union(q1, q2).order_by(desc(text("last_reacted_at")))
# извлечение ожидаемой структуры данных
shouts = get_shouts_with_stats(combined_query, limit, offset=offset)
return shouts
@query.field("load_shouts_followed")
@login_required
async def load_shouts_followed(_, info, limit=50, offset=0) -> List[Shout]:
"""
Загружает публикации, на которые подписан пользователь.
:param info: Информация о контексте GraphQL.
:param limit: Количество публикаций для загрузки.
:param offset: Смещение для пагинации.
:return: Список публикаций.
"""
user_id = info.context["user_id"]
with local_session() as session:
author = session.query(Author).filter(Author.user == user_id).first()
if author:
try:
author_id: int = author.dict()["id"]
shouts = await reacted_shouts_updates(author_id, limit, offset)
return shouts
except Exception as error:
logger.debug(error)
return []
@query.field("load_shouts_followed_by")
async def load_shouts_followed_by(_, info, slug: str, limit=50, offset=0) -> List[Shout]:
"""
Загружает публикации, на которые подписан автор по slug.
:param info: Информация о контексте GraphQL.
:param slug: Slug автора.
:param limit: Количество публикаций для загрузки.
:param offset: Смещение для пагинации.
:return: Список публикаций.
"""
with local_session() as session:
author = session.query(Author).filter(Author.slug == slug).first()
if author:
try:
author_id: int = author.dict()["id"]
shouts = await reacted_shouts_updates(author_id, limit, offset)
return shouts
except Exception as error:
logger.debug(error)
return []

View File

@ -38,7 +38,9 @@ def add_topic_stat_columns(q):
Shout.deleted_at.is_(None), Shout.deleted_at.is_(None),
), ),
) )
.add_columns(func.count(distinct(aliased_shout.shout)).label("shouts_stat")) # Подсчет уникальных публикаций для темы .add_columns(
func.count(distinct(aliased_shout.shout)).label("shouts_stat")
) # Подсчет уникальных публикаций для темы
) )
aliased_follower = aliased(TopicFollower) aliased_follower = aliased(TopicFollower)
@ -80,7 +82,9 @@ def add_author_stat_columns(q):
aliased_shout.deleted_at.is_(None), aliased_shout.deleted_at.is_(None),
), ),
) )
.add_columns(func.count(distinct(aliased_shout.id)).label("shouts_stat")) # Подсчет уникальных публикаций автора .add_columns(
func.count(distinct(aliased_shout.id)).label("shouts_stat")
) # Подсчет уникальных публикаций автора
) )
# Добавляем количество подписчиков автора # Добавляем количество подписчиков автора