Files
Solo_bot/client.py
T
2024-09-20 03:48:41 +03:00

122 lines
4.6 KiB
Python

import requests
import json
from config import ADMIN_USERNAME, ADMIN_PASSWORD, GET_INBOUNDS_URL, DATABASE_PATH
import uuid
from auth import login_with_credentials
from datetime import datetime, timedelta
import aiosqlite
def add_client(session, client_id: str, email: str, tg_id: str, limit_ip: int, total_gb: int, expiry_time: int, enable: bool, flow: str):
url = 'https://solonet.pocomacho.ru:62553/solonet/panel/api/inbounds/addClient'
# Преобразуем email в нижний регистр
email = email.lower()
# Формируем данные клиента
client_data = {
"id": client_id,
"alterId": 0,
"email": email,
"limitIp": limit_ip,
"totalGB": total_gb,
"expiryTime": expiry_time,
"enable": enable,
"tgId": tg_id,
"subId": "",
"flow": flow,
}
# Преобразуем данные клиента в строку JSON
settings = json.dumps({"clients": [client_data]})
# Формируем тело запроса
data = {
"id": 1, # Если id динамически изменяется, замените это значение
"settings": settings
}
headers = {
'Content-Type': 'application/json', # Если требуется токен авторизации
}
response = session.post(url, json=data, headers=headers)
# Выводим информацию о запросе и ответе
print(f"Запрос на добавление клиента: {data}")
print(f"Статус ответа: {response.status_code}")
print(f"Ответ от сервера: {response.text}")
if response.status_code == 200:
print(f"Клиент добавлен: email={email}")
else:
print(f"Ошибка при добавлении клиента: {response.status_code}, {response.text}")
import json
def extend_client_key(session, tg_id, client_id, email: str, new_expiry_time: int) -> bool:
# Получаем текущие данные клиента
response = session.get(f"https://solonet.pocomacho.ru:62553/solonet/panel/api/inbounds/getClientTraffics/{email}")
print(f"GET {response.url} Status: {response.status_code}")
print(f"GET Response: {response.text}")
if response.status_code != 200:
print(f"Ошибка при получении данных клиента: {response.status_code} - {response.text}")
return False
client_data = response.json().get("obj", {})
print(client_data)
if not client_data:
print("Не удалось получить данные клиента.")
return False
# Получаем текущий срок действия
current_expiry_time = client_data.get('expiryTime', 0)
# Если нет текущего срока, используем новый срок
if current_expiry_time == 0:
current_expiry_time = new_expiry_time
# Определяем новый срок окончания
updated_expiry_time = max(current_expiry_time, new_expiry_time)
# Формируем данные для обновления
payload = {
"id": 1, # Если id динамически изменяется, замените это значение
"settings": json.dumps({
"clients": [
{
"id": client_id,
"alterId": 0,
"email": email.lower(), # Приведение email к нижнему регистру
"limitIp": 2,
"totalGB": 429496729600000,
"expiryTime": updated_expiry_time,
"enable": True,
"tgId": tg_id,
"subId": "",
"flow": "xtls-rprx-vision"
}
]
})
}
headers = {
'Content-Type': 'application/json',
'Accept': 'application/json'
}
try:
response = session.post(f"https://solonet.pocomacho.ru:62553/solonet/panel/api/inbounds/updateClient/{client_id}", json=payload, headers=headers)
print(f"POST {response.url} Status: {response.status_code}")
print(f"POST Request Data: {json.dumps(payload, indent=2)}")
print(f"POST Response: {response.text}")
if response.status_code == 200:
return True
else:
print(f"Ошибка при продлении ключа: {response.status_code} - {response.text}")
return False
except Exception as e:
print(f"Ошибка запроса: {e}")
return False