Update main.py

main.py

@@ -27,6 +27,8 @@ beijing_time = os.environ.get("BEIJING_TIME")
 # 将字符串转换为布尔值
 if force_push_message:
     force_push_message = eval(force_push_message)
+else:
+    force_push_message = ""
 
 
 # MD5加密
@@ -320,6 +322,319 @@ grades_updated_push_integrated_send_info = (
 )
 
 
+# 如果是第一次运行,则提示程序运行成功
+if run_count == 2:
+    print(first_run_text)
+
+    # 推送信息
+    first_run_text_response_text = send_message(
+        token,
+        "正方教务管理系统成绩推送",
+        first_time_run_integrated_send_info,
+    )
+
+    # 解析 JSON 数据
+    first_run_text_response_dict = json.loads(first_run_text_response_text)
+
+    # 删除 "data" 字段
+    if "data" in first_run_text_response_dict:
+        first_run_text_response_dict.pop("data")
+
+    # 输出响应内容
+    print(first_run_text_response_dict)
+else:
+    # 如果非第一次运行,则输出成绩信息
+    if grade:
+        print(f"新成绩：{encrypted_integrated_grade_info}")
+        print(f"旧成绩：{old_grade_content}")
+    else:
+        print("成绩为空")
+    print("------")
+
+    # 对grade.txt和old_grade.txt两个文件的内容进行比对,输出成绩是否更新
+    if grade_content != old_grade_content or force_push_message:
+        # 判断是否选中了强制推送信息
+        print("强制推送信息" if force_push_message else "成绩已更新")
+
+        # 推送信息
+        response_text = send_message(
+            token,
+            "正方教务管理系统成绩推送",
+            grades_updated_push_integrated_send_info,
+        )
+
+        # 解析 JSON 数据
+        response_dict = json.loads(response_text)
+
+        # 删除 "data" 字段
+        if "data" in response_dict:
+            response_dict.pop("data")
+
+        # 输出响应内容
+        print(response_dict)
+    else:
+        print("成绩未更新")
+
+# 更新info.txt
+with open(info_file_path, "r") as info_file:
+    info_file_content = info_file.read()
+    if info_file_content != encrypted_info:
+        with open("info.txt", "w") as info_file:
+            info_file.write(encrypted_info)
+
+# 删除 __pycache__ 缓存目录及其内容
+current_directory = os.getcwd()
+cache_folder = os.path.join(current_directory, "__pycache__")
+# 检查目录是否存在
+if os.path.exists(cache_folder):
+    # 删除目录及其内容
+    shutil.rmtree(cache_folder)        sys.exit()
+
+# 获取个人信息
+info = student_client.get_info()["data"]
+
+# 整合个人信息
+integrated_info = (
+    f"个人信息：\n"
+    f"学号：{info['sid']}\n"
+    f"班级：{info['class_name']}\n"
+    f"姓名：{info['name']}"
+)
+
+# 加密个人信息
+encrypted_info = md5_encrypt(integrated_info)
+
+# 定义info.txt文件路径
+info_file_path = "info.txt"
+
+# 初始化运行次数
+run_count = 2
+
+# 判断info.txt文件是否存在
+if not os.path.exists(info_file_path):
+    # 如果文件不存在,创建并写入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文件中保存的个人信息与获取到的个人信息一致,则代表非第一次运行程序
+        if info_file_content == encrypted_info:
+            # 非第一次运行程序
+            run_count = 1
+
+
+# 获取已选课程信息
+selected_courses_data = student_client.get_selected_courses().get("data", {})
+selected_courses = selected_courses_data.get("courses", [])
+
+# 第一次运行程序则运行两遍,否则运行一遍
+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") as old_grade_file:
+        old_grade_file.truncate()
+
+    # 将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_data = student_client.get_grade("").get("data", {})
+    grade = grade_data.get("courses", [])
+
+    # 成绩不为空时则对成绩信息进行处理
+    if grade:
+        # 遍历 grade 中的每个字典，将 title 中的中文括号替换为英文括号
+        for course_data_grade in grade:
+            course_data_grade["title"] = (
+                course_data_grade["title"].replace("（", "(").replace("）", ")")
+            )
+
+        # 清空grade.txt文件内容
+        with open("grade.txt", "w") as grade_file:
+            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_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
+        )
+
+        # GPA计算 (学分*绩点)的总和/学分总和
+        gpa = "{:.2f}".format(total_xfjd / total_credit)
+
+        # 百分制GPA计算 (百分制成绩*学分)的总和/学分总和
+        percentage_gpa = "{:.2f}".format(
+            sum_of_percentage_grades_multiplied_by_credits / total_credit
+        )
+
+        # 初始化输出成绩信息字符串
+        integrated_grade_info = "成绩信息："
+
+        # 遍历前8条成绩信息
+        for i, course in enumerate(sorted_grade[:8]):
+            # 整合成绩信息
+            integrated_grade_info += (
+                f"\n"
+                f"教学班ID：{course['class_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成绩为空\n------"
+
+    # 加密保存成绩
+    encrypted_integrated_grade_info = md5_encrypt(integrated_grade_info)
+
+    # 将加密后的成绩信息写入grade.txt文件
+    with open("grade.txt", "w") as grade_file:
+        grade_file.write(encrypted_integrated_grade_info)
+
+
+# 成绩信息不为空时整合GPA信息
+if grade:
+    # 整合个人信息
+    integrated_info += (
+        f"\n当前GPA：{gpa}\n"
+        f"当前百分制GPA：{percentage_gpa}\n"
+        f"------"
+    )
+
+# 读取grade.txt和old_grade.txt文件的内容
+with open("grade.txt", "r") as grade_file, open("old_grade.txt", "r") as old_grade_file:
+    grade_content = grade_file.read()
+    old_grade_content = old_grade_file.read()
+
+# 第一次运行时的提示文本
+first_run_text = (
+    "你的程序运行成功\n"
+    "从现在开始,程序将会每隔 30 分钟自动检测成绩是否有更新\n"
+    "若有更新,将通过微信推送及时通知你\n"
+    "------"
+)
+
+# 整合MD5值
+integrated_grade_info += f"\n" f"MD5：{encrypted_integrated_grade_info}"
+
+# 已选课程信息不为空时,处理未公布成绩的课程和异常课程
+if selected_courses:
+    # 初始化空字典用于存储未公布成绩的课程,按学年学期分组
+    ungraded_courses_by_semester = {}
+    # 初始化空字典用于存储异常的课程,按学年学期分组
+    abnormal_courses_by_semester = {}
+
+    # 获取成绩列表中的class_id集合
+    grade_class_ids = {course["class_id"] for course in grade}
+
+    # 初始化输出内容
+    selected_courses_filtering = ""
+
+    # 遍历selected_courses和grade中的每个课程
+    for course in selected_courses + grade:
+        # 获取课程的class_id和学年学期
+        yearsemester_id = course["class_name"].split("(")[1].split(")")[0]
+        year, semester, seq = yearsemester_id.split("-")
+
+        # 构建年学期名称,例如 "a至b学年第c学期"
+        yearsemester_name = f"{year}至{semester}学年第{seq}学期"
+
+        # 判断课程是否未公布成绩或为异常课程
+        if course["class_id"] not in grade_class_ids:
+            # 未公布成绩
+            ungraded_courses_by_semester.setdefault(yearsemester_name, []).append(
+                f"{course['title'].replace('（', '(').replace('）', ')')} - {course['teacher']}"
+            )
+        elif course["class_id"] not in {
+            course["class_id"] for course in selected_courses
+        }:
+            # 异常课程
+            abnormal_courses_by_semester.setdefault(yearsemester_name, []).append(
+                f"{course['title'].replace('（', '(').replace('）', ')')} - {course['teacher']}"
+            )
+
+    # 构建输出内容
+    if ungraded_courses_by_semester:
+        # 存在未公布成绩的课程
+        selected_courses_filtering += "------\n未公布成绩的课程："
+        for i, (semester, courses) in enumerate(ungraded_courses_by_semester.items()):
+            if i > 0:
+                selected_courses_filtering += "\n------"
+            selected_courses_filtering += f"\n{semester}："
+            for course in courses:
+                selected_courses_filtering += f"\n{course}"
+
+    if abnormal_courses_by_semester:
+        # 存在异常的课程
+        if ungraded_courses_by_semester:
+            # 如果存在课程,添加分隔线
+            selected_courses_filtering += "\n"
+        selected_courses_filtering += "------\n异常的课程："
+        for i, (semester, courses) in enumerate(abnormal_courses_by_semester.items()):
+            if i > 0:
+                selected_courses_filtering += "\n------"
+            selected_courses_filtering += f"\n{semester}："
+            for course in courses:
+                selected_courses_filtering += f"\n{course}"
+else:
+    selected_courses_filtering = "------\n已选课程信息为空"
+
+# 工作流信息
+workflow_info = (
+    f"------\n"
+    f"工作流信息：\n"
+    f"Force Push Message：{force_push_message}\n"
+    f"Triggered By：{github_event_name}\n"
+    f"Run By：{github_triggering_actor}\n"
+    f"Repository Name：{repository_name}\n"
+    f"Commit SHA：{github_sha}\n"
+    f"Workflow Name：{github_workflow}\n"
+    f"Workflow Number：{github_run_number}\n"
+    f"Workflow ID：{github_run_id}\n"
+    f"Beijing Time：{beijing_time}"
+)
+
+# 整合所有信息
+# 注意此处integrated_send_info保存的是未加密的信息,仅用于信息推送
+# 若是在 Github Actions 等平台运行,请不要使用print(integrated_send_info)
+integrated_send_info = (
+    f"{integrated_info}\n"
+    f"{integrated_grade_info}\n"
+    f"{selected_courses_filtering}\n"
+    f"{workflow_info}"
+)
+
+# 整合首次运行时需要使用到的所有信息
+first_time_run_integrated_send_info = f"{first_run_text}\n" f"{integrated_send_info}"
+
+# 整合成绩已更新时需要使用到的所有信息
+grades_updated_push_integrated_send_info = (
+    f"{'强制推送信息成功' if force_push_message else '教务管理系统成绩已更新'}\n"
+    f"------\n"
+    f"{integrated_send_info}"
+)
+
+
 # 如果是第一次运行,则提示程序运行成功
 if run_count == 2:
     print(first_run_text)