94 lines
4.5 KiB
Python
94 lines
4.5 KiB
Python
from fastapi import FastAPI, Form, Request
|
|
from fastapi.responses import HTMLResponse, JSONResponse
|
|
from fastapi.templating import Jinja2Templates
|
|
from fastapi.staticfiles import StaticFiles
|
|
import httpx
|
|
import uuid
|
|
import string
|
|
import random
|
|
import xml.etree.ElementTree as ET
|
|
|
|
app = FastAPI()
|
|
|
|
templates = Jinja2Templates(directory="app/templates")
|
|
|
|
ascii_list = list(string.ascii_letters + string.digits)
|
|
DESKTOP_NAME = "DESKTOP-" + "".join(random.sample(ascii_list, 7)).upper()
|
|
MACHINE_ID = uuid.uuid4()
|
|
USERNAME = "".join(random.sample(ascii_list, 7)).upper()
|
|
VERSION = 2024100
|
|
BUILD_NUMBER = "2024.1.4+Build+CL-241.18034.45"
|
|
|
|
# 配置静态文件
|
|
app.mount("/static", StaticFiles(directory="app/static"), name="static")
|
|
|
|
# 获取并解析 XML 响应
|
|
def get_xml_data(server: str, ticket_path: str):
|
|
client = httpx.Client(verify=False)
|
|
try:
|
|
response = client.get(server + ticket_path, timeout=3) # 设置超时为10秒
|
|
response.raise_for_status() # 检查响应状态
|
|
root = ET.fromstring(response.text)
|
|
|
|
return {
|
|
"action": root.find("action").text,
|
|
"confirmation_stamp": root.find("confirmationStamp").text,
|
|
"lease_signature": root.find("leaseSignature").text,
|
|
"response_code": root.find("responseCode").text,
|
|
"salt": root.find("salt").text,
|
|
"server_lease": root.find("serverLease").text,
|
|
"server_uid": root.find("serverUid").text,
|
|
"validation_deadline_period": root.find("validationDeadlinePeriod").text,
|
|
"validation_period": root.find("validationPeriod").text,
|
|
"message": root.find("message").text
|
|
}
|
|
except httpx.ReadTimeout:
|
|
return {"error": "Timeout: Please try again later."}
|
|
except Exception as e:
|
|
return {"error": f"An unexpected error occurred: {str(e)}. It seems that this server is not avaliable or you can try again later."}
|
|
|
|
@app.get("/", response_class=HTMLResponse)
|
|
async def index(request: Request):
|
|
return templates.TemplateResponse("index.html", {"request": request})
|
|
|
|
@app.post("/fetch", response_class=HTMLResponse)
|
|
async def fetch_data(request: Request, server: str = Form(...)):
|
|
RELEASE_TICKET_PATH = f"/rpc/releaseTicket.action?buildNumber={BUILD_NUMBER}&clientVersion=16&hostName={DESKTOP_NAME}&machineId={MACHINE_ID}&productCode=cfc7082d-ae43-4978-a2a2-46feb1679405&productFamilyId=cfc7082d-ae43-4978-a2a2-46feb1679405&salt=1726973268812&secure=false&ticketId=rsoaehfbbg&userName={USERNAME}"
|
|
OBTAIN_TICKET_PATH = f"/rpc/obtainTicket.action?buildDate=20240409&{BUILD_NUMBER}&clientVersion=16&hostName={DESKTOP_NAME}&machineId={MACHINE_ID}&productCode=cfc7082d-ae43-4978-a2a2-46feb1679405&productFamilyId=cfc7082d-ae43-4978-a2a2-46feb1679405&salt=1726973269845&secure=false&userName={USERNAME}&version={VERSION}&versionNumber={VERSION}"
|
|
|
|
release_ticket_data = get_xml_data(server, RELEASE_TICKET_PATH)
|
|
obtain_ticket_data = get_xml_data(server, OBTAIN_TICKET_PATH)
|
|
|
|
return templates.TemplateResponse("index.html", {
|
|
"request": request,
|
|
"release_data": release_ticket_data,
|
|
"obtain_data": obtain_ticket_data,
|
|
"server": server
|
|
})
|
|
|
|
@app.get("/api/fetch")
|
|
@app.post("/api/fetch")
|
|
async def api_fetch(server: str = Form(...)):
|
|
return await handle_fetch(server)
|
|
|
|
async def handle_fetch(server: str):
|
|
RELEASE_TICKET_PATH = f"/rpc/releaseTicket.action?buildNumber={BUILD_NUMBER}&clientVersion=16&hostName={DESKTOP_NAME}&machineId={MACHINE_ID}&productCode=cfc7082d-ae43-4978-a2a2-46feb1679405&productFamilyId=cfc7082d-ae43-4978-a2a2-46feb1679405&salt=1726973268812&secure=false&ticketId=rsoaehfbbg&userName={USERNAME}"
|
|
OBTAIN_TICKET_PATH = f"/rpc/obtainTicket.action?buildDate=20240409&{BUILD_NUMBER}&clientVersion=16&hostName={DESKTOP_NAME}&machineId={MACHINE_ID}&productCode=cfc7082d-ae43-4978-a2a2-46feb1679405&productFamilyId=cfc7082d-ae43-4978-a2a2-46feb1679405&salt=1726973269845&secure=false&userName={USERNAME}&version={VERSION}&versionNumber={VERSION}"
|
|
|
|
release_ticket_data = get_xml_data(server, RELEASE_TICKET_PATH)
|
|
obtain_ticket_data = get_xml_data(server, OBTAIN_TICKET_PATH)
|
|
|
|
available = False
|
|
if obtain_ticket_data.get("response_code") == "OK":
|
|
available = True
|
|
elif obtain_ticket_data.get("response_code") == "Error":
|
|
available = False
|
|
|
|
return JSONResponse(content={
|
|
"available": available,
|
|
"message": obtain_ticket_data.get("message", ""),
|
|
"data": {
|
|
"release_ticket_data": release_ticket_data,
|
|
"obtain_ticket_data": obtain_ticket_data,
|
|
}
|
|
}) |