Optimize some code

scripts/user_login.py

@@ -1,7 +1,62 @@
+import os
+import re
 import sys
 from pprint import pprint
 from .zfn_api import Client
 
+# 从环境变量中提取教务系统的URL、用户名、密码和TOKEN等信息
+force_push_message = os.environ.get("FORCE_PUSH_MESSAGE")
+github_actions = os.environ.get("GITHUB_ACTIONS")
+github_ref_name = os.environ.get("GITHUB_REF_NAME")
+github_event_name = os.environ.get("GITHUB_EVENT_NAME")
+github_actor = os.environ.get("GITHUB_ACTOR")
+github_actor_id = os.environ.get("GITHUB_ACTOR_ID")
+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")
+github_step_summary = os.environ.get("GITHUB_STEP_SUMMARY")
+
+# 工作流信息
+workflow_info = (
+    f"------\n"
+    f"工作流信息：\n"
+    f"Force Push Message：{force_push_message}\n"
+    f"Branch Name：{github_ref_name}\n"
+    f"Triggered By：{github_event_name}\n"
+    f"Initial Run By：{github_actor}\n"
+    f"Initial Run By ID：{github_actor_id}\n"
+    f"Initiated 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}"
+)
+
+copyright_text = "Copyright © 2024 NianBroken. All rights reserved."
+
+
+def write_github_summary(run_log, lgn_code):
+    summary_log = (
+        f"# 正方教务管理系统成绩推送\n"
+        f"你因 **{run_log}** 原因而登录失败，错误代码为 **{lgn_code}**。\n"
+        f"若你不明白或不理解为什么登录失败，请到上游仓库的 "
+        f"[Issue](https://github.com/NianBroken/ZFCheckScores/issues/new 'Issue') 中寻求帮助。\n"
+        f"{workflow_info}\n"
+        f"{copyright_text}"
+    )
+    # 将任意个数的换行替换为两个换行
+    summary_log = re.sub("\n+", "\n\n", summary_log)
+
+    # 将登录需要验证码写入到 GitHub Actions 的环境文件中
+    with open(github_step_summary, "w", encoding="utf-8") as file:
+        file.write(summary_log)
+
 
 def login(url, username, password):
     cookies = {}
@@ -23,7 +78,13 @@ def login(url, username, password):
     if cookies == {}:
         lgn = student_client.login(username, password)
         if lgn["code"] == 1001:
-            sys.exit("你的教务系统登录时需要验证码，因此你无法使用此项目。")
+            run_log = "登录需要验证码"
+
+            # 如果是Github Actions运行,则将运行日志写入到GitHub Actions的日志文件中
+            if github_actions:
+                write_github_summary(run_log, lgn["code"])
+
+            sys.exit(f"你因{run_log}原因而登录失败，错误代码为{lgn['code']}。")
             """
             verify_data = lgn["data"]
             with open(os.path.abspath("kaptcha.png"), "wb") as pic:
@@ -37,6 +98,12 @@ def login(url, username, password):
             """
         elif lgn["code"] != 1000:
             pprint(lgn)
-            sys.exit("登录失败")
+            run_log = lgn["msg"]
+
+            # 如果是Github Actions运行,则将运行日志写入到GitHub Actions的日志文件中
+            if github_actions:
+                write_github_summary(run_log, lgn["code"])
+
+            sys.exit(1)
 
     return student_client