yuketangHelperBUU/get_info.py at main · MuWinds/yuketangHelperBUU · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
# -*- coding: utf-8 -*-
# version 1
# developed by MuWinds
import requests
import time
import get_websockets
import json


def getUniversityId(domain):
    url = (
        "https://"
        + domain
        + "/edu_admin/get_custom_university_info/?current=1&_="
        + str((round(time.time() * 1000)))
    )
    headers = {
        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_14_4) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/87.0.4280.67 Safari/537.36",
        "Content-Type": "application/json",
    }
    response = requests.get(url=url, headers=headers).json()
    data = response["data"]
    print(data["university_id"])
    return data["university_id"]


def getWebSocketInfo(domain):
    ws = None
    try:
        ws = get_websockets.WebSocketQrcode()
        message = ws.run(domain)
        return message
    except KeyboardInterrupt:
        print("\n检测到用户中断，正在关闭WebSocket连接...")
        if ws:
            ws.close()  # 确保 WebSocket 连接被关闭
        raise  # 重新抛出异常让外层处理
    except Exception as e:
        print(f"WebSocket连接发生错误: {e}")
        if ws:
            ws.close()
        return None


def getCookies(domain):
    is_cached = input(
        "是否用已有cookies?是输入1(否则会将原来的cookies以新获得cookies覆盖)"
    )
    if is_cached == "1":
        filename = input("请输入cookies文件名:")
        ck_file = open(filename + ".txt", "r")
        cookie = ck_file.read()
    else:
        university_id = getUniversityId(domain)
        login_message = getWebSocketInfo(domain)
        # message转json
        message = json.loads(login_message)
        auth_info = message["Auth"]
        user_id = message["UserID"]
        verify_url = (
            "https://"
            + domain
            + "/edu_admin/account/login/verify-origin-system-bind?term=latest&uv_id="
            + str(university_id)
        )
        verify_header = {
            "referer": "https://" + domain + "/pro/portal/home/",
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/135.0.0.0 Safari/537.36",
            "Content-Type": "application/json",
            "Cookie": "university_id=3325;platform_id=3;xtbz=cloud;platform_type=1;",
            "Origin": "https://" + domain,
            "Platform-Id": "3",
            "University-Id": str(university_id),
            "Terminal-Type": "web",
            "X-Client": "web",
            "X-Csrftoken": "undefined",
            "Xtbz": "cloud",
        }
        verify_form = {"auth": auth_info, "origin_user_id": str(user_id)}
        response = requests.post(verify_url, json=verify_form, headers=verify_header)
        cookie = response.headers.get("Set-Cookie")
        filename = input("请输入cookies文件名:")
        ck_write = open(filename + ".txt", "w")
        ck_write.write(cookie)
    return cookie


def extract_specific_cookies(cookie_string):
    cookies = {}
    # 将cookie字符串分割成单独的cookie条目
    for cookie in cookie_string.split(","):
        # 进一步处理每个cookie，提取键和值
        key_value = cookie.split(";")[0].strip()  # 只获取键值对部分
        if "=" in key_value:
            key, value = key_value.split("=", 1)
            cookies[key] = value
    print(cookies.get("csrftoken"))
    print(cookies.get("sessionid"))
    # 返回特定的cookie
    return cookies.get("csrftoken"), cookies.get("sessionid")