Update main.py

main.py

@@ -45,11 +45,11 @@ student_client = Client(
     timeout=timeout,
 )
 
-# 如果cookies为空字典，则进行登录
+# 如果cookies为空字典,则进行登录
 if not cookies:
     login_result = student_client.login(username, password)
     if login_result["code"] == 1001:
-        # 如果需要验证码，获取验证码并进行登录
+        # 如果需要验证码,获取验证码并进行登录
         verify_data = login_result["data"]
         # 将验证码图片写入文件
         with open(os.path.abspath("kaptcha.png"), "wb") as pic:
@@ -90,21 +90,21 @@ run_count = 2
 
 # 判断info.txt文件是否存在
 if not os.path.exists(info_file_path):
-    # 如果文件不存在，创建并写入encrypted_info的内容
+    # 如果文件不存在,创建并写入encrypted_info的内容
     with open(info_file_path, "w") as info_file:
         info_file.write(encrypted_info)
 else:
-    # 如果文件存在，读取文件内容并比较
+    # 如果文件存在,读取文件内容并比较
     with open(info_file_path, "r") as info_file:
         info_file_content = info_file.read()
-        # 如果info.txt文件中保存的个人信息与获取到的个人信息不一致，则代表是第一次运行程序
+        # 若info.txt文件中保存的个人信息与获取到的个人信息一致,则代表非第一次运行程序
         if info_file_content == encrypted_info:
             # 非第一次运行程序
             run_count = 1
 
-# 第一次运行程序则运行两遍，否则运行一遍
+# 第一次运行程序则运行两遍,否则运行一遍
 for _ in range(run_count):
-    # 如果grade.txt文件不存在，则创建文件
+    # 如果grade.txt文件不存在,则创建文件
     if not os.path.exists("grade.txt"):
         open("grade.txt", "w").close()
 
@@ -113,71 +113,69 @@ for _ in range(run_count):
         old_grade_file.truncate()
 
     # 获取成绩信息
-    grade_data = student_client.get_grade("").get("data", {})
+    grade_data = student_client.get_grade("2024").get("data", {})
     grade = grade_data.get("courses", [])
 
-    # 成绩为空时退出程序
+    # 成绩不为空时则对成绩信息进行处理
-    if not grade:
+    if grade:
-        print("成绩为空")
+        # 将grade.txt文件中的内容写入old_grade.txt文件内
-        with open("info.txt", "w") as info_file:
+        with open("grade.txt", "r") as grade_file, open(
-            # 将加密后的个人信息写入info.txt文件
+            "old_grade.txt", "w"
-            info_file.write(encrypted_info)
+        ) as old_grade_file:
-        sys.exit()
+            old_grade_file.write(grade_file.read())
 
-    # 将grade.txt文件中的内容写入old_grade.txt文件内
+        # 清空grade.txt文件内容
-    with open("grade.txt", "r") as grade_file, open(
+        with open("grade.txt", "w") as grade_file:
-        "old_grade.txt", "w"
+            grade_file.truncate()
-    ) as old_grade_file:
-        old_grade_file.write(grade_file.read())
 
-    # 清空grade.txt文件内容
+        # 按照提交时间降序排序
-    with open("grade.txt", "w") as grade_file:
+        sorted_grade = sorted(grade, key=lambda x: x["submission_time"], reverse=True)
-        grade_file.truncate()
 
-    # 按照提交时间降序排序
+        # 学分总和
-    sorted_grade = sorted(grade, key=lambda x: x["submission_time"], reverse=True)
+        total_credit = sum(float(course["credit"]) for course in grade)
 
-    # 学分总和
+        # 学分绩点总和
-    total_credit = sum(float(course["credit"]) for course in grade)
+        total_xfjd = sum(float(course["xfjd"]) for course in grade)
 
