From 33ddfc6c31f38db8663bd07dcb3e877fbc4899ec Mon Sep 17 00:00:00 2001
From: Untone <anton.rewin@gmail.com>
Date: Mon, 3 Feb 2025 23:22:45 +0300
Subject: [PATCH] after-handlers-cache

---
 CHANGELOG.md      |  1 +
 cache/triggers.py | 47 ++++++++++++++++++++++++++++++++---------------
 2 files changed, 33 insertions(+), 15 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 8d20ca18..48bd1800 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,6 +1,7 @@
 #### [0.4.8]
 - `Reaction.deleted_at` filter on `update_reaction` resolver added
 - `triggers` module updated with `after_shout_handler`, `after_reaction_handler` for cache revalidation
+- `after_shout_handler`, `after_reaction_handler` now also handle `deleted_at` field
 
 #### [0.4.7]
 - `get_my_rates_shouts` resolver added with:
diff --git a/cache/triggers.py b/cache/triggers.py
index e92bb12e..237c69e4 100644
--- a/cache/triggers.py
+++ b/cache/triggers.py
@@ -51,29 +51,44 @@ def after_shout_handler(mapper, connection, target):
     # Проверяем изменение статуса публикации
     was_published = target.published_at is not None and target.deleted_at is None
     
-    if was_published:
-        # Обновляем счетчики для авторов
-        for author in target.authors:
-            revalidation_manager.mark_for_revalidation(author.id, "authors")
-            
-        # Обновляем счетчики для тем
-        for topic in target.topics:
-            revalidation_manager.mark_for_revalidation(topic.id, "topics")
+    # Всегда обновляем счетчики для авторов и тем при любом изменении поста
+    for author in target.authors:
+        revalidation_manager.mark_for_revalidation(author.id, "authors")
+        
+    for topic in target.topics:
+        revalidation_manager.mark_for_revalidation(topic.id, "topics")
+        
+    # Обновляем сам пост
+    revalidation_manager.mark_for_revalidation(target.id, "shouts")
 
 
 def after_reaction_handler(mapper, connection, target):
     """Обработчик для комментариев"""
-    if not isinstance(target, Reaction) or target.kind != ReactionKind.COMMENT.value:
+    if not isinstance(target, Reaction):
         return
         
-    # Проверяем что комментарий относится к опубликованному посту
+    # Проверяем что это комментарий
+    is_comment = target.kind == ReactionKind.COMMENT.value
+    
+    # Получаем связанный пост
     shout = target.shout
-    if shout and shout.published_at and not shout.deleted_at:
-        # Обновляем счетчики для автора комментария
-        revalidation_manager.mark_for_revalidation(target.created_by.id, "authors")
+    if not shout:
+        return
         
-        # Обновляем счетчики для поста
-        revalidation_manager.mark_for_revalidation(shout.id, "shouts")
+    # Обновляем счетчики для автора комментария
+    if target.created_by:
+        revalidation_manager.mark_for_revalidation(target.created_by.id, "authors")
+    
+    # Обновляем счетчики для поста и его авторов/тем
+    revalidation_manager.mark_for_revalidation(shout.id, "shouts")
+    
+    if is_comment and shout.published_at and not shout.deleted_at:
+        # Для комментариев к опубликованным постам обновляем также:
+        for author in shout.authors:
+            revalidation_manager.mark_for_revalidation(author.id, "authors")
+            
+        for topic in shout.topics:
+            revalidation_manager.mark_for_revalidation(topic.id, "topics")
 
 
 def events_register():
@@ -98,8 +113,10 @@ def events_register():
     event.listen(Author, "after_update", mark_for_revalidation)
     event.listen(Topic, "after_update", mark_for_revalidation)
     event.listen(Shout, "after_update", after_shout_handler)
+    event.listen(Shout, "after_delete", after_shout_handler)
 
     event.listen(Reaction, "after_insert", after_reaction_handler)
+    event.listen(Reaction, "after_update", after_reaction_handler)
     event.listen(Reaction, "after_delete", after_reaction_handler)
 
     logger.info("Event handlers registered successfully.")