Adding admin operations to the balance

database/hot_leads.py

@@ -2,6 +2,7 @@ from sqlalchemy import func, select
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from database.models import Key, Payment, User
+from core.constants import PAYMENT_SYSTEMS_EXCLUDED
 
 
 async def get_hot_leads(session: AsyncSession):
@@ -16,7 +17,7 @@ async def get_hot_leads(session: AsyncSession):
         .where(User.trial == 1)
         .where(Payment.amount > 0)
         .where(Payment.status == "success")
-        .where(Payment.payment_system.notin_(["referral", "coupon", "cashback"]))
+        .where(Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED))
         .where(~Payment.tg_id.in_(sub_active))
     )
 

database/statistics.py

@@ -4,6 +4,7 @@ from sqlalchemy import and_, exists, func, not_, select
 from sqlalchemy.ext.asyncio import AsyncSession
 
 from database.models import Key, Payment, Referral, Tariff, User
+from core.constants import PAYMENT_SYSTEMS_EXCLUDED
 
 
 async def count_total_users(session: AsyncSession) -> int:
@@ -116,7 +117,7 @@ async def sum_payments_since(session: AsyncSession, since: date) -> float:
             and_(
                 Payment.created_at >= since,
                 Payment.status == "success",
-                Payment.payment_system.notin_(["referral", "coupon", "cashback"]),
+                Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED),
             )
         )
     )
@@ -130,7 +131,7 @@ async def sum_payments_between(session: AsyncSession, start: date, end: date) ->
                 Payment.created_at >= start,
                 Payment.created_at < end,
                 Payment.status == "success",
-                Payment.payment_system.notin_(["referral", "coupon", "cashback"]),
+                Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED),
             )
         )
     )
@@ -142,7 +143,7 @@ async def sum_total_payments(session: AsyncSession) -> float:
         select(func.coalesce(func.sum(Payment.amount), 0)).where(
             and_(
                 Payment.status == "success",
-                Payment.payment_system.notin_(["referral", "coupon", "cashback"]),
+                Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED),
             )
         )
     )
@@ -158,7 +159,7 @@ async def count_hot_leads(session: AsyncSession) -> int:
         select(Payment.tg_id)
         .where(Payment.amount > 0)
         .where(Payment.status == "success")
-        .where(Payment.payment_system.notin_(["referral", "coupon", "cashback"]))
+        .where(Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED))
         .where(not_(exists(subquery_active_keys.where(Key.tg_id == Payment.tg_id))))
         .distinct()
     )

database/tracking_sources.py

@@ -4,9 +4,7 @@ from sqlalchemy.ext.asyncio import AsyncSession
 
 from database.models import Payment, TrackingSource, User
 from logger import logger
-
-
-EXCLUDED_PAYMENT_MARKERS = ["coupon", "referral", "cashback"]
+from core.constants import PAYMENT_SYSTEMS_EXCLUDED
 
 
 async def create_tracking_source(session: AsyncSession, name: str, code: str, type_: str, created_by: int):
@@ -43,7 +41,11 @@ async def get_all_tracking_sources(session: AsyncSession) -> list[dict]:
     payments_subq = (
         select(func.count(func.distinct(Payment.tg_id)))
         .join(User, Payment.tg_id == User.tg_id)
-        .where((User.source_code == TrackingSource.code) & (Payment.status == "success"))
+        .where(
+            (User.source_code == TrackingSource.code)
+            & (Payment.status == "success")
+            & Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED)
+        )
         .correlate(TrackingSource)
         .scalar_subquery()
     )
@@ -103,7 +105,7 @@ async def get_tracking_source_stats(session: AsyncSession, code: str) -> dict |
         .where(
             (User.source_code == code)
             & (Payment.status == "success")
-            & not_(Payment.payment_system.in_(EXCLUDED_PAYMENT_MARKERS))
+            & Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED)
             & (Payment.created_at >= created_at)
         )
         .scalar_subquery()
@@ -115,7 +117,7 @@ async def get_tracking_source_stats(session: AsyncSession, code: str) -> dict |
         .where(
             (User.source_code == code)
             & (Payment.status == "success")
-            & not_(Payment.payment_system.in_(EXCLUDED_PAYMENT_MARKERS))
+            & Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED)
             & (Payment.created_at >= created_at)
         )
         .scalar_subquery()
@@ -146,7 +148,7 @@ async def get_tracking_source_stats(session: AsyncSession, code: str) -> dict |
         .where(
             (User.source_code == code)
             & (Payment.status == "success")
-            & not_(Payment.payment_system.in_(EXCLUDED_PAYMENT_MARKERS))
+            & Payment.payment_system.notin_(PAYMENT_SYSTEMS_EXCLUDED)
             & (Payment.created_at >= created_at)
         )
         .subquery()