tgadmin/webadmin/function.py

from fastapi import FastAPI, HTTPException, Request, Cookie
from cryptography.fernet import Fernet
from sqlalchemy import select
from db.db import async_session
from db import crud
from db.models import User
from exc import NotAuthenticatedException, NewsletterException, NotAccessException
import logging
import os
import sys
from sqlalchemy.ext.asyncio import AsyncSession
import asyncio

sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '../')))

from bot.db.models import UserTg
from bot.config import bot
from aiogram.utils.keyboard import InlineKeyboardBuilder, ReplyKeyboardBuilder

def start_markup(): # старт кнопки

    markup = (
        InlineKeyboardBuilder()
        .button(text='Тич-Пицца', callback_data='pizza') # 
        .button(text='Ебидоеби', callback_data='ebi')
        .adjust(1, repeat=True)
        .as_markup()
    )
    
    return markup

key = os.getenv('SECRET_KEY_AUTH')
cipher_suite = Fernet(key)


def encrypt_token(token: str) -> str:
    return cipher_suite.encrypt(token.encode()).decode()

def decrypt_token(encrypted_token: str) -> str:
    return cipher_suite.decrypt(encrypted_token.encode()).decode()

async def get_current_user(request: Request, session_token: str = Cookie(None)):
    
    if not session_token:
        return None
    try:
        decrypted_token = decrypt_token(session_token)
    except Exception as ex:
        logging.error(ex)
        return None
    async with async_session() as db_session:
        session = await crud.get_session_by_token(db_session, decrypted_token)
        if not session:
            return None
        user = await db_session.get(User, session.user_id)
        return user

async def get_authenticated_user(request: Request, session_token: str = Cookie(None)) -> User:
    user = await get_current_user(request, session_token)
    if user is None: 
        raise NotAuthenticatedException(status_code=401, detail="Not authenticated")
    return user

async def get_user_with_access(request: Request, session_token: str = Cookie(None)) -> User:
    user = await get_current_user(request, session_token)
    if user is None: 
        raise NotAuthenticatedException(status_code=401, detail="Not authenticated")
    if user.has_access == False:
        raise NotAccessException(status_code=401, detail="Нет доступа")
    
    return user


async def send_newsletter(message: str, db_session: AsyncSession) -> dict:
    try:
        # Создание запроса для получения всех пользователей
        query = select(UserTg).where(UserTg.has_access == True) # Запрос на выбор всех записей из UserTg
        
        # Выполнение запроса и получение всех пользователей
        result = await db_session.execute(query)
        users = result.scalars().all()  # Получаем все записи
        
        
        for user in users:
            try:
                await bot.send_message(chat_id=user.user_id, text=message)
                await asyncio.sleep(2.5)
            except Exception as ex:
                pass
    except Exception as ex:
        logging.error(ex)
        raise NewsletterException(status_code=502, detail=f"Ошибка, рассылка не была начата - {ex}")

async def send_msg(user_id: int):
    try:
        markup = start_markup()
        await bot.send_message(chat_id=user_id, text='⭐️ Добро пожаловать в бота, используйте кнопки ниже', reply_markup=markup)
    except Exception as ex:
        logging.error(f"ERROR send msg - {ex}")