Private
Public Access
forked from FOSS/AutoShop-Djimbo-Simple
116 lines
4.5 KiB
Python
116 lines
4.5 KiB
Python
# - *- coding: utf- 8 - *-
|
|
from aiogram import BaseMiddleware
|
|
from cachetools import TTLCache
|
|
|
|
from tgbot.data.config import BOT_USER_CACHE_TTL
|
|
from tgbot.database import Referralx, Settingsx, Userx
|
|
from tgbot.utils.const_functions import clear_html
|
|
from tgbot.utils.misc.bot_logging import bot_logger
|
|
|
|
|
|
class ExistsUserMiddleware(BaseMiddleware):
|
|
# Создание кеша пользователей
|
|
def __init__(self, cache_ttl: int = BOT_USER_CACHE_TTL) -> None:
|
|
self.users = Userx()
|
|
self.referrals = Referralx()
|
|
self.cache = TTLCache(maxsize=10_000, ttl=cache_ttl)
|
|
|
|
@staticmethod
|
|
def _get_referrer_id(event) -> int | None:
|
|
event_text = getattr(event, "text", None)
|
|
|
|
if not isinstance(event_text, str):
|
|
return None
|
|
|
|
start_data = event_text.split(maxsplit=1)
|
|
|
|
if len(start_data) != 2 or start_data[0] != "/start":
|
|
return None
|
|
|
|
referral_data = start_data[1]
|
|
|
|
if not referral_data.startswith("r_") or not referral_data[2:].isdigit():
|
|
return None
|
|
|
|
return int(referral_data[2:])
|
|
|
|
# Добавление или обновление пользователя перед handler-ом
|
|
async def __call__(self, handler, event, data):
|
|
this_user = data.get("event_from_user")
|
|
|
|
if this_user is not None and not this_user.is_bot:
|
|
user_id = this_user.id
|
|
user_login = (this_user.username or "").lower()
|
|
user_name = clear_html(this_user.first_name) or ""
|
|
user_surname = clear_html(this_user.last_name) or ""
|
|
user_fullname = clear_html(this_user.full_name) or " ".join(filter(None, [user_name, user_surname]))
|
|
|
|
user_data = (
|
|
user_login,
|
|
user_name,
|
|
user_surname,
|
|
user_fullname,
|
|
)
|
|
|
|
cached_user = self.cache.get(user_id)
|
|
existing_user = None
|
|
|
|
if cached_user is None:
|
|
existing_user = await self.users.get(user_id=user_id)
|
|
|
|
is_new_user = cached_user is None and existing_user is None
|
|
|
|
if cached_user is None or cached_user["data"] != user_data:
|
|
user = await self.users.upsert(
|
|
user_id=user_id,
|
|
user_login=user_data[0],
|
|
user_name=user_data[1],
|
|
user_surname=user_data[2],
|
|
user_fullname=user_data[3],
|
|
)
|
|
else:
|
|
user = cached_user["user"]
|
|
|
|
referrer_id = self._get_referrer_id(event)
|
|
|
|
if is_new_user and referrer_id is not None:
|
|
registration_status = await self.referrals.register_referral(
|
|
referral_id=user_id,
|
|
referrer_id=referrer_id,
|
|
)
|
|
|
|
if registration_status == "ok":
|
|
user = await self.users.get_required(user_id=user_id)
|
|
get_settings = await Settingsx().get()
|
|
bot = data.get("bot") or getattr(event, "bot", None)
|
|
|
|
if bot is not None:
|
|
notifications = (
|
|
(
|
|
referrer_id,
|
|
"<b>🤝 Новый реферал</b>\n\n"
|
|
f"▪️ Пользователь: <a href='tg://user?id={user_id}'>{user.user_name}</a>\n"
|
|
f"▪️ Начислено: <code>{get_settings.referral_bonus_rub}₽</code>",
|
|
),
|
|
(
|
|
user_id,
|
|
"<b>🤝 Реферальный бонус</b>\n\n"
|
|
f"▪️ Вам начислено: <code>{get_settings.referral_bonus_rub}₽</code>",
|
|
),
|
|
)
|
|
|
|
for recipient_id, notification_text in notifications:
|
|
try:
|
|
await bot.send_message(recipient_id, notification_text)
|
|
except Exception:
|
|
bot_logger.warning(
|
|
"Не удалось отправить уведомление о реферальной регистрации",
|
|
exc_info=True,
|
|
)
|
|
|
|
self.cache[user_id] = {"data": user_data, "user": user}
|
|
|
|
data["User"] = user
|
|
|
|
return await handler(event, data)
|