90% сеньоров плачат на таких коммитах

src/usecase/placement/update_placement.py

@@ -4,6 +4,8 @@ from typing import TYPE_CHECKING
 
 from src import domain, dto
 
+from .helpers import generate_post_url
+
 if TYPE_CHECKING:
     from .. import Usecase
 
@@ -17,9 +19,7 @@ async def update_placement(
     user_id: uuid.UUID,
     workspace_id: uuid.UUID,
 ) -> dto.PlacementOutput:
-    context = await self.ensure_workspace_permission(
-        workspace_id, user_id, domain.PermissionKey.PLACEMENTS_WRITE
-    )
+    context = await self.ensure_workspace_permission(workspace_id, user_id, domain.PermissionKey.PLACEMENTS_WRITE)
 
     placement = await self.database.get_placement(workspace_id, placement_id)
     if not placement:
@@ -39,6 +39,21 @@ async def update_placement(
 
     await self.database.update_placement(placement)
 
+    # Compute post-related fields
+    ad_post_url = None
+    views_count = None
+    views_availability = domain.PostViewsAvailability.UNKNOWN
+    last_views_fetch_at = None
+
+    if placement.post:
+        ad_post_url = generate_post_url(placement.post.channel.username, placement.post.message_id)
+
+        # Get latest views data
+        views_data = await self.database.get_latest_views_data(placement.post.id)
+        if views_data:
+            views_count, last_views_fetch_at = views_data
+            views_availability = domain.PostViewsAvailability.AVAILABLE
+
     return dto.PlacementOutput(
         id=placement.id,
         project_id=placement.project_id,
@@ -50,13 +65,13 @@ async def update_placement(
         placement_date=placement.placement_date,
         cost=placement.cost,
         comment=placement.comment,
-        ad_post_url=placement.ad_post_url,
+        ad_post_url=ad_post_url,
         invite_link_type=placement.invite_link_type,
         invite_link=placement.invite_link,
         status=placement.status,
         subscriptions_count=placement.subscriptions_count,
-        views_count=placement.views_count,
-        views_availability=placement.views_availability,
-        last_views_fetch_at=placement.last_views_fetch_at,
+        views_count=views_count,
+        views_availability=views_availability,
+        last_views_fetch_at=last_views_fetch_at,
         created_at=placement.created_at,
     )