Update main.yml

.github/workflows/main.yml

@@ -1,126 +1,309 @@
-name: CheckScores
+# 必要的库
+import base64
+import hashlib
+import os
+import sys
+import shutil
+import json
+from pprint import pprint
+from zfn_api import Client
+from pushplus import send_message
 
-on:
+# 从环境变量中提取教务系统的URL、用户名、密码和TOKEN等信息
-  schedule:
+url = os.environ.get("URL")
-    - cron: "*/30 * * * *" # 每30分钟执行一次
+username = os.environ.get("USERNAME")
-  workflow_dispatch: # 允许手动触发workflow
+password = os.environ.get("PASSWORD")
+token = os.environ.get("TOKEN")
+github_event_name = os.environ.get("GITHUB_EVENT_NAME")
+github_triggering_actor = os.environ.get("GITHUB_TRIGGERING_ACTOR")
+repository_name = os.environ.get("REPOSITORY_NAME")
+github_sha = os.environ.get("GITHUB_SHA")
+github_workflow = os.environ.get("GITHUB_WORKFLOW")
+github_run_number = os.environ.get("GITHUB_RUN_NUMBER")
+github_run_id = os.environ.get("GITHUB_RUN_ID")
+beijing_time = os.environ.get("BEIJING_TIME")
 
-jobs:
-  CheckScores:
-    # 在最新版的Ubuntu系统上运行
-    runs-on: ubuntu-latest
 
-    steps:
+# MD5加密
-      # 使用GitHub Actions提供的动作来检出代码库
+def md5_encrypt(string):
-      - name: Checkout Repository
+    return hashlib.md5(string.encode()).hexdigest()
-        uses: actions/checkout@main
 
-      # 配置Python环境
-      - name: Set up Python
-        uses: actions/setup-python@main
-        with:
-          python-version: "*"
 
-      # 使用pip安装项目的依赖项
+# 初始化变量
-      - name: Install dependencies
+cookies = {}
-        run: |
+base_url = url
-          pip install requests rsa pyquery
+raspisanie = []
+ignore_type = []
+detail_category_type = []
+timeout = 5
 
-      # 配置Git用户信息
+# 创建教务系统客户端对象
-      - name: Configure Git
+student_client = Client(
-        run: |
+    cookies=cookies,
-          git config --global user.email "actions@github.com"
+    base_url=base_url,
-          git config --global user.name "GitHub Actions"
+    raspisanie=raspisanie,
+    ignore_type=ignore_type,
+    detail_category_type=detail_category_type,
+    timeout=timeout,
+)
 
-      # 添加上游仓库作为远程仓库，并获取上游仓库的main分支的最新更改
+# 登录
-      - name: Fetch upstream changes
+if not cookies:
-        run: |
+    login_result = student_client.login(username, password)
-          git remote add upstream https://github.com/NianBroken/ZFCheckScores.git
+    if login_result["code"] == 1001:
-          git fetch upstream main
+        # 如果需要验证码,获取验证码并进行登录
+        verify_data = login_result["data"]
+        # 将验证码图片写入文件
+        with open(os.path.abspath("kaptcha.png"), "wb") as pic:
+            pic.write(base64.b64decode(verify_data.pop("kaptcha_pic")))
+        # 输入验证码
+        verify_data["kaptcha"] = input("输入验证码：")
+        # 使用验证码进行登录
+        login_result = student_client.login_with_kaptcha(**verify_data)
 
-      # 创建一个备份目录，并将当前目录中的所有.txt文件复制到备份目录
+        if login_result["code"] != 1000:
-      - name: Backup local .txt files
+            pprint(login_result)
-        run: |
+            sys.exit()
-          mkdir backup
+        pprint(login_result)
-          cp *.txt backup/ || echo "No .txt files to backup"
 
-      # 将上游main分支的更改强制推送到当前main分支
+    elif login_result["code"] != 1000:
-      - name: Force push changes from upstream to current main branch
+        pprint(login_result)
-        run: |
+        sys.exit()
-          git checkout main
-          git reset --hard upstream/main
-          git push origin main --force || (for i in {1..10}; do
-          echo "------" &&
-          echo "你需要手动同步上游分支" &&
-          echo "同步方法如下：" &&
-          echo "1. 打开你的仓库首页：https://github.com/${{ github.repository }}" &&
-          echo "2. 点击 Sync fork"&&
-          echo "3. 点击 Update branch 或 Discard xxx commits";
-          done; exit 1)
 
