153 lines
4.6 KiB
Python
153 lines
4.6 KiB
Python
import os
|
||
import json
|
||
from loghelper import log
|
||
from ruamel.yaml import YAML
|
||
|
||
# 这个字段现在还没找好塞什么地方好，就先塞config这里了
|
||
serverless = False
|
||
# 提示需要更新config版本
|
||
update_config_need = False
|
||
|
||
config = {
|
||
'enable': True, 'version': 6,
|
||
'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, 'cancel_like_posts': True, 'share_post': True
|
||
},
|
||
'games': {
|
||
'cn': {
|
||
'enable': True,
|
||
'useragent': 'Mozilla/5.0 (Linux; Android 12; Unspecified Device) AppleWebKit/537.36 (KHTML, like Gecko) '
|
||
'Version/4.0 Chrome/103.0.5060.129 Mobile Safari/537.36',
|
||
'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_json = f"{path}/config.json"
|
||
config_Path = f"{path}/config.yaml"
|
||
|
||
|
||
def load_config_json():
|
||
with open(config_Path_json, "r") as f:
|
||
data = json.load(f)
|
||
if data.get('version') == 5:
|
||
config_json = data
|
||
try:
|
||
config_json["mihoyobbs"]["like_post"]
|
||
except KeyError:
|
||
pass
|
||
else:
|
||
config_json["mihoyobbs"]["read_posts"] = config_json["mihoyobbs"]["read_post"]
|
||
config_json["mihoyobbs"]["like_posts"] = config_json["mihoyobbs"]["like_post"]
|
||
del config_json["mihoyobbs"]["like_post"]
|
||
del config_json["mihoyobbs"]["read_post"]
|
||
else:
|
||
log.error("config版本过低，请手动更新到基于yaml版本的新版本配置文件，更新完成后请删除json版的配置文件")
|
||
exit(1)
|
||
log.info("v5Config加载完毕")
|
||
return config_json
|
||
|
||
|
||
def update_config():
|
||
global config
|
||
global update_config_need
|
||
update_config_need = True
|
||
log.info("正在更新config....")
|
||
config_json = load_config_json()
|
||
config['account'] = config_json['account']
|
||
config['mihoyobbs'].update(config_json['mihoyobbs'])
|
||
del config['mihoyobbs']['un_like']
|
||
config['mihoyobbs']['cancel_like_posts'] = config_json['mihoyobbs']['un_like']
|
||
for i in config_json['games']['cn'].keys():
|
||
if i == 'enable':
|
||
continue
|
||
config['games']['cn'][i] = config_json['games']['cn'][i]
|
||
config['games']['os'] = config_json['games']['os']
|
||
save_config()
|
||
log.info('config更新完毕')
|
||
if not serverless:
|
||
os.remove(config_Path_json)
|
||
else:
|
||
log.error("请本地更新一下config")
|
||
|
||
|
||
def load_config():
|
||
yaml = YAML()
|
||
global config
|
||
with open(config_Path, "r", encoding='utf-8') as f:
|
||
config = yaml.load(f)
|
||
log.info("Config加载完毕")
|
||
|
||
|
||
def save_config():
|
||
global serverless
|
||
yaml = YAML()
|
||
if serverless:
|
||
log.info("云函数执行，无法保存")
|
||
return None
|
||
with open(config_Path, "w+") as f:
|
||
try:
|
||
f.seek(0)
|
||
f.truncate()
|
||
yaml.dump(config, f)
|
||
f.flush()
|
||
except OSError:
|
||
serverless = True
|
||
log.info("Cookie保存失败")
|
||
exit(-1)
|
||
else:
|
||
log.info("Config保存完毕")
|
||
|
||
|
||
def clear_cookies():
|
||
global config
|
||
global serverless
|
||
yaml = YAML()
|
||
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"
|
||
try:
|
||
f.seek(0)
|
||
f.truncate()
|
||
yaml.dump(config, f)
|
||
f.flush()
|
||
except OSError:
|
||
serverless = True
|
||
log.info("Cookie删除失败")
|
||
else:
|
||
log.info("Cookie删除完毕")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
# 初始化配置文件
|
||
# try:
|
||
# account_cookie = config['account']['cookie']
|
||
# config = load_config()
|
||
# config['account']['cookie'] = account_cookie
|
||
# except OSError:
|
||
# pass
|
||
# save_config()
|
||
# update_config()
|
||
pass
|