Доделал наконец

- Покупку, но надо подправить пару моментов
- И выдача конфигураций
- Проверил работу panel.py наконец, исправил пару приколов
This commit is contained in:
2024-10-26 19:38:27 +03:00
parent 7437525d0d
commit 824e007786
5 changed files with 318 additions and 375 deletions

View File

@@ -9,27 +9,11 @@ from dateutil.relativedelta import relativedelta
with open('config.json', 'r') as file : config = json.load(file)
def generate_date(months):
now = datetime.now()
# Преобразуем months в число
try:
months = int(months) # или float(months), если месяцы могут быть дробными
except ValueError:
raise TypeError("months должно быть числом")
future_date = now + timedelta(days=30 * months)
return future_date.isoformat()
def generate_random_string(length=8):
characters = string.ascii_letters + string.digits
return ''.join(secrets.choice(characters) for _ in range(length))
def generate_uuid():
return str(uuid.uuid4())
class PanelInteraction:
def __init__(self, base_url, login_data, logger_):
self.base_url = base_url
@@ -47,7 +31,8 @@ class PanelInteraction:
def login(self):
login_url = self.base_url + "/login"
response = requests.post(login_url, data=self.login_data)
self.logger.info(f"Login URL : {login_url}")
response = requests.post(login_url, data=self.login_data, verify=False)
if response.status_code == 200:
session_id = response.cookies.get("3x-ui")
return session_id
@@ -55,27 +40,32 @@ class PanelInteraction:
self.logger.error(f"Login failed: {response.status_code}")
return None
def getInboundInfo(self,inboundId):
def getInboundInfo(self, inboundId):
url = f"{self.base_url}/panel/api/inbounds/get/{inboundId}"
response = requests.get(url, headers=self.headers)
if response.status_code == 200:
return response.json()
else:
self.logger.error(f"Failed to get inbound info: {response.status_code}")
self.logger.debug("Response:", response.text)
return None
try:
response = requests.get(url, headers=self.headers, verify=False)
if response.status_code == 200:
return response.json()
else:
self.logger.error(f"Failed to get inbound info: {response.status_code}")
self.logger.debug("Response:", response.text)
return None
finally:
self.logger.info("Finished attempting to get inbound info.")
def get_client_traffic(self, email):
url = f"{self.base_url}/panel/api/inbounds/getClientTraffics/{email}"
response = requests.get(url, headers=self.headers)
if response.status_code == 200:
return response.json()
else:
self.logger.error(f"Failed to get client traffic: {response.status_code}")
self.logger.debug("Response:", response.text)
return None
try:
response = requests.get(url, headers=self.headers, verify=False)
if response.status_code == 200:
return response.json()
else:
self.logger.error(f"Failed to get client traffic: {response.status_code}")
self.logger.debug("Response:", response.text)
return None
finally:
self.logger.info("Finished attempting to get client traffic.")
def update_client_expiry(self, client_uuid, new_expiry_time, client_email):
url = f"{self.base_url}/panel/api/inbounds/updateClient"
update_data = {
@@ -96,23 +86,27 @@ class PanelInteraction:
]
})
}
response = requests.post(url, headers=self.headers, json=update_data)
if response.status_code == 200:
self.logger.debug("Client expiry time updated successfully.")
else:
self.logger.error(f"Failed to update client: {response.status_code} {response.text}")
try:
response = requests.post(url, headers=self.headers, json=update_data, verify=False)
if response.status_code == 200:
self.logger.debug("Client expiry time updated successfully.")
else:
self.logger.error(f"Failed to update client: {response.status_code} {response.text}")
finally:
self.logger.info("Finished attempting to update client expiry.")
def add_client(self, inbound_id, months):
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": generate_random_string(),
"email": email,
"limitIp": 2,
"totalGB": 0,
"expiryTime": generate_date(months),
"flow":"xtls-rprx-vision",
"expiryTime": expiry_date,
"enable": True,
"tgId": "",
"subId": ""
@@ -123,11 +117,13 @@ class PanelInteraction:
"id": inbound_id,
"settings": json.dumps(client_info)
}
response = requests.post(url, headers=self.headers, json=payload)
if response.status_code == 200:
self.logger.debug("Client added successfully!")
return response.json()
else:
self.logger.error(f"Failed to add client: {response.status_code}")
self.logger.debug("Response:", response.text)
return None
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.")