Zakhar Izmaylov
a2c71a49f2
- Consolidated and simplified code formatting by removing unnecessary line breaks and improving readability in various functions. - Updated the handling of backup file sending in backup.py for better clarity. - Streamlined error handling and logging messages in several handlers to enhance consistency. - Adjusted the Makefile to exclude specific files during Ruff checks and formatting. - Made minor adjustments to function signatures and parameter handling for improved clarity and consistency. This commit enhances code maintainability and readability without altering functionality.
86 lines
2.7 KiB
Python
86 lines
2.7 KiB
Python
import asyncio
|
|
import uuid
|
|
from datetime import datetime, timedelta
|
|
from typing import Any
|
|
|
|
import pytz
|
|
from py3xui import AsyncApi
|
|
|
|
from client import add_client
|
|
from config import ADMIN_PASSWORD, ADMIN_USERNAME, LIMIT_IP, PUBLIC_LINK, SUPERNODE, TOTAL_GB, TRIAL_TIME
|
|
from database import get_servers_from_db, store_key, use_trial
|
|
from handlers.texts import INSTRUCTIONS
|
|
from handlers.utils import generate_random_email, get_least_loaded_cluster
|
|
from logger import logger
|
|
|
|
|
|
async def create_trial_key(tg_id: int, session: Any):
|
|
try:
|
|
trial_status = await session.fetchval("SELECT trial FROM connections WHERE tg_id = $1", tg_id)
|
|
if trial_status == 1:
|
|
return {"error": "Вы уже использовали пробную версию."}
|
|
except Exception as e:
|
|
logger.error(f"Ошибка при проверке триала: {e}")
|
|
|
|
client_id = str(uuid.uuid4())
|
|
base_email = generate_random_email()
|
|
public_link = f"{PUBLIC_LINK}{base_email}/{tg_id}"
|
|
instructions = INSTRUCTIONS
|
|
result = {"key": public_link, "instructions": instructions, "email": base_email}
|
|
|
|
moscow_tz = pytz.timezone("Europe/Moscow")
|
|
current_time = datetime.now(moscow_tz)
|
|
expiry_time = current_time + timedelta(days=TRIAL_TIME)
|
|
expiry_timestamp = int(expiry_time.timestamp() * 1000)
|
|
|
|
clusters = await get_servers_from_db()
|
|
least_loaded_cluster = await get_least_loaded_cluster()
|
|
if least_loaded_cluster not in clusters:
|
|
raise ValueError(f"Кластер {least_loaded_cluster} не найден в базе данных.")
|
|
|
|
servers_in_cluster = clusters[least_loaded_cluster]
|
|
tasks = []
|
|
|
|
for server_info in servers_in_cluster:
|
|
server_name = server_info.get("server_name", "unknown")
|
|
|
|
if SUPERNODE:
|
|
email = f"{base_email}_{server_name.lower()}"
|
|
else:
|
|
email = base_email
|
|
|
|
tasks.append(
|
|
add_client(
|
|
AsyncApi(
|
|
server_info["api_url"],
|
|
username=ADMIN_USERNAME,
|
|
password=ADMIN_PASSWORD,
|
|
),
|
|
client_id,
|
|
email,
|
|
tg_id,
|
|
limit_ip=LIMIT_IP,
|
|
total_gb=TOTAL_GB,
|
|
expiry_time=expiry_timestamp,
|
|
enable=True,
|
|
flow="xtls-rprx-vision",
|
|
inbound_id=int(server_info["inbound_id"]),
|
|
sub_id=base_email,
|
|
)
|
|
)
|
|
|
|
await asyncio.gather(*tasks)
|
|
|
|
await store_key(
|
|
tg_id,
|
|
client_id,
|
|
base_email,
|
|
expiry_timestamp,
|
|
public_link,
|
|
server_id=least_loaded_cluster,
|
|
session=session,
|
|
)
|
|
|
|
await use_trial(tg_id, session)
|
|
return result
|