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