From 333b85a9260be20df9aeb7733e2c8d28977350dc Mon Sep 17 00:00:00 2001
From: LinChiQ <111717457+LinChiQ@users.noreply.github.com>
Date: Sun, 4 Jun 2023 18:28:28 +0800
Subject: [PATCH] Add files via upload

---
 PickUp.py          | 235 +++++++++++++++++++++++++++++++++++++++++++
 ReserveTomorrow.py | 241 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 476 insertions(+)
 create mode 100644 PickUp.py
 create mode 100644 ReserveTomorrow.py

diff --git a/PickUp.py b/PickUp.py
new file mode 100644
index 0000000..6b11dc5
--- /dev/null
+++ b/PickUp.py
@@ -0,0 +1,235 @@
+from requests import post
+from requests import Session
+from json import loads
+from json import decoder
+from yagmail import SMTP
+from time import sleep
+import urllib.request
+import urllib.parse
+import http.cookiejar
+import http.cookies
+from threading import Thread
+from time import localtime
+
+# 该部分出于GitHub上的GoToLibCookie项目
+
+
+def get_code(url):
+    query = urllib.parse.urlparse(url).query
+    codes = urllib.parse.parse_qs(query).get('code')
+    if codes:
+        return codes.pop()
+    else:
+        raise ValueError("未找到授权码！")
+
+
+def get_cookie_string(code):
+    cookiejar = http.cookiejar.MozillaCookieJar()
+    opener = urllib.request.build_opener(
+        urllib.request.HTTPCookieProcessor(cookiejar))
+    response = opener.open(
+        "http://wechat.v2.traceint.com/index.php/urlNew/auth.html?" + urllib.parse.urlencode({
+            "r": "https://web.traceint.com/web/index.html",
+            "code": code,
+            "state": 1
+        })
+    )
+    cookie_items = []
+    for cookie in cookiejar:
+        cookie_items.append(f"{cookie.name}={cookie.value}")
+    cookie_string = '; '.join(cookie_items)
+    return cookie_string
+
+
+def KeepCookie(cookie_string, mail, receiver_mail):
+    session = Session()
+    cookie = http.cookies.SimpleCookie()
+    cookie.load(cookie_string)
+    for key, morsel in cookie.items():
+        session.cookies.set(key, morsel)
+    while True:
+        if session.cookies.keys().count("Authorization") > 1:
+            session.cookies.set("Authorization", domain="", value=None)
+        res = session.post("http://wechat.v2.traceint.com/index.php/graphql/", json={
+            "query": 'query getUserCancleConfig { userAuth { user { holdValidate: getSchConfig(fields: "hold_validate", extra: true) } } }',
+            "variables": {},
+            "operationName": "getUserCancleConfig"
+        })
+        try:
+            result = res.json()
+        except decoder.JSONDecodeError as err:
+            print("Error: %s" % err)
+            break
+        if result.get("errors") and result.get("errors")[0].get("code") != 0:
+            print("Cookie过期！")
+            mail.send(receiver_mail, 'Cookie过期！',
+                      'Cookie过期，请重新微信扫码并联系作者本人：2251698595')
+            exit(0)
+        print("Cookie正常！")
+        sleep(60)
+
+
+class DataUsed(object):
+    def __init__(self, cookies):
+        self.main_url = 'https://wechat.v2.traceint.com/index.php/graphql/'
+        self.headers = {
+            'Host': 'wechat.v2.traceint.com',
+            'Connection': 'keep-alive',
+            'Content-Length': '729',
+            'Origin': 'https://web.traceint.com',
+            'Content-Type': 'application/json',
+            'User-Agent': "Mozilla/5.0 (Windows NT 6.1WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.138 Safari/537.36 NetType/WIFI MicroMessenger/7.0.20.1781(0x6700143B) WindowsWechat(0x6307062c)",
+            'App-Version': '2.0.9',
+            'Accept': '*/*',
+            'Cookie': cookies,
+            'Sec-Fetch-Site': 'same-site',
+            'Sec-Fetch-Mode': 'cors',
+            'Sec-Fetch-Dest': 'empty',
+            'Referer': 'https://web.traceint.com/web/index.html',
+            'Accept-Encoding': 'gzip, deflate, br',
+            'Accept-Language': "zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7"
+        }
+        self.libs_json = {"operationName": "list", "query": "query list {\n userAuth {\n reserve {\n libs(libType: -1) {\n lib_id\n lib_floor\n is_open\n lib_name\n lib_type\n lib_group_id\n lib_comment\n lib_rt {\n seats_total\n seats_used\n seats_booking\n seats_has\n reserve_ttl\n open_time\n open_time_str\n close_time\n close_time_str\n advance_booking\n }\n }\n libGroups {\n id\n group_name\n }\n reserve {\n isRecordUser\n }\n }\n record {\n libs {\n lib_id\n lib_floor\n is_open\n lib_name\n lib_type\n lib_group_id\n lib_comment\n lib_color_name\n lib_rt {\n seats_total\n seats_used\n seats_booking\n seats_has\n reserve_ttl\n open_time\n open_time_str\n close_time\n close_time_str\n advance_booking\n }\n }\n }\n rule {\n signRule\n }\n }\n}"}
+
+    def GetFreeLibJson(self, free_lib_id):
+        return {"operationName": "libLayout", "query": "query libLayout($libId: Int, $libType: Int) {\n userAuth {\n reserve {\n libs(libType: $libType, libId: $libId) {\n lib_id\n is_open\n lib_floor\n lib_name\n lib_type\n lib_layout {\n seats_total\n seats_booking\n seats_used\n max_x\n max_y\n seats {\n x\n y\n key\n type\n name\n seat_status\n status\n }\n }\n }\n }\n }\n}", "variables": {"libId": free_lib_id}}
+
+    def GetFreeSeatJson(self, free_lib_id, free_seat_key):
+        return {"operationName": "reserueSeat", "query": "mutation reserueSeat($libId: Int!, $seatKey: String!, $captchaCode: String, $captcha: String!) {\n userAuth {\n reserve {\n reserueSeat(\n libId: $libId\n seatKey: $seatKey\n captchaCode: $captchaCode\n captcha: $captcha\n )\n }\n }\n}", "variables": {"seatKey": f"{free_seat_key}", "libId": free_lib_id, "captchaCode": "", "captcha": ""}}
+
+
+def GetLibsData(data_object):
+    req = post(data_object.main_url,
+               headers=data_object.headers, json=data_object.libs_json)
+    try:
+        text = loads(req.text)
+        print("获得自习信息成功！")
+    except Exception as e:
+        print("未知错误", e)
+        return False
+    try:
+        return text['data']['userAuth']['reserve']['libs']
+    except Exception as e:
+        print("请微信重新扫码并重新运行程序！！！")
+
+
+def FindFreeLib(libs_data, esacpe_seats):
+    step = 0
+    last = 0
+    if int(esacpe_seats[-1]) != 0:
+        for i in esacpe_seats:
+            if int(i) == last + 1:
+                libs_data.pop(0)
+                last += 1
+            else:
+                libs_data.pop(int(i) - last - 1)
+                last += 1
+    for lib_data in libs_data:
+        seats_has = lib_data['lib_rt']['seats_has']
+        lib_name = lib_data['lib_name']
+        if (seats_has != 0):
+            print(f"找到{lib_name}存在{seats_has}个空闲座位！")
+            step += 1
+            return lib_data['lib_id'], lib_data['lib_name']
+    if step == 0:
+        print("暂时不存在空闲图书室！")
+        return False, step
+
+
+def GetFreeLibSeats(data_object, free_lib_id, free_lib_name):
+    free_lib_json = data_object.GetFreeLibJson(free_lib_id)
+    req = post(data_object.main_url,
+               headers=data_object.headers, json=free_lib_json)
+    req.encoding = 'utf-8'
+    try:
+        text = loads(req.text)
+        print(f"成功获得{free_lib_name}的座位表！")
+    except Exception as e:
+        print("发生未知错误 ", e)
+    free_seats = text['data']['userAuth']['reserve']['libs'][0]['lib_layout']['seats']
+    keys = []
+    for free_seat in free_seats:
+        seat_status = free_seat['seat_status']
+        status = free_seat['status']
+        if seat_status == 1 and status == False:
+            keys.append(free_seat['key'])
+            print("找到空闲座位，", free_seat['key'])
+    return keys
+
+
+def ReserveSeat(data_object, free_lib_id, free_seat_keys, free_lib_name, mail, receiver_mail):
+    step = 0
+    for free_seat_key in free_seat_keys:
+        free_seat_json = data_object.GetFreeSeatJson(
+            free_lib_id, free_seat_key)
+        req = post(data_object.main_url,
+                   json=free_seat_json, headers=data_object.headers)
+        req.encoding = 'utf-8'
+        try:
+            text = loads(req.text)
+        except Exception as e:
+            print("未知错误 ", e)
+        if text['data']['userAuth']['reserve']['reserueSeat'] is True:
+            print(f"座位{free_seat_key}预定成功！")
+            mail.send(receiver_mail, '座位预定成功！',
+                      f'{free_lib_name}座位{free_seat_key}预定成功！')
+            step += 1
+            return True
+        else:
+            print(f"座位{free_seat_key}预定失败！", text)
+            sleep(1)
+            continue
+    if step == 0:
+        return False
+
+
+def main():
+    print("欢迎来到当日捡漏！\n\n")
+    mail = SMTP('lateyoung111@163.com',
+                'DJAFUFGCIMJXXRXU', 'smtp.163.com')
+    url = input("请输入微信复制的链接：")
+    receiver_mail = input("请输入接受通知邮箱（例：123456@qq.com）：")
+    escaped_seats = input("请输入不想去的楼层（输入0选择所有楼层，输入14代表一楼四楼不选择）：")
+    escaped_seats = list(escaped_seats)
+    code = get_code(url)
+    cookie_string = get_cookie_string(code)
+    print("\n获取Cookie成功！\n")
+    data_used = DataUsed('FROM_TYPE=weixin; v=5.5;'+cookie_string)
+    KeepLive = Thread(target=KeepCookie, args=(
+        cookie_string, mail, receiver_mail))
+    KeepLive.setDaemon(True)
+    KeepLive.start()
+    print("\nCookie保持在线中！\n")
+    sleep(1)
+    hour = int(input("请输入抢座小时（立马开始输入0）："))
+    min = int(input("请输入抢座分钟（立马开始输入0）："))
+    if hour != 0 and min != 0:
+        if localtime().tm_hour < hour:
+            print(f"时间已过！现在是{localtime().tm_hour}：{localtime().tm_min}")
+        elif localtime().tm_hour == hour and localtime().tm_min > min:
+            print(f"时间已过！现在是{localtime().tm_hour}：{localtime().tm_min}")
+        else:
+            print(f"程序进入等待状态，运行时间为：{hour}：{min}")
+            while True:
+                if localtime().tm_hour == hour and localtime().tm_min == min:
+                    break
+    print("程序开始运行！")
+    while True:
+        libs_data = GetLibsData(data_used)
+        free_lib_id, free_lib_name = FindFreeLib(libs_data, escaped_seats)
+        if free_lib_id is False:
+            print("3秒后执行下一次请求！")
+            sleep()
+            continue
+        free_seat_keys = GetFreeLibSeats(data_used, free_lib_id, free_lib_name)
+        success = ReserveSeat(data_used, free_lib_id,
+                              free_seat_keys, free_lib_name, mail, receiver_mail)
+        if success is True:
+            print("程序退出！")
+            exit(0)
+        else:
+            continue
+
+
+if __name__ == '__main__':
+    main()
diff --git a/ReserveTomorrow.py b/ReserveTomorrow.py
new file mode 100644
index 0000000..ea81a03
--- /dev/null
+++ b/ReserveTomorrow.py
@@ -0,0 +1,241 @@
+from requests import post
+from yagmail import SMTP
+from json import decoder
+from json import loads
+from time import sleep
+import urllib.request
+import urllib.parse
+import http.cookiejar
+import http.cookies
+from threading import Thread
+from time import localtime
+from requests import Session
+from websocket import create_connection
+from websocket import WebSocket
+
+
+def get_code(url):
+    query = urllib.parse.urlparse(url).query
+    codes = urllib.parse.parse_qs(query).get('code')
+    if codes:
+        return codes.pop()
+    else:
+        raise ValueError("未找到授权码！")
+
+
+def get_cookie_string(code):
+    cookiejar = http.cookiejar.MozillaCookieJar()
+    opener = urllib.request.build_opener(
+        urllib.request.HTTPCookieProcessor(cookiejar))
+    response = opener.open(
+        "http://wechat.v2.traceint.com/index.php/urlNew/auth.html?" + urllib.parse.urlencode({
+            "r": "https://web.traceint.com/web/index.html",
+            "code": code,
+            "state": 1
+        })
+    )
+    cookie_items = []
+    for cookie in cookiejar:
+        cookie_items.append(f"{cookie.name}={cookie.value}")
+    cookie_string = '; '.join(cookie_items)
+    return cookie_string
+
+
+def KeepCookie(cookie_string, mail, receiver_mail):
+    session = Session()
+    cookie = http.cookies.SimpleCookie()
+    cookie.load(cookie_string)
+    for key, morsel in cookie.items():
+        session.cookies.set(key, morsel)
+    while True:
+        if session.cookies.keys().count("Authorization") > 1:
+            session.cookies.set("Authorization", domain="", value=None)
+        res = session.post("http://wechat.v2.traceint.com/index.php/graphql/", json={
+            "query": 'query getUserCancleConfig { userAuth { user { holdValidate: getSchConfig(fields: "hold_validate", extra: true) } } }',
+            "variables": {},
+            "operationName": "getUserCancleConfig"
+        })
+        try:
+            result = res.json()
+        except decoder.JSONDecodeError as err:
+            print("Error: %s" % err)
+            break
+        if result.get("errors") and result.get("errors")[0].get("code") != 0:
+            print("Cookie过期！")
+            mail.send(receiver_mail, 'Cookie过期！',
+                      'Cookie过期，请重新微信扫码并联系作者本人：2251698595')
+            exit(0)
+        print("Cookie正常！")
+        sleep(60)
+
+
+class DataUsed(object):
+    def __init__(self, cookies):
+        self.main_url = 'https://wechat.v2.traceint.com/index.php/graphql/'
+        self.headers = {
+            'Host': 'wechat.v2.traceint.com',
+            'Connection': 'keep-alive',
+            'Content-Length': '729',
+            'Origin': 'https://web.traceint.com',
+            'Content-Type': 'application/json',
+            'User-Agent': "Mozilla/5.0 (Windows NT 6.1WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.138 Safari/537.36 NetType/WIFI MicroMessenger/7.0.20.1781(0x6700143B) WindowsWechat(0x6307062c)",
+            'App-Version': '2.0.9',
+            'Accept': '*/*',
+            'Cookie': cookies,
+            'Sec-Fetch-Site': 'same-site',
+            'Sec-Fetch-Mode': 'cors',
+            'Sec-Fetch-Dest': 'empty',
+            'Referer': 'https://web.traceint.com/web/index.html',
+            'Accept-Encoding': 'gzip, deflate, br',
+            'Accept-Language': "zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7"
+        }
+
+
+def HandleCookie(mail, receiver_mail):
+    url = input("请输入微信复制的链接：")
+    code = get_code(url)
+    cookie_string = get_cookie_string(code)
+    print("获取Cookie成功！")
+    data_used = DataUsed('FROM_TYPE=weixin; v=5.5;'+cookie_string)
+    KeepLive = Thread(target=KeepCookie, args=(
+        cookie_string, mail, receiver_mail))
+    KeepLive.setDaemon(True)
+    KeepLive.start()
+    print("Cookie保持在线中！")
+    sleep(1)
+    return data_used, 'FROM_TYPE=weixin; v=5.5;'+cookie_string
+
+
+def GetMail():
+    mail = SMTP('lateyoung111@163.com',
+                'DJAFUFGCIMJXXRXU', 'smtp.163.com')
+    receiver_mail = input("请输入接受通知邮箱（例：123456@qq.com）：")
+    return mail, receiver_mail
+
+
+def GetInQueue(cookie_string, hour, min):
+    headers = ["Pragma: no-cache",
+               "Cache-Control: no-cache",
+               "User-Agent: Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/81.0.4044.138 Safari/537.36 NetType/WIFI MicroMessenger/7.0.20.1781(0x6700143B) WindowsWechat(0x6307001e)",
+               f"Cookie: {cookie_string}",
+               "Accept-Language: zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7",
+               "Sec-WebSocket-Extensions: permessage-deflate; client_max_window_bits",
+               ]
+    if int(hour) != 0:
+        print("程序进入等待！")
+        while True:
+            if localtime().tm_hour == int(hour) and localtime().tm_min == int(min):
+                print("时间到，正在排队！")
+                break
+    ws = WebSocket()
+    ws.connect("wss://wechat.v2.traceint.com/ws?ns=prereserve/queue",
+               header=headers)
+    if ws.connected:
+        # 接收实时数据，并打印出来
+        while True:
+            ws.send('{"ns":"prereserve/queue","msg":""}')
+            a = ws.recv()
+            if a.find('u6392') != -1:  # 排队成功返回的第一个字符
+                print("排队结束！")
+                break
+        # 关闭连接
+        ws.close()
+    return True
+
+
+def QueryIndex(data_used, escaped_libs):
+    step = 0
+    last = 0
+    data = {"operationName": "index",
+            "query": "query index {\n userAuth {\n user {\n prereserveAuto: getSchConfig(extra: true, fields: \"prereserve.auto\")\n }\n currentUser {\n sch {\n isShowCommon\n }\n }\n prereserve {\n libs {\n is_open\n lib_floor\n lib_group_id\n lib_id\n lib_name\n num\n seats_total\n }\n }\n oftenseat {\n prereserveList {\n id\n info\n lib_id\n seat_key\n status\n }\n }\n }\n}"}
+    req = post(data_used.main_url, headers=data_used.headers, json=data)
+    try:
+        text = loads(req.text)
+    except Exception as e:
+        print("未知错误，", e)
+    libs = text['data']['userAuth']['prereserve']['libs']
+    if int(escaped_libs[-1]) != 0:
+        for i in escaped_libs:
+            if int(i) == last + 1:
+                libs.pop(0)
+                last += 1
+            else:
+                libs.pop(int(i) - last - 1)
+                last += 1
+    while True:
+        for lib in libs:
+            if lib['num'] != 0:
+                print(f"找到{lib['lib_floor']}存在空闲{lib['num']}！")
+                step += 1
+                return lib['lib_id'], lib['lib_floor']
+        if step == 0:
+            print("暂时不存在空闲！")
+            print("1秒后进行下一次请求！")
+            sleep(1)
+
+
+def LibLayOut(data_used, lib_id, lib_floor):
+    data = {"operationName": "libLayout",
+            "query": "query libLayout($libId: Int!) {\n userAuth {\n prereserve {\n libLayout(libId: $libId) {\n max_x\n max_y\n seats_booking\n seats_total\n seats_used\n seats {\n key\n name\n seat_status\n status\n type\n x\n y\n }\n }\n }\n }\n}", "variables": {"libId": lib_id}}
+    req = post(data_used.main_url, headers=data_used.headers, json=data)
+    try:
+        text = loads(req.text)
+        print(f"获取{lib_floor}座位成功！")
+    except Exception as e:
+        print("未知错误，", e)
+    seats = text['data']['userAuth']['prereserve']['libLayout']['seats']
+    keys = []
+    for seat in seats:
+        if seat['status'] is False and seat['type'] == 1:
+            keys.append(seat['key'])
+            print("找到空闲座位，", seat['key'])
+    return keys
+
+
+def TomorrowReserve(data_used, keys, lib_id, lib_floor, mail, receiver_mail):
+    for key in keys:
+        data = {
+            "operationName": "save",
+            "query": "mutation save($key: String!, $libid: Int!, $captchaCode: String, $captcha: String) {\n userAuth {\n prereserve {\n save(key: $key, libId: $libid, captcha: $captcha, captchaCode: $captchaCode)\n }\n }\n}",
+            "variables": {
+                "key": f"{key}.",
+                "libid": lib_id,
+                "captchaCode": "",
+                "captcha": ""
+            }
+        }
+        req = post(data_used.main_url, headers=data_used.headers, json=data)
+        try:
+            text = loads(req.text)
+        except Exception as e:
+            print("未知错误，", e)
+        if text['data']['userAuth']['prereserve']['save'] is True:
+            print(f"座位{lib_floor}的{key}预定成功！")
+            mail.send(receiver_mail, '预定成功！', f"座位{lib_floor}的{key}预定成功！")
+            return True
+        else:
+            print(f"座位{lib_floor}的{key}预定失败！")
+            continue
+
+
+def main():
+    min = 0
+    mail, receiver_mail = GetMail()
+    hour = input("请输入预约时间（输入0即刻预约，若想在9点准时开始请输入21）：")
+    if int(hour) != 0:
+        min = input("请输入预约分钟（若想在9点准时开始请输入0）：")
+    escaped_libs = input("请输入不想预约的楼层（0为全部预约，输入17代表一楼七楼不预约）：")
+    escaped_libs = list(escaped_libs)
+    data_used, cookie_string = HandleCookie(mail, receiver_mail)
+    queue = GetInQueue(cookie_string, hour, min)
+    if not queue:
+        print("排队错误！")
+        exit(0)
+    lib_id, lib_floor = QueryIndex(data_used, escaped_libs)
+    keys = LibLayOut(data_used, lib_id, lib_floor)
+    TomorrowReserve(data_used, keys, lib_id, lib_floor, mail, receiver_mail)
+
+
+if __name__ == '__main__':
+    print("\n欢迎来到明日预约！\n")
+    main()