130 lines
5.6 KiB
Python
130 lines
5.6 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}"
|