add

main.py

@@ -0,0 +1,143 @@
+import base64
+import hashlib
+import os
+import sys
+from pprint import pprint
+from zfn_api import Client
+from pushplus import send_message
+
+# 教务系统的URL、用户名和密码
+url = os.environ.get("URL")
+username = os.environ.get("USERNAME")
+password = os.environ.get("PASSWORD")
+
+
+# 定义一个md5加密的封装函数
+def md5_encrypt(string):
+    # 将字符串转换为字节
+    byte_string = string.encode()
+    # 创建一个MD5对象
+    md5_object = hashlib.md5(byte_string)
+    # 返回MD5加密后的十六进制字符串
+    return md5_object.hexdigest()
+
+
+# 初始化相关变量
+cookies = {}  # 初始化cookies为空字典
+base_url = url
+raspisanie = []
+ignore_type = []
+detail_category_type = []
+timeout = 5
+stu = Client(
+    cookies=cookies,
+    base_url=base_url,
+    raspisanie=raspisanie,
+    ignore_type=ignore_type,
+    detail_category_type=detail_category_type,
+    timeout=timeout,
+)
+
+# 如果cookies为空字典，则进行登录
+if not cookies:
+    lgn = stu.login(username, password)
+    if lgn["code"] == 1001:
+        verify_data = lgn["data"]
+        # 将验证码保存为图片文件
+        with open(os.path.abspath("kaptcha.png"), "wb") as pic:
+            pic.write(base64.b64decode(verify_data.pop("kaptcha_pic")))
+        verify_data["kaptcha"] = input("输入验证码：")
+        ret = stu.login_with_kaptcha(**verify_data)
+        if ret["code"] != 1000:
+            pprint(ret)
+            sys.exit()
+        pprint(ret)
+    elif lgn["code"] != 1000:
+        pprint(lgn)
+        sys.exit()
+
+# 获取个人信息
+info = stu.get_info()["data"]
+# 整合个人信息
+integrate_info = (
+    f"个人信息：\n" f"学号：{info['sid']}\n" f"班级：{info['class_name']}\n" f"姓名：{info['name']}"
+)
+
+# 获取成绩信息
+firstrun_file_path = "firstrun.txt"
+
+# 如果firstrun.txt文件不存在，创建并写入true
+if not os.path.exists(firstrun_file_path):
+    with open(firstrun_file_path, "w") as firstrun_file:
+        firstrun_file.write("true")
+
+# 第一次运行则运行两遍，否则运行一遍
+run_count = 2 if open(firstrun_file_path).read() == "true" else 1
+for _ in range(run_count):
+    # 如果grade.txt文件不存在，创建文件
+    if not os.path.exists("grade.txt"):
+        open("grade.txt", "w").close()
+
+    # 清空old_grade.txt文件内容
+    with open("old_grade.txt", "w"):
+        pass
+
+    # 将grade.txt文件中的内容写入到old_grade.txt文件内。
+    with open("grade.txt", "r") as grade_file, open(
+        "old_grade.txt", "w"
+    ) as old_grade_file:
+        old_grade_file.write(grade_file.read())
+
+    # 清空grade.txt文件内容
+    with open("grade.txt", "w"):
+        pass
+
+    # 将下列代码输出的所有内容都写入到grade.txt文件内。
+    integrated_grade_info = "成绩信息："
+    grade = stu.get_grade(2023)["data"]["courses"]
+    for course in grade:
+        # 整合成绩信息
+        integrated_grade_info += (
+            f"\n课程ID: {course['course_id']}\n"
+            f"课程名称: {course['title']}\n"
+            f"任课教师: {course['teacher']}\n"
+            f"成绩: {course['grade']}\n"
+            f"提交时间: {course['submission_time']}\n"
+            f"提交人姓名: {course['name_of_submitter']}\n"
+            f"------"
+        )
+
+    # 加密保存成绩
+    encrypt_integrated_grade_info = md5_encrypt(integrated_grade_info)
+
+    # 将加密后的成绩信息写入grade.txt文件
+    with open("grade.txt", "w") as grade_file:
+        grade_file.write(encrypt_integrated_grade_info)
+
+# 对grade.txt和old_grade.txt两个文件的内容进行比对
+with open("grade.txt", "r") as grade_file, open("old_grade.txt", "r") as old_grade_file:
+    # 读取grade.txt和old_grade.txt文件的内容
+    grade_content = grade_file.read()
+    old_grade_content = old_grade_file.read()
+
+# 输出成绩信息
+print(f"新成绩：{encrypt_integrated_grade_info}")
+print(f"旧成绩：{old_grade_content}")
+print("------")
+
+# 整合所有信息
+integrate_send_info = f"{integrate_info}\n------\n{integrated_grade_info}"
+
+# 输出成绩是否更新
+if grade_content == old_grade_content:
+    print("成绩未更新")
+else:
+    print("成绩已更新")
+    response_text = send_message("教务成绩已更新", integrate_send_info)
+    print(response_text)
+
+# 更新firstrun.txt文件内容为false
+if open(firstrun_file_path).read() == "true":
+    with open(firstrun_file_path, "w") as firstrun_file:
+        # 将firstrun.txt文件内容更新为false
+        firstrun_file.write("false")