-      # 从备份目录将.txt文件还原到当前目录，然后删除备份目录
+# 获取个人信息
-      - name: Restore local .txt files and delete backup
+info = student_client.get_info()["data"]
-        run: |
-          cp backup/*.txt . || echo "No .txt files to restore"
-          rm -rf backup
 
-      # 运行主程序main.py
+# 整合个人信息
-      - name: Run main.py
+integrated_info = (
-        env:
+    f"个人信息：\n"
-          URL: ${{ secrets.URL }}
+    f"学号：{info['sid']}\n"
-          USERNAME: ${{ secrets.USERNAME }}
+    f"班级：{info['class_name']}\n"
-          PASSWORD: ${{ secrets.PASSWORD }}
+    f"姓名：{info['name']}"
-          TOKEN: ${{ secrets.TOKEN }}
+)
-          GITHUB_EVENT_NAME: ${{ github.event_name }}
-          GITHUB_TRIGGERING_ACTOR: ${{ github.triggering_actor }}
-          REPOSITORY_NAME: ${{ github.repository }}
-          GITHUB_SHA: ${{ github.sha }}
-          GITHUB_WORKFLOW: ${{ github.workflow }}
-          GITHUB_RUN_NUMBER: ${{ github.run_number }}
-          GITHUB_RUN_ID: ${{ github.run_id }}
-        run: |
-          export BEIJING_TIME=$(TZ='Asia/Shanghai' date +'%Y-%m-%d %H:%M:%S:%3N')
-          python main.py
-          echo "------"
-          # 判断环境变量是否为空
-          if [ -z "$URL" ]; then
-            echo "URL secret is empty!"
-          fi
-          if [ -z "$USERNAME" ]; then
-            echo "USERNAME secret is empty!"
-          fi
-          if [ -z "$PASSWORD" ]; then
-            echo "PASSWORD secret is empty!"
-          fi
-          if [ -z "$TOKEN" ]; then
-            echo "TOKEN secret is empty!"
-          fi
-          # 方便通过截图快速定位到用户
-          echo "Triggered By: ${{ github.triggering_actor }}"
-          echo "Run By: ${{ github.triggering_actor }}"
-          echo "Repository Name: ${{ github.repository }}"
-          echo "Commit SHA: ${{ github.sha }}"
-          echo "Workflow Name: ${{ github.workflow }}"
-          echo "Workflow Number: ${{ github.run_number }}"
-          echo "Workflow ID: ${{ github.run_id }}"
-          echo "Beijing Time: $BEIJING_TIME"
 
-      # 删除__pycache__文件夹
+# 加密个人信息
-      - name: Delete __pycache__ folder
+encrypted_info = md5_encrypt(integrated_info)
-        run: |
-          rm -rf __pycache__
 
-      # 将当前目录中的所有更改添加到Git的暂存区
+# 定义info.txt文件路径
-      - name: Add changes
+info_file_path = "info.txt"
-        run: |
-          git add .
 
-      # 提交更改到Git仓库
+# 初始化运行次数
-      - name: Commit changes
+run_count = 2
-        run: |
-          git commit -m "Update from GitHub Actions" || true
 
-      # 将更改强制推送到main分支
+# 判断info.txt文件是否存在
-      - name: Force push changes to main branch
+if not os.path.exists(info_file_path):
-        run: |
+    # 如果文件不存在,创建并写入encrypted_info的内容
-          git push origin main --force
+    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
+
+# 第一次运行程序则运行两遍,否则运行一遍
+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_data = student_client.get_grade("").get("data", {})
+    grade = grade_data.get("courses", [])
+
+    # 将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())
+
+    # 成绩不为空时则对成绩信息进行处理
+    if grade:
+        # 清空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['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成绩为空\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}"
+
+# 工作流信息
+workflow_info = (
+    f"------\n"
+    f"工作流信息：\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"{workflow_info}"
+)
+
+# 整合首次运行时需要使用到的所有信息
+first_time_run_integrated_send_info = (
+    f"{first_run_text}\n"
+    f"{integrated_send_info}"
+)
+
+# 整合成绩已更新时需要使用到的所有信息
+grades_updated_push_integrated_send_info = (
+    f"教务管理系统成绩已更新\n"
+    f"------\n"
+    f"{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:
+        print("成绩未更新")
+    else:
+        print("成绩已更新")
+
+        # 推送信息
+        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)
+
+# 更新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)