From a3c67161686f8b9185491163f0d5357cc31d1343 Mon Sep 17 00:00:00 2001 From: YinFengQi <134215872+YinFengQi@users.noreply.github.com> Date: Sun, 22 Dec 2024 12:17:11 +0800 Subject: [PATCH] Update main.py(first commit) --- main.py | 109 +++++++++++++++++++++++++++++++++++++++++++------------- 1 file changed, 84 insertions(+), 25 deletions(-) diff --git a/main.py b/main.py index ba639e9..d3c0d87 100644 --- a/main.py +++ b/main.py @@ -1,10 +1,56 @@ +from tkinter import * +from tkinter.ttk import * +import platform +import ctypes + +import matplotlib.pyplot as plt +from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, NavigationToolbar2Tk + from Crypto.Cipher import AES from Crypto.Util.Padding import unpad import base64 import json -import matplotlib.pyplot as plt import requests -import platform + +class I_dont_know_what_is_a_good_name_for_this_fucking_gui_class_only_used_once(Tk): + def __init__(self, idserial, servicehall, serverid): + super().__init__() + self.title("THU Eating") + self.tk_idserial = StringVar(master=self, value=idserial) + self.tk_servicehall = StringVar(master=self, value=servicehall) + self.tk_serverid = StringVar(master=self, value=serverid) + + self.bottom_frame = Frame(self) + self.input_frame = Frame(self.bottom_frame) + Label(self.input_frame, text="学号:") .grid(row=0, column=0, sticky="e") + Label(self.input_frame, text="servicehall:").grid(row=1, column=0, sticky="e") + Label(self.input_frame, text="serverid:") .grid(row=2, column=0, sticky="e") + Entry(self.input_frame, width=30, textvariable=self.tk_idserial) .grid(row=0, column=1) + Entry(self.input_frame, width=30, textvariable=self.tk_servicehall).grid(row=1, column=1) + Entry(self.input_frame, width=30, textvariable=self.tk_serverid) .grid(row=2, column=1) + self.input_frame.pack(side="left") + + Button(self.bottom_frame, text="run script", command=self.on_button_clicked).pack(side="right") + self.bottom_frame.pack(fill="x",side="bottom", padx=20, pady=20) + + self.plt_canvas = FigureCanvasTkAgg(plt.figure(), master=self) + self.plt_canvas.get_tk_widget().pack(fill="both", side="top", expand=True) + self.plt_toolbar = NavigationToolbar2Tk(self.plt_canvas, self) + self.plt_toolbar.update() + + + def on_button_clicked(self): + idserial = self.tk_idserial .get() + servicehall = self.tk_servicehall.get() + serverid = self.tk_serverid .get() + + with open("config.json", "w", encoding='utf-8') as f: + json.dump({"idserial": idserial, "servicehall": servicehall, "serverid":serverid}, f, indent=4) + print("running script") + process_data(idserial, servicehall, serverid) + self.plt_canvas.draw() + + def decrypt_aes_ecb(encrypted_data: str) -> str: @@ -18,33 +64,25 @@ def decrypt_aes_ecb(encrypted_data: str) -> str: return decrypted_data.decode('utf-8') -idserial = "" -servicehall = "" -all_data = dict() -if __name__ == "__main__": - # 读入账户信息 - try: - with open("config.json", "r", encoding='utf-8') as f: - account = json.load(f) - idserial = account["idserial"] - servicehall = account["servicehall"] - except Exception as e: - print("账户信息读取失败，请重新输入") - idserial = input("请输入学号: ") - servicehall = input("请输入服务代码: ") - with open("config.json", "w", encoding='utf-8') as f: - json.dump({"idserial": idserial, "servicehall": servicehall}, f, indent=4) - - # 发送请求，得到加密后的字符串 +def process_data(idserial, servicehall, serverid): + all_data = dict() url = f"https://card.tsinghua.edu.cn/business/querySelfTradeList?pageNumber=0&pageSize=5000&starttime=2024-01-01&endtime=2024-12-31&idserial={idserial}&tradetype=-1" cookie = { "servicehall": servicehall, + "serverid": serverid, } response = requests.post(url, cookies=cookie) - # 解密字符串 - encrypted_string = json.loads(response.text)["data"] + try: + encrypted_string = json.loads(response.text)["data"] + except json.decoder.JSONDecodeError as e: + if "登录" in response.text: # 如果填入的服务代码无效，就会获得一个在线服务系统的登录页面 + print("登录失败，你需要重新获取服务代码") + return + else: + print("未知错误") + raise e decrypted_string = decrypt_aes_ecb(encrypted_string) # 整理数据 @@ -68,7 +106,7 @@ def decrypt_aes_ecb(encrypted_data: str) -> str: else: plt.rcParams['font.sans-serif'] = ['SimHei'] - plt.figure(figsize=(12, len(all_data) / 66 * 18)) + # plt.figure(figsize=(12, len(all_data) / 66 * 18)) plt.barh(list(all_data.keys()), list(all_data.values())) for index, value in enumerate(list(all_data.values())): plt.text(value + 0.01 * max(all_data.values()), @@ -80,5 +118,26 @@ def decrypt_aes_ecb(encrypted_data: str) -> str: plt.xlim(0, 1.2 * max(all_data.values())) plt.title("华清大学食堂消费情况") plt.xlabel("消费金额（元）") - plt.savefig("result.png") - plt.show() + + +if __name__ == "__main__": + # 读入账户信息 + idserial = ""; servicehall = ""; serverid = "" + try: + with open("config.json", "r", encoding='utf-8') as f: + account = json.load(f) + idserial = account["idserial"]; servicehall = account["servicehall"]; serverid = account["serverid"] + except Exception as _: + print("未正确读取到账户信息") + + # 适配高分屏 + match platform.system(): + case "Windows": + ctypes.windll.shcore.SetProcessDpiAwareness(1) + case "Darwin": # 等待补充 + pass + case "Linux": + pass + # 创建窗口 + window = I_dont_know_what_is_a_good_name_for_this_fucking_gui_class_only_used_once(idserial, servicehall, serverid) + window.mainloop()