MihoyoBBSTools/config.py

import os
import json
from loghelper import log

# 这个字段现在还没找好塞什么地方好，就先塞config这里了
serverless = False

config = {
    'enable': True, 'version': 5,
    'account': {
        'cookie': '',
        'login_ticket': '',
        'stuid': '',
        'stoken': ''
    },
    'mihoyobbs': {
        'enable': True, 'checkin': True, 'checkin_multi': True, 'checkin_multi_list': [2, 5],
        'read_posts': True, 'like_posts': True, 'un_like': True, 'share_post': True
    },
    'games': {
        'cn': {
            'enable': True,
            'genshin': {'auto_checkin': True, 'black_list': []},
            'hokai2': {'auto_checkin': False, 'black_list': []},
            'honkai3rd': {'auto_checkin': False, 'black_list': []},
            'tears_of_themis': {'auto_checkin': False, 'black_list': []},
        },
        'os': {
            'enable': False, 'cookie': '',
            'genshin': {'auto_checkin': False, 'black_list': []}
        }
    }
}


path = os.path.dirname(os.path.realpath(__file__)) + "/config"
config_Path = f"{path}/config.json"


def load_v4(data: dict):
    global config
    # 配置开关
    config["enable"] = data["enable_Config"]
    # 账号 cookie
    config["account"]["login_ticket"] = data["mihoyobbs_Login_ticket"]
    config["account"]["stuid"] = data["mihoyobbs_Stuid"]
    config["account"]["stoken"] = data["mihoyobbs_Stoken"]
    config["account"]["cookie"] = data["mihoyobbs_Cookies"]
    # bbs 相关设置(自己之前造的孽)
    config["mihoyobbs"]["enable"] = data["mihoyobbs"]["bbs_Global"]
    config["mihoyobbs"]["checkin"] = data["mihoyobbs"]["bbs_Signin"]
    config["mihoyobbs"]["checkin_multi"] = data["mihoyobbs"]["bbs_Signin_multi"]
    config["mihoyobbs"]["checkin_multi_list"] = data["mihoyobbs"]["bbs_Signin_multi_list"]
    config["mihoyobbs"]["read_posts"] = data["mihoyobbs"]["bbs_Read_posts"]
    config["mihoyobbs"]["like_posts"] = data["mihoyobbs"]["bbs_Like_posts"]
    config["mihoyobbs"]["un_like"] = data["mihoyobbs"]["bbs_Unlike"]
    config["mihoyobbs"]["share_post"] = data["mihoyobbs"]["bbs_Share"]
    # 游戏相关设置 v4只支持原神和崩坏3，所以其他选项默认关闭
    config["games"]["cn"]["genshin"]["auto_checkin"] = data["genshin_Auto_sign"]
    config["games"]["cn"]["honkai3rd"]["auto_checkin"] = data["honkai3rd_Auto_sign"]


def load_config():
    global config
    with open(config_Path, "r") as f:
        data = json.load(f)
        if data.get('version') == 5:
            config = data
            try:
                config["mihoyobbs"]["like_post"]
            except KeyError:
                pass
            else:
                config["mihoyobbs"]["read_posts"] = config["mihoyobbs"]["read_post"]
                config["mihoyobbs"]["like_posts"] = config["mihoyobbs"]["like_post"]
                del config["mihoyobbs"]["like_post"]
                del config["mihoyobbs"]["read_post"]
                save_config()
        else:
            load_v4(data)
            log.info("升级v5 config")
            # 直接升级到v5 config
            save_config()
        f.close()
        log.info("Config加载完毕")


def save_config():
    global serverless
    if serverless:
        log.info("云函数执行，无法保存")
        return None
    with open(config_Path, "r+") as f:
        temp_text = json.dumps(config, sort_keys=False, indent=4, separators=(', ', ': '))
        try:
            f.seek(0)
            f.truncate()
            f.write(temp_text)
            f.flush()
        except OSError:
            serverless = True
            log.info("Cookie保存失败")
            exit(-1)
        else:
            log.info("Config保存完毕")
        f.close()


def clear_cookies():
    global config
    global serverless
    if serverless:
        log.info("云函数执行，无法保存")
        return None
    with open(config_Path, "r+") as f:
        config["enable"] = False
        config["account"]["login_ticket"] = ""
        config["account"]["stuid"] = ""
        config["account"]["stoken"] = ""
        config["account"]["cookie"] = "CookieError"
        temp_text = json.dumps(config, sort_keys=False, indent=4, separators=(', ', ': '))
        try:
            f.seek(0)
            f.truncate()
            f.write(temp_text)
            f.flush()
        except OSError:
            serverless = True
            log.info("Cookie删除失败")
        else:
            log.info("Cookie删除完毕")
        f.close()

if __name__ == "__main__":
    # 初始化配置文件
    # try:
    #     account_cookie = config['account']['cookie']
    #     config = load_config()
    #     config['account']['cookie'] = account_cookie
    # except OSError:
    #     pass
    # save_config()
    pass