96 lines
4.8 KiB
Python
96 lines
4.8 KiB
Python
import logging
|
|
import aiosqlite
|
|
from typing import Optional ,Dict ,Callable ,Any ,TYPE_CHECKING ,Tuple
|
|
from aiogram import Bot
|
|
from datetime import datetime
|
|
|
|
from config .settings import Settings
|
|
from db .database import get_user
|
|
from bot .middlewares .i18n import JsonI18n
|
|
|
|
if TYPE_CHECKING :
|
|
from .subscription_service import SubscriptionService
|
|
|
|
|
|
class ReferralService :
|
|
def __init__ (self ,db_conn_provider :Callable [[],Any ],settings :Settings ,subscription_service_instance :'SubscriptionService',bot :Bot ,i18n :JsonI18n ):
|
|
self .db_conn_provider =db_conn_provider
|
|
self .settings =settings
|
|
self .subscription_service =subscription_service_instance
|
|
self .bot =bot
|
|
self .i18n =i18n
|
|
|
|
|
|
|
|
|
|
async def process_new_user_referral (self ,referee_user_id :int ,inviter_user_id :Optional [int ]):
|
|
if inviter_user_id and referee_user_id !=inviter_user_id :
|
|
|
|
|
|
logging .info (f"Referral link used: User {referee_user_id} was invited by {inviter_user_id}.")
|
|
|
|
pass
|
|
|
|
async def apply_referral_bonuses_for_payment (
|
|
self ,referee_user_id :int ,purchased_subscription_months :int ,
|
|
db_conn :aiosqlite .Connection
|
|
)->Dict [str ,Any ]:
|
|
"""Applies bonuses. Notifies inviter. Returns referee bonus details."""
|
|
referee_final_end_date :Optional [datetime ]=None
|
|
referee_bonus_applied_days :Optional [int ]=None
|
|
try :
|
|
referee_user_row =await get_user (referee_user_id ,db_conn =db_conn )
|
|
if not referee_user_row or referee_user_row ['referred_by_id']is None :
|
|
logging .debug (f"User {referee_user_id} not referred or inviter ID missing. No referral bonuses.")
|
|
return {"referee_bonus_applied_days":None ,"referee_new_end_date":None }
|
|
|
|
inviter_user_id =referee_user_row ['referred_by_id']
|
|
inviter_user_row =await get_user (inviter_user_id ,db_conn =db_conn )
|
|
|
|
referee_name =referee_user_row ['first_name']or f"User {referee_user_id}"
|
|
|
|
default_lang_for_placeholder =getattr (self .settings ,'DEFAULT_LANGUAGE','en')
|
|
inviter_name =inviter_user_row ['first_name']if inviter_user_row else self .i18n .gettext (default_lang_for_placeholder ,"friend_placeholder")
|
|
|
|
inviter_bonus =self .settings .referral_bonus_inviter .get (purchased_subscription_months )
|
|
referee_bonus =self .settings .referral_bonus_referee .get (purchased_subscription_months )
|
|
|
|
if inviter_bonus and inviter_bonus >0 and inviter_user_row :
|
|
new_end_date_inviter =await self .subscription_service .extend_subscription_for_referral (
|
|
user_id =inviter_user_id ,bonus_days =inviter_bonus ,db_conn =db_conn
|
|
)
|
|
if new_end_date_inviter :
|
|
logging .info (f"Bonus applied for inviter {inviter_user_id}.")
|
|
try :
|
|
inviter_lang =inviter_user_row .get ('language_code',default_lang_for_placeholder )
|
|
_i =lambda k ,**kw :self .i18n .gettext (inviter_lang ,k ,**kw )
|
|
await self .bot .send_message (
|
|
inviter_user_id ,
|
|
_i ("referral_bonus_inviter_notification_extended",
|
|
days =inviter_bonus ,
|
|
referee_name =referee_name ,
|
|
new_end_date =new_end_date_inviter .strftime ('%Y-%m-%d')
|
|
)
|
|
)
|
|
except Exception as e :
|
|
logging .error (f"Failed to send bonus notification to inviter {inviter_user_id}: {e}")
|
|
else :logging .warning (f"Failed to apply bonus subscription extension for inviter {inviter_user_id}.")
|
|
|
|
if referee_bonus and referee_bonus >0 :
|
|
new_end_date_referee =await self .subscription_service .extend_subscription_for_referral (
|
|
user_id =referee_user_id ,bonus_days =referee_bonus ,db_conn =db_conn ,is_referee_bonus =True
|
|
)
|
|
if new_end_date_referee :
|
|
logging .info (f"Bonus applied for referee {referee_user_id}.")
|
|
referee_final_end_date =new_end_date_referee
|
|
referee_bonus_applied_days =referee_bonus
|
|
else :logging .warning (f"Failed to apply bonus subscription extension for referee {referee_user_id}.")
|
|
|
|
return {"referee_bonus_applied_days":referee_bonus_applied_days ,"referee_new_end_date":referee_final_end_date }
|
|
except Exception as e :
|
|
logging .error (f"Error in apply_referral_bonuses_for_payment (db_conn: {db_conn is not None}): {e}",exc_info =True )
|
|
|
|
raise
|
|
|
|
def generate_referral_link (self ,bot_username :str ,inviter_user_id :int )->str :
|
|
return f"https://t.me/{bot_username}?start=ref_{inviter_user_id}" |