-    # 学分绩点总和
+        # (百分制成绩*学分)的总和
-    total_xfjd = sum(float(course["xfjd"]) for course in grade)
+        sum_of_percentage_grades_multiplied_by_credits = sum(
-
+            float(course["percentage_grades"]) * float(course["credit"])
-    # (百分制成绩*学分)的总和
+            for course in grade
-    sum_of_percentage_grades_multiplied_by_credits = sum(
-        float(course["percentage_grades"]) * float(course["credit"]) for course in grade
-    )
-
-    # GPA计算 (学分*绩点)的总和/学分总和
-    gpa = "{:.2f}".format(total_xfjd / total_credit)
-
-    # 百分制GPA计算 (百分制成绩*学分)的总和/学分总和
-    percentage_gpa = "{:.2f}".format(
-        sum_of_percentage_grades_multiplied_by_credits / total_credit
-    )
-
-    # 整合个人信息
-    integrated_info += (
-        f"当前GPA：{gpa}\n" f"当前百分制GPA：{percentage_gpa}\n" f"------"
-    )
-
-    # 初始化输出成绩信息字符串
-    integrated_grade_info = "成绩信息："
-
-    # 遍历前8条成绩信息
-    for i, course in enumerate(sorted_grade[:8]):
-        # 整合成绩信息
-        integrated_grade_info += (
-            f"\n"
-            f"课程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"------"
         )
 
+        # GPA计算 (学分*绩点)的总和/学分总和
+        gpa = "{:.2f}".format(total_xfjd / total_credit)
+
+        # 百分制GPA计算 (百分制成绩*学分)的总和/学分总和
+        percentage_gpa = "{:.2f}".format(
+            sum_of_percentage_grades_multiplied_by_credits / total_credit
+        )
+
+        # 整合个人信息
+        integrated_info += (
+            f"当前GPA：{gpa}\n" f"当前百分制GPA：{percentage_gpa}\n" f"------"
+        )
+
+        # 初始化输出成绩信息字符串
+        integrated_grade_info = "成绩信息："
+
+        # 遍历前8条成绩信息
+        for i, course in enumerate(sorted_grade[:8]):
+            # 整合成绩信息
+            integrated_grade_info += (
+                f"\n"
+                f"课程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"------"
+            )
+    else:
+        # 成绩为空时将成绩信息定义为"成绩为空"
+        integrated_grade_info = "成绩为空\n------"
+
     # 加密保存成绩
     encrypted_integrated_grade_info = md5_encrypt(integrated_grade_info)
 
@@ -193,8 +191,8 @@ with open("grade.txt", "r") as grade_file, open("old_grade.txt", "r") as old_gra
 # 第一次运行时的提示文本
 first_run_text = (
     "你的程序运行成功\n"
-    "从现在开始，程序将会每隔 30 分钟自动检测成绩是否有更新\n"
+    "从现在开始,程序将会每隔 30 分钟自动检测成绩是否有更新\n"
-    "若有更新，将通过微信推送及时通知你\n"
+    "若有更新,将通过微信推送及时通知你\n"
     "------"
 )
 
@@ -212,13 +210,8 @@ workflow_info = (
     f"Beijing Time：{beijing_time}"
 )
 
-# 如果是第一次运行 则提示程序运行成功
+# 如果是第一次运行,则提示程序运行成功
-# 如果非第一次运行 则输出成绩信息
+if run_count == 2:
-if run_count == 1:
-    print(f"新成绩：{encrypted_integrated_grade_info}")
-    print(f"旧成绩：{old_grade_content}")
-    print("------")
-else:
     print(first_run_text)
 
     # 推送信息
@@ -235,16 +228,23 @@ else:
 
     # 输出响应内容
     print(first_run_text_response_dict)
+else:
+    # 如果非第一次运行,则输出成绩信息
+    if grade:
+        print(f"新成绩：{encrypted_integrated_grade_info}")
+        print(f"旧成绩：{old_grade_content}")
+    else:
+        print("成绩为空")
+    print("------")
 
 # 整合所有信息
-# 注意此处integrated_send_info保存的是未加密的信息，仅用于信息推送
+# 注意此处integrated_send_info保存的是未加密的信息,仅用于信息推送
-# 若是在 Github Actions 等平台运行，请不要使用print(integrated_send_info)
+# 若是在 Github Actions 等平台运行,请不要使用print(integrated_send_info)
 integrated_send_info = f"{integrated_info}\n{integrated_grade_info}\n{workflow_info}"
 
-# 对grade.txt和old_grade.txt两个文件的内容进行比对，输出成绩是否更新
+# 对grade.txt和old_grade.txt两个文件的内容进行比对,输出成绩是否更新
 if grade_content == old_grade_content:
-    if run_count == 1:
+    print("成绩未更新")
-        print("成绩未更新")
 else:
     print("成绩已更新")
 
@@ -261,6 +261,7 @@ else:
     # 输出响应内容
     print(response_dict)
 
+
 # 更新info.txt
 with open(info_file_path, "r") as info_file:
     info_file_content = info_file.read()