web.py

from fastapi import FastAPI, Request
from fastapi.responses import HTMLResponse
from fastapi.responses import JSONResponse
from fastapi.templating import Jinja2Templates
import requests
import hashlib
import json
import time
import uvicorn

def login_api(studentId, password, captcha,
          api_url="http://szgxjx.csu.edu.cn/api/security-server/dietc/sys/authorizing/login"):
    ## 密码转化为md5
    password_md5 = hashlib.md5(password.encode()).hexdigest()
    payload = {
        "queryParam": "QHP2xQRpUEvs1766113234391",
        "password": password_md5,
        "username": studentId,
        "code": captcha
        }
    headers = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/143.0.0.0 Safari/537.36",
        "Connection": "keep-alive",
        "Accept": "application/json, text/plain, */*",
        "Accept-Encoding": "gzip, deflate",
        "Content-Type": "application/json",
        "DNT": "1",
        "Origin": "http://szgxjx.csu.edu.cn",
        "Referer": "http://szgxjx.csu.edu.cn/zndxstudent/",
        "Accept-Language": "zh-CN,zh;q=0.9"
        }
    response = requests.post(api_url, data=json.dumps(payload), headers=headers)
    response.raise_for_status()
    return response.json()

app = FastAPI()

templates = Jinja2Templates(directory="templates")


@app.get("/", response_class=HTMLResponse)
async def main(request: Request):
    current_timestamp = str(int(time.time() * 1000))
    captcha_url = "http://szgxjx.csu.edu.cn/api/security-server/dietc/servlet/patchcaServlet?action=QHP2xQ&appid=RpUEvs&key=1766113234391"
    return templates.TemplateResponse(
        request=request, name="main.html", context={"welcome": "CSU工训报告自动化系统", "captcha_url": captcha_url}
    )

@app.post("/login", response_class=JSONResponse)
async def login(request: Request):
    ## 解析body中的json数据
    data = await request.json()
    studentId = data.get("username")
    password = data.get("password")
    captcha = data.get("captcha")
    result = login_api(studentId, password, captcha)
    response = {
        "success": False,
        "message": ""
    }
    if result['status'] != 200:
        response['message'] = result.get('message', '登录失败，未知错误')
    else:
        response['success'] = True
        response['message'] = "登录成功"
        response['name'] = result['datas']['user']['realName']
        response['token'] = result['datas']['token']
    return JSONResponse(content=response)
    

uvicorn.run(app, host="127.0.0.1", port=8000)