From d499574c4d8bb400b87fefd4d4d1270dcaa7103f Mon Sep 17 00:00:00 2001 From: Egor Date: Sat, 6 Sep 2025 01:27:27 +0300 Subject: [PATCH] Update referral.py --- app/database/crud/referral.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/app/database/crud/referral.py b/app/database/crud/referral.py index 237fdf8e..b00337ef 100644 --- a/app/database/crud/referral.py +++ b/app/database/crud/referral.py @@ -281,14 +281,17 @@ async def get_user_referral_stats(db: AsyncSession, user_id: int) -> dict: month_ago = datetime.utcnow() - timedelta(days=30) month_earned = await get_referral_earnings_sum(db, user_id, start_date=month_ago) - from app.database.models import Subscription + from app.database.models import Subscription, SubscriptionStatus + current_time = datetime.utcnow() + active_referrals_result = await db.execute( select(func.count(User.id)) .join(Subscription, User.id == Subscription.user_id) .where( and_( User.referred_by_id == user_id, - Subscription.is_active == True + Subscription.status == SubscriptionStatus.ACTIVE.value, + Subscription.end_date > current_time ) ) )