31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
|
from aiogram.types import Message, CallbackQuery
|
||
|
|
from config import settings
|
||
|
|
from aiogram import Bot
|
||
|
|
from aiogram.dispatcher.middlewares.base import BaseMiddleware
|
||
|
|
from typing import Any, Awaitable, Callable, Dict
|
||
|
|
from sql_function import databasework
|
||
|
|
|
||
|
|
|
||
|
|
class CreateUserMiddleware(BaseMiddleware): # ---- > мидлвар создание юзера
|
||
|
|
async def __call__(
|
||
|
|
self,
|
||
|
|
handler: Callable[[CallbackQuery, Dict[str, Any]], Awaitable[Any]],
|
||
|
|
event: Message,
|
||
|
|
data: Dict[str, Any],
|
||
|
|
) -> Any:
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
check_user = await databasework.check_user(event.from_user.id) # ---> достаем юзера
|
||
|
|
if check_user == None: # если юзера нет -
|
||
|
|
await databasework.create_user(event.from_user.id, event.from_user.username) # - создаем
|
||
|
|
|
||
|
|
|
||
|
|
elif check_user[2] != event.from_user.username: # обновляем юзернейм если изменился
|
||
|
|
await databasework.update_username_user(event.from_user.username, event.from_user.id)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
return await handler(event, data)
|
||
|
|
|