machka-pasla
aa97618165
- Replaced legacy payment notification functions with a new NotificationService for improved code organization and maintainability. - Updated user management handlers to utilize the new notification system for payment confirmations. - Enhanced user card formatting with localized labels for better clarity and user experience. - Introduced context management for the PanelApiService to ensure proper session handling. - Updated localization files to include new labels and messages for user management and notifications.
215 lines
8.9 KiB
Python
215 lines
8.9 KiB
Python
import logging
|
|
import hmac
|
|
import hashlib
|
|
import json
|
|
from typing import Optional
|
|
|
|
from aiohttp import web
|
|
from aiogram import Bot
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from config.settings import Settings
|
|
from bot.middlewares.i18n import JsonI18n
|
|
from bot.services.subscription_service import SubscriptionService
|
|
from bot.services.panel_api_service import PanelApiService
|
|
from bot.services.referral_service import ReferralService
|
|
from .notification_service import NotificationService
|
|
from bot.keyboards.inline.user_keyboards import get_connect_and_main_keyboard
|
|
from db.dal import payment_dal, user_dal, subscription_dal
|
|
|
|
|
|
def convert_period_to_months(period: Optional[str]) -> int:
|
|
"""Map Tribute subscription period strings to months."""
|
|
if not period:
|
|
return 1
|
|
|
|
mapping = {
|
|
"monthly": 1,
|
|
"quarterly": 3,
|
|
"3-month": 3,
|
|
"3months": 3,
|
|
"3-months": 3,
|
|
"q": 3,
|
|
"halfyearly": 6,
|
|
"yearly": 12,
|
|
"annual": 12,
|
|
"y": 12,
|
|
}
|
|
return mapping.get(period.lower(), 1)
|
|
|
|
|
|
class TributeService:
|
|
def __init__(self, bot: Bot, settings: Settings, i18n: JsonI18n,
|
|
async_session_factory: sessionmaker,
|
|
panel_service: PanelApiService,
|
|
subscription_service: SubscriptionService,
|
|
referral_service: ReferralService):
|
|
self.bot = bot
|
|
self.settings = settings
|
|
self.i18n = i18n
|
|
self.async_session_factory = async_session_factory
|
|
self.panel_service = panel_service
|
|
self.subscription_service = subscription_service
|
|
self.referral_service = referral_service
|
|
|
|
async def handle_webhook(self, raw_body: bytes,
|
|
signature_header: Optional[str]) -> web.Response:
|
|
settings = self.settings
|
|
bot = self.bot
|
|
i18n = self.i18n
|
|
async_session_factory = self.async_session_factory
|
|
subscription_service = self.subscription_service
|
|
referral_service = self.referral_service
|
|
|
|
if settings.TRIBUTE_API_KEY:
|
|
if not signature_header:
|
|
return web.Response(status=403, text="no_signature")
|
|
expected_sig = hmac.new(settings.TRIBUTE_API_KEY.encode(), raw_body,
|
|
hashlib.sha256).hexdigest()
|
|
if not hmac.compare_digest(expected_sig, signature_header):
|
|
return web.Response(status=403, text="invalid_signature")
|
|
|
|
try:
|
|
payload = json.loads(raw_body.decode())
|
|
except Exception:
|
|
return web.Response(status=400, text="bad_request")
|
|
|
|
logging.info(
|
|
"Tribute webhook data: %s",
|
|
json.dumps(payload, ensure_ascii=False),
|
|
)
|
|
|
|
event_name = payload.get('name')
|
|
data = payload.get('payload', {})
|
|
user_id = data.get('telegram_user_id')
|
|
price_val = (
|
|
data.get('amount')
|
|
or data.get('amount_paid')
|
|
or data.get('price')
|
|
)
|
|
|
|
if not user_id or price_val is None:
|
|
return web.Response(status=200, text="ok_missing_fields")
|
|
|
|
period_val = data.get('period')
|
|
months = convert_period_to_months(period_val)
|
|
price_rub = price_val / 100
|
|
|
|
async with async_session_factory() as session:
|
|
if event_name == 'new_subscription':
|
|
provider_payment_id = str(data.get('subscription_id'))
|
|
existing_payment = await payment_dal.get_payment_by_provider_payment_id(
|
|
session, provider_payment_id)
|
|
if existing_payment:
|
|
logging.info(
|
|
"Duplicate Tribute payment webhook ignored for provider_payment_id %s",
|
|
provider_payment_id,
|
|
)
|
|
payment_record = existing_payment
|
|
else:
|
|
payment_record = await payment_dal.create_payment_record(
|
|
session,
|
|
{
|
|
'user_id': user_id,
|
|
'amount': float(price_rub),
|
|
'currency': 'RUB',
|
|
'status': 'succeeded',
|
|
'description': 'Tribute subscription',
|
|
'subscription_duration_months': months,
|
|
'provider_payment_id': provider_payment_id,
|
|
'provider': 'tribute',
|
|
},
|
|
)
|
|
activation_details = await subscription_service.activate_subscription(
|
|
session,
|
|
user_id,
|
|
months,
|
|
float(price_rub),
|
|
payment_record.payment_id,
|
|
provider='tribute',
|
|
)
|
|
referral_bonus = await referral_service.apply_referral_bonuses_for_payment(
|
|
session, user_id, months)
|
|
await session.commit()
|
|
|
|
db_user = await user_dal.get_user_by_id(session, user_id)
|
|
lang = db_user.language_code if db_user and db_user.language_code else settings.DEFAULT_LANGUAGE
|
|
_ = lambda k, **kw: i18n.gettext(lang, k, **kw)
|
|
|
|
applied_ref_days = referral_bonus.get('referee_bonus_applied_days') if referral_bonus else None
|
|
final_end = (referral_bonus.get('referee_new_end_date')
|
|
if referral_bonus else None)
|
|
if not final_end:
|
|
final_end = activation_details.get('end_date')
|
|
|
|
if final_end:
|
|
config_link = activation_details.get("subscription_url") or _(
|
|
"config_link_not_available"
|
|
)
|
|
|
|
if applied_ref_days:
|
|
inviter_name_display = _('friend_placeholder')
|
|
if db_user and db_user.referred_by_id:
|
|
inviter = await user_dal.get_user_by_id(session, db_user.referred_by_id)
|
|
if inviter and inviter.first_name:
|
|
inviter_name_display = inviter.first_name
|
|
elif inviter and inviter.username:
|
|
inviter_name_display = f"@{inviter.username}"
|
|
success_msg = _(
|
|
"payment_successful_with_referral_bonus_full",
|
|
months=months,
|
|
base_end_date=activation_details["end_date"].strftime('%Y-%m-%d'),
|
|
bonus_days=applied_ref_days,
|
|
final_end_date=final_end.strftime('%Y-%m-%d'),
|
|
inviter_name=inviter_name_display,
|
|
config_link=config_link,
|
|
)
|
|
else:
|
|
success_msg = _(
|
|
"payment_successful_full",
|
|
months=months,
|
|
end_date=final_end.strftime('%Y-%m-%d'),
|
|
config_link=config_link,
|
|
)
|
|
markup = get_connect_and_main_keyboard(
|
|
lang, i18n, settings, config_link
|
|
)
|
|
|
|
try:
|
|
await bot.send_message(
|
|
user_id,
|
|
success_msg,
|
|
reply_markup=markup,
|
|
parse_mode="HTML",
|
|
disable_web_page_preview=True,
|
|
)
|
|
except Exception as e:
|
|
logging.error(
|
|
f"Failed to send Tribute payment success message to user {user_id}: {e}")
|
|
|
|
# Send notification about payment
|
|
try:
|
|
notification_service = NotificationService(bot, settings, i18n)
|
|
user = await user_dal.get_user_by_id(session, user_id)
|
|
await notification_service.notify_payment_received(
|
|
user_id=user_id,
|
|
amount=float(price_rub),
|
|
currency="RUB",
|
|
months=months,
|
|
payment_provider="tribute",
|
|
username=user.username if user else None
|
|
)
|
|
except Exception as e:
|
|
logging.error(f"Failed to send tribute payment notification: {e}")
|
|
else:
|
|
await session.commit()
|
|
return web.Response(status=200, text="ok")
|
|
|
|
|
|
async def tribute_webhook_route(request: web.Request):
|
|
"""AIOHTTP route handler for Tribute webhook calls."""
|
|
tribute_service: TributeService = request.app['tribute_service']
|
|
raw_body = await request.read()
|
|
signature_header = request.headers.get('trbt-signature')
|
|
return await tribute_service.handle_webhook(raw_body, signature_header)
|