43 lines
1.9 KiB
Python
43 lines
1.9 KiB
Python
|
import aiohttp # Для выполнения HTTP-запросов
|
|||
|
|
from loguru import logger
|
|||
|
|
from database.db_main import User, session
|
|||
|
|
from datetime import datetime
|
|||
|
|
from .crypto import decrypt_password
|
|||
|
|
|
|||
|
|
# Функция для отправки POST-запроса к API и получения токена
|
|||
|
|
async def get_token_from_api(email, password):
|
|||
|
|
url = "https://api.otask.ru/api/v1/auth/login"
|
|||
|
|
headers = {
|
|||
|
|
"Content-Type": "application/json",
|
|||
|
|
"Accept": "application/json"
|
|||
|
|
}
|
|||
|
|
body = {
|
|||
|
|
"email": email,
|
|||
|
|
"password": password
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
async with aiohttp.ClientSession() as session:
|
|||
|
|
async with session.post(url, headers=headers, json=body) as response:
|
|||
|
|
if response.status == 200:
|
|||
|
|
return await response.json() # Возвращаем токен, если запрос успешен
|
|||
|
|
else:
|
|||
|
|
logger.error(f"Failed to get token. Status code: {response.status}")
|
|||
|
|
return None
|
|||
|
|
|
|||
|
|
|
|||
|
|
# Функция для обновления токена
|
|||
|
|
async def update_token(user: User):
|
|||
|
|
# Выполняем запрос на обновление токена
|
|||
|
|
new_tokens = await get_token_from_api(user.name, decrypt_password(user.password)) # Получаем новый токен
|
|||
|
|
if new_tokens:
|
|||
|
|
user.token = new_tokens['token']
|
|||
|
|
user.refresh_token = new_tokens['refresh_token']
|
|||
|
|
user.expires_in = new_tokens['expires_in'] # Если у вас есть это поле в модели
|
|||
|
|
user.last_token_update = datetime.now() # Обновляем дату последнего обновления токена
|
|||
|
|
session.commit()
|
|||
|
|
logger.info(f"Токен обновлён для пользователя: {user.telegram_id}")
|
|||
|
|
else:
|
|||
|
|
logger.warning(f"Не удалось обновить токен для пользователя: {user.telegram_id}")
|
|||
|
|
|
|||
|
|
|