139 lines
5.1 KiB
Python
139 lines
5.1 KiB
Python
import requests
|
|
import uuid
|
|
import string
|
|
import secrets
|
|
import json
|
|
from utils.LogCon import setup_logger, load_config
|
|
from datetime import datetime, timedelta
|
|
from dateutil.relativedelta import relativedelta
|
|
|
|
config = load_config()
|
|
|
|
|
|
|
|
def generate_uuid():
|
|
return str(uuid.uuid4())
|
|
|
|
class PanelInteraction:
|
|
def __init__(self, base_url, login_data, logger_):
|
|
self.base_url = base_url
|
|
self.login_data = login_data
|
|
self.logger = logger_
|
|
self.session_id = self.login()
|
|
if self.session_id:
|
|
self.headers = {
|
|
'Accept': 'application/json',
|
|
'Cookie': f'3x-ui={self.session_id}',
|
|
'Content-Type': 'application/json'
|
|
}
|
|
else:
|
|
raise ValueError("Login failed, session_id is None")
|
|
|
|
def login(self):
|
|
login_url = self.base_url + "/login"
|
|
self.logger.info(f"Attempting to login at: {login_url}")
|
|
try:
|
|
response = requests.post(login_url, data=self.login_data, verify=False, timeout=10)
|
|
response.raise_for_status()
|
|
session_id = response.cookies.get("3x-ui")
|
|
if session_id:
|
|
return session_id
|
|
else:
|
|
self.logger.error(f"Login failed: {response.status_code}")
|
|
self.logger.debug(f"Response content: {response.text}")
|
|
return None
|
|
except requests.RequestException as e:
|
|
self.logger.error(f"Login request failed: {e}")
|
|
return None
|
|
|
|
def getInboundInfo(self, inboundId):
|
|
url = f"{self.base_url}/panel/api/inbounds/get/{inboundId}"
|
|
try:
|
|
response = requests.get(url, headers=self.headers, verify=False, timeout=10)
|
|
response.raise_for_status()
|
|
if response:
|
|
return response.json()
|
|
else:
|
|
self.logger.error(f"Failed to get inbound info: {response.status_code}")
|
|
self.logger.debug("Response:", response.text)
|
|
return None
|
|
except requests.RequestException as e:
|
|
self.logger.error(f"Get inbound request failed: {e}")
|
|
|
|
def get_client_traffic(self, email):
|
|
url = f"{self.base_url}/panel/api/inbounds/getClientTraffics/{email}"
|
|
try:
|
|
response = requests.get(url, headers=self.headers, verify=False, timeout=10)
|
|
response.raise_for_status()
|
|
if response:
|
|
return response.json()
|
|
else:
|
|
self.logger.error(f"Failed to get client traffic: {response.status_code}")
|
|
self.logger.debug("Response:", response.text)
|
|
return None
|
|
except requests.RequestException as e:
|
|
self.loggin.error(f"Get client request failed: {e}")
|
|
|
|
def update_client_expiry(self, client_uuid, new_expiry_time, client_email):
|
|
url = f"{self.base_url}/panel/api/inbounds/updateClient"
|
|
update_data = {
|
|
"id": 1,
|
|
"settings": json.dumps({
|
|
"clients": [
|
|
{
|
|
"id": client_uuid,
|
|
"alterId": 0,
|
|
"email": client_email,
|
|
"limitIp": 2,
|
|
"totalGB": 0,
|
|
"expiryTime": new_expiry_time,
|
|
"enable": True,
|
|
"tgId": "",
|
|
"subId": ""
|
|
}
|
|
]
|
|
})
|
|
}
|
|
try:
|
|
response = requests.post(url, headers=self.headers, json=update_data, verify=False)
|
|
response.raise_for_status()
|
|
if response:
|
|
self.logger.debug("Client expiry time updated successfully.")
|
|
else:
|
|
self.logger.error(f"Failed to update client: {response.status_code}, {response.text}")
|
|
except requests.RequestException as e:
|
|
self.logger.error(f"Update client request failed: {e}")
|
|
|
|
def add_client(self, inbound_id, expiry_date,email):
|
|
url = f"{self.base_url}/panel/api/inbounds/addClient"
|
|
client_info = {
|
|
"clients": [
|
|
{
|
|
"id": generate_uuid(),
|
|
"alterId": 0,
|
|
"email": email,
|
|
"limitIp": 2,
|
|
"totalGB": 0,
|
|
"flow":"xtls-rprx-vision",
|
|
"expiryTime": expiry_date,
|
|
"enable": True,
|
|
"tgId": "",
|
|
"subId": ""
|
|
}
|
|
]
|
|
}
|
|
payload = {
|
|
"id": inbound_id,
|
|
"settings": json.dumps(client_info)
|
|
}
|
|
try:
|
|
response = requests.post(url, headers=self.headers, json=payload, verify=False)
|
|
if response.status_code == 200:
|
|
return response.json()
|
|
else:
|
|
self.logger.error(f"Failed to add client: {response.status_code}")
|
|
self.logger.debug("Response:", response.text)
|
|
return None
|
|
finally:
|
|
self.logger.info("Finished attempting to add client.")
|