From c2795c7b44ad27ecce7466d8fda79d5cc290c53d Mon Sep 17 00:00:00 2001
From: Itamar Shefi <itamarshefi@gmail.com>
Date: Sun, 4 Feb 2024 16:22:39 +0200
Subject: [PATCH] Fix subscription record logic

---
 logic/user_logic.py | 22 +++++++++++-----------
 1 file changed, 11 insertions(+), 11 deletions(-)

diff --git a/logic/user_logic.py b/logic/user_logic.py
index 4da7302..900d111 100644
--- a/logic/user_logic.py
+++ b/logic/user_logic.py
@@ -80,18 +80,18 @@ async def subscribe(self, subscription: schemas.Subscription) -> bool:
             query = select(tables.UserSubscription)
             query = query.where(tables.UserSubscription.uuid == subscription.message_id)
             if session.exec(query).one_or_none() is None:
-                return False
-
-            session.add(
-                tables.UserSubscription(
-                    user_id=user.id,
-                    amount=subscription.amount,
-                    tier_name=subscription.tier_name,
-                    uuid=subscription.message_id,
-                    timestamp=subscription.timestamp,
+                session.add(
+                    tables.UserSubscription(
+                        user_id=user.id,
+                        amount=subscription.amount,
+                        tier_name=subscription.tier_name,
+                        uuid=subscription.message_id,
+                        timestamp=subscription.timestamp,
+                    )
                 )
-            )
-            return True
+                return True
+            else:
+                return False
 
     def get_subscription_expiry(self, user: tables.User) -> datetime.datetime | None:
         with hs_transaction(self.session, expire_on_commit=False) as session: