# 导入必要的库
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

# 从环境变量中提取教务系统的URL、用户名、密码和TOKEN
url = os.environ.get("URL")
username = os.environ.get("USERNAME")
password = os.environ.get("PASSWORD")
token = os.environ.get("TOKEN")


# 定义一个封装MD5加密的函数
def md5_encrypt(string):
    # 将字符串编码为字节
    byte_string = string.encode()
    # 创建MD5对象
    md5_object = hashlib.md5(byte_string)
    # 返回MD5哈希值的十六进制表示
    return md5_object.hexdigest()


# 初始化相关变量
cookies = {}
base_url = url
raspisanie = []
ignore_type = []
detail_category_type = []
timeout = 5

# 创建教务系统客户端对象
student_client = 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:
    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:
            pic.write(base64.b64decode(verify_data.pop("kaptcha_pic")))
        # 输入验证码
        verify_data["kaptcha"] = input("输入验证码：")
        # 使用验证码进行登录
        login_result = student_client.login_with_kaptcha(**verify_data)

        if login_result["code"] != 1000:
            pprint(login_result)
            sys.exit()
        pprint(login_result)

    elif login_result["code"] != 1000:
        pprint(login_result)
        sys.exit()

# 获取个人信息
info = student_client.get_info()["data"]

# 整合个人信息
integrated_info = (
    f"个人信息：\n" f"学号：{info['sid']}\n" f"班级：{info['class_name']}\n" f"姓名：{info['name']}\n"
)

# 加密个人信息
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

# 第一次运行程序则运行两遍，否则运行一遍
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.txt文件内容
    with open("grade.txt", "w") as grade_file:
        grade_file.truncate()

    # 获取成绩信息
    grade = student_client.get_grade("")["data"]["courses"]

    # 按照提交时间降序排序
    sorted_grade = sorted(grade, key=lambda x: x["submission_time"], reverse=True)

    # 学分绩点总和
    total_xfjd = sum(float(course["xfjd"]) for course in grade)

    # 学分总和
    total_credit = sum(float(course["credit"]) for course in grade)

    # (学分*绩点)的总和/学分总和
    gpa = "{:.2f}".format(total_xfjd / total_credit)

    integrated_info += f"当前GPA：{gpa}\n------"

    # 初始化输出成绩信息字符串
    integrated_grade_info = "成绩信息："

    # 遍历前8条成绩信息
    for i, course in enumerate(sorted_grade[:8]):
        # 整合成绩信息
        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------"
        )

    # 加密保存成绩
    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)

# 读取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()

# 输出成绩信息
print(f"新成绩：{encrypted_integrated_grade_info}")
print(f"旧成绩：{old_grade_content}")
print("------")

# 整合所有信息
# 注意此处integrated_send_info保存的是未加密的信息，仅用于信息推送
# 若是在 Github Actions 等平台运行，请不要使用print(integrated_send_info)
integrated_send_info = f"{integrated_info}\n{integrated_grade_info}"

# 对grade.txt和old_grade.txt两个文件的内容进行比对，输出成绩是否更新
if grade_content == old_grade_content:
    print("成绩未更新")
else:
    print("成绩已更新")
    response_text = send_message(token, "教务成绩已更新", 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)

# 其他日志
print(f"run_count：{run_count}")