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

tolerate-double-follow
All checks were successful
Deploy on push / deploy (push) Successful in 6s
Details

Browse Source
This commit is contained in:
Untone 2024-11-02 12:33:52 +03:00
parent 5f3f00366f
commit 1b43f742d3
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
resolvers/follower.py
View File

@ -70,10 +70,11 @@ async def follow(_, info, what, slug):
if entity_id: if entity_id:
logger.debug("Проверка существующей подписки") logger.debug("Проверка существующей подписки")
with local_session() as session: with local_session() as session:
existing_sub = session.query(follower_class).filter( existing_sub = (
follower_class.follower == follower_id, session.query(follower_class)
getattr(follower_class, entity_type) == entity_id .filter(follower_class.follower == follower_id, getattr(follower_class, entity_type) == entity_id)
).first() .first()
)
if existing_sub: if existing_sub:
logger.info(f"Пользователь {follower_id} уже подписан на {what.lower()} с ID {entity_id}") logger.info(f"Пользователь {follower_id} уже подписан на {what.lower()} с ID {entity_id}")
else: else:
@ -172,7 +173,7 @@ async def unfollow(_, info, what, slug):
if get_cached_follows_method: if get_cached_follows_method:
logger.debug("Получение подписок из кэша") logger.debug("Получение подписок из кэша")
existing_follows = await get_cached_follows_method(follower_id) existing_follows = await get_cached_follows_method(follower_id)
follows = filter(lambda x: x['id'] != entity_id, existing_follows) follows = filter(lambda x: x["id"] != entity_id, existing_follows)
logger.debug("Обновлен список подписок") logger.debug("Обновлен список подписок")
if what == "AUTHOR": if what == "AUTHOR":