[Feature Request] 用图形界面封装了一下

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()