mirror of
https://github.com/djimboy/djimbo_template_aio3.git
synced 2026-07-25 09:44:29 +00:00
dj
56 lines
2.0 KiB
Python
56 lines
2.0 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.db_users import UsersRepository
|
|
from tgbot.utils.const_functions import clear_html
|
|
|
|
|
|
# Проверка юзера в БД и его добавление/обновление
|
|
class ExistsUserMiddleware(BaseMiddleware):
|
|
def __init__(self, cache_ttl: int = BOT_USER_CACHE_TTL) -> None:
|
|
self.users = UsersRepository()
|
|
self.cache = TTLCache(maxsize=10_000, ttl=cache_ttl)
|
|
|
|
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 ""
|
|
user_name = clear_html(this_user.first_name)
|
|
user_surname = clear_html(this_user.last_name)
|
|
user_fullname = clear_html(this_user.first_name)
|
|
|
|
if user_name is None: user_name = ""
|
|
if user_surname is None: user_surname = ""
|
|
if user_fullname is None: user_fullname = ""
|
|
|
|
if len(user_surname) >= 1: user_fullname += f" {user_surname}"
|
|
|
|
user_data = (
|
|
user_login.lower(),
|
|
user_name,
|
|
user_surname,
|
|
user_fullname,
|
|
)
|
|
|
|
cached_user = self.cache.get(user_id)
|
|
|
|
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],
|
|
)
|
|
self.cache[user_id] = {"data": user_data, "user": user}
|
|
else:
|
|
user = cached_user["user"]
|
|
|
|
data['User'] = user
|
|
|
|
return await handler(event, data)
|