feat: handle unsubscribe

src/usecase/subscription/handle_subscription.py

@@ -17,20 +17,12 @@ async def handle_subscription(
     first_name: str | None = None,
     last_name: str | None = None,
 ) -> None:
-    """
-    Обрабатывает подписку пользователя в целевой канал.
-
-    Находит закуп по invite_link и создаёт запись о подписке,
-    инкрементируя счётчик подписок у закупа.
-    """
     async with self.database.transaction():
-        # Находим закуп по пригласительной ссылке
         placement = await self.database.get_placement_by_invite_link(invite_link)
         if not placement:
             log.warning('Placement not found for invite_link: %s', invite_link)
             return
 
-        # Создаём или обновляем подписчика
         subscriber = domain.Subscriber(
             telegram_id=user_telegram_id,
             username=username,
@@ -39,27 +31,53 @@ async def handle_subscription(
         )
         subscriber = await self.database.upsert_subscriber(subscriber)
 
-        # Проверяем, не создана ли уже подписка для этого пользователя
-        existing_subscription = await self.database.get_subscription_by_subscriber_and_placement(
-            subscriber.id, placement.id
+        active_subscription = await self.database.get_active_subscription_by_subscriber_and_channel(
+            subscriber.id, placement.target_channel.telegram_id
         )
-        if existing_subscription:
-            log.info(
-                'Subscription already exists for subscriber %s and placement %s',
+
+        if active_subscription:
+            # Пользователь уже подписан на канал через другой placement
+            # Это не должно случиться (Telegram не даст подписаться дважды),
+            # но если случилось - логируем и игнорируем
+            log.warning(
+                'User %s (telegram_id: %s) already has active subscription to channel %s via placement %s, '
+                'ignoring new subscription attempt via placement %s',
                 subscriber.id,
+                user_telegram_id,
+                placement.target_channel_id,
+                active_subscription.placement_id,
+                placement.id,
+            )
+            return
+
+        # Проверяем, была ли раньше подписка через ЭТОТ placement (для реактивации)
+        existing_sub = await self.database.get_subscription_by_subscriber_and_placement(subscriber.id, placement.id)
+
+        if existing_sub and existing_sub.status == domain.SubscriptionStatus.UNSUBSCRIBED:
+            # Реактивируем старую подписку через тот же placement
+            existing_sub.status = domain.SubscriptionStatus.ACTIVE
+            existing_sub.unsubscribed_at = None
+            await self.database.update_subscription(existing_sub)
+
+            placement.subscriptions_count += 1
+            await self.database.update_placement(placement)
+
+            log.info(
+                'Subscription reactivated: subscriber %s (telegram_id: %s) resubscribed via same placement %s',
+                subscriber.id,
+                user_telegram_id,
                 placement.id,
             )
             return
 
-        # Создаём запись о подписке
         subscription = domain.Subscription(
             placement_id=placement.id,
             subscriber_id=subscriber.id,
+            target_channel_id=placement.target_channel_id,
             invite_link=invite_link,
         )
         await self.database.create_subscription(subscription)
 
-        # Увеличиваем счётчик подписок у закупа
         placement.subscriptions_count += 1
         await self.database.update_placement(placement)
 

src/usecase/subscription/handle_unsubscription.py

@@ -0,0 +1,49 @@
+import datetime
+import logging
+from typing import TYPE_CHECKING
+
+from src import domain
+
+if TYPE_CHECKING:
+    from .. import Usecase
+
+log = logging.getLogger(__name__)
+
+
+async def handle_unsubscription(self: 'Usecase', user_telegram_id: int, channel_telegram_id: int) -> None:
+    async with self.database.transaction():
+        subscriber = await self.database.get_subscriber(user_telegram_id)
+        if not subscriber:
+            log.warning('Subscriber not found for telegram_id: %s', user_telegram_id)
+            return
+
+        subscriptions = await self.database.get_active_subscriptions_by_subscriber_and_channel(
+            subscriber.id, channel_telegram_id
+        )
+
+        if not subscriptions:
+            log.info(
+                'No active subscriptions found for subscriber %s (telegram_id: %s) in channel %s',
+                subscriber.id,
+                user_telegram_id,
+                channel_telegram_id,
+            )
+            return
+
+        for subscription in subscriptions:
+            subscription.status = domain.SubscriptionStatus.UNSUBSCRIBED
+            subscription.unsubscribed_at = datetime.datetime.now(datetime.UTC)
+            await self.database.update_subscription(subscription)
+
+            # Декрементируем счётчик подписок у закупа
+            placement = subscription.placement
+            if placement.subscriptions_count > 0:
+                placement.subscriptions_count -= 1
+                await self.database.update_placement(placement)
+
+            log.info(
+                'Subscription marked as unsubscribed: subscriber %s (telegram_id: %s) unsubscribed from placement %s',
+                subscriber.id,
+                user_telegram_id,
+                placement.id,
+            )