Vladless
68 lines
2.7 KiB
Python
68 lines
2.7 KiB
Python
import random
|
|
import asyncpg
|
|
import time
|
|
import uuid # Импортируем модуль для генерации UUID
|
|
from config import DATABASE_URL, SERVERS, ADMIN_USERNAME, ADMIN_PASSWORD
|
|
from auth import login_with_credentials, link # Импортируйте необходимые функции
|
|
from client import add_client
|
|
from database import store_key, add_connection
|
|
from handlers.texts import INSTRUCTIONS
|
|
from datetime import datetime, timedelta
|
|
|
|
def generate_random_email():
|
|
"""Генерирует случайный набор символов."""
|
|
random_string = ''.join(random.choices('abcdefghijklmnopqrstuvwxyz0123456789', k=6))
|
|
return random_string
|
|
|
|
async def create_trial_key(tg_id: int):
|
|
conn = await asyncpg.connect(DATABASE_URL)
|
|
try:
|
|
server_id = await get_least_loaded_server(conn)
|
|
session = await login_with_credentials(server_id, ADMIN_USERNAME, ADMIN_PASSWORD)
|
|
current_time = datetime.utcnow()
|
|
|
|
expiry_time = current_time + timedelta(days=30, hours=3)
|
|
expiry_timestamp = int(expiry_time.timestamp() * 1000)
|
|
|
|
client_id = str(uuid.uuid4())
|
|
email = generate_random_email()
|
|
response = await add_client(
|
|
session, server_id, client_id, email, tg_id,
|
|
limit_ip=1, total_gb=0, expiry_time=expiry_timestamp,
|
|
enable=True, flow="xtls-rprx-vision"
|
|
)
|
|
if response.get("success"):
|
|
connection_link = await link(session, server_id, client_id, email)
|
|
|
|
existing_connection = await conn.fetchrow('SELECT * FROM connections WHERE tg_id = $1', tg_id)
|
|
|
|
if existing_connection:
|
|
await conn.execute('UPDATE connections SET trial = 1 WHERE tg_id = $1', tg_id)
|
|
else:
|
|
await add_connection(tg_id, 0, 1)
|
|
|
|
await store_key(tg_id, client_id, email, expiry_timestamp, connection_link, server_id)
|
|
|
|
instructions = INSTRUCTIONS
|
|
return {
|
|
'key': connection_link,
|
|
'instructions': instructions
|
|
}
|
|
else:
|
|
return {'error': 'Не удалось добавить клиента на панель'}
|
|
finally:
|
|
await conn.close()
|
|
|
|
async def get_least_loaded_server(conn):
|
|
least_loaded_server_id = None
|
|
min_load_percentage = float('inf')
|
|
|
|
for server_id, server in SERVERS.items():
|
|
count = await conn.fetchval('SELECT COUNT(*) FROM keys WHERE server_id = $1', server_id)
|
|
percent_full = (count / 60) * 100 if count <= 60 else 100
|
|
if percent_full < min_load_percentage:
|
|
min_load_percentage = percent_full
|
|
least_loaded_server_id = server_id
|
|
|
|
return least_loaded_server_id
|