133 lines
4.4 KiB
Python
133 lines
4.4 KiB
Python
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()
|