Vladless
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
import asyncpg
|
|
import uuid
|
|
from config import DATABASE_URL, SERVERS, ADMIN_USERNAME, ADMIN_PASSWORD
|
|
from auth import login_with_credentials, link_subscription
|
|
from client import add_client
|
|
from database import store_key, add_connection
|
|
from handlers.texts import INSTRUCTIONS
|
|
from datetime import datetime, timedelta
|
|
from handlers.utils import generate_random_email, get_least_loaded_server
|
|
|
|
|
|
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=1, 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_subscription(email, server_id)
|
|
|
|
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()
|