|
|
@ -2,6 +2,7 @@ import json |
|
|
|
import os |
|
|
|
import os |
|
|
|
import random |
|
|
|
import random |
|
|
|
import re |
|
|
|
import re |
|
|
|
|
|
|
|
import sqlite3 |
|
|
|
import time |
|
|
|
import time |
|
|
|
import traceback |
|
|
|
import traceback |
|
|
|
# import tomli_w |
|
|
|
# import tomli_w |
|
|
@ -32,15 +33,19 @@ def toml_to_json(toml_path, to_json_path): |
|
|
|
# tomli_w.dump(json_dict, f) |
|
|
|
# tomli_w.dump(json_dict, f) |
|
|
|
|
|
|
|
|
|
|
|
class config_get(object): |
|
|
|
class config_get(object): |
|
|
|
def __init__(self, config_path=None): |
|
|
|
def __init__(self, custom_path=None): |
|
|
|
""" |
|
|
|
""" |
|
|
|
config_path: 自定义配置文件路径 |
|
|
|
config_path: 自定义配置文件路径 |
|
|
|
config_file: 实际使用的配置文件路径 |
|
|
|
config_file: 实际使用的配置文件路径 |
|
|
|
config_format: 实际使用的配置文件格式 |
|
|
|
config_format: 实际使用的配置文件格式 |
|
|
|
""" |
|
|
|
""" |
|
|
|
self.config_path = config_path |
|
|
|
if custom_path is None: |
|
|
|
self.config_file = self.get_config_file() |
|
|
|
self.config_path = self.get_config_path() |
|
|
|
self.config_format = self.get_config_format() |
|
|
|
self.config_file = self.get_config_file() |
|
|
|
|
|
|
|
self.config_format = self.get_config_format() |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
self.config_file = custom_path |
|
|
|
|
|
|
|
self.config_format = self.get_config_format() |
|
|
|
|
|
|
|
|
|
|
|
def get_config_format(self): |
|
|
|
def get_config_format(self): |
|
|
|
if self.config_file.endswith('.toml'): |
|
|
|
if self.config_file.endswith('.toml'): |
|
|
@ -48,30 +53,40 @@ class config_get(object): |
|
|
|
else: |
|
|
|
else: |
|
|
|
return "json" |
|
|
|
return "json" |
|
|
|
|
|
|
|
|
|
|
|
def get_config_file(self): |
|
|
|
@staticmethod |
|
|
|
ql_new = '/ql/config/env.sh' |
|
|
|
def get_config_path(): |
|
|
|
json_config_file = '/ql/config/check.json' |
|
|
|
ql_old = "/ql/config/" |
|
|
|
toml_config_file = '/ql/config/check.toml' |
|
|
|
ql_new = "/ql/data/config/" |
|
|
|
print('开始检查环境\n') |
|
|
|
if os.path.isdir(ql_old): |
|
|
|
if os.path.exists(ql_new): |
|
|
|
print('成功 当前环境为青龙面板v2.12- 继续执行\n') |
|
|
|
print('成功 当前环境为青龙面板继续执行\n') |
|
|
|
return ql_old |
|
|
|
if self.config_path is not None: |
|
|
|
elif os.path.isdir(ql_new): |
|
|
|
print("使用了自定义路径的配置文件") |
|
|
|
print('成功 当前环境为青龙面板v2.12+ 继续执行\n') |
|
|
|
return self.config_path |
|
|
|
return ql_new |
|
|
|
elif os.path.exists(toml_config_file): |
|
|
|
|
|
|
|
print("未使用自定义配置文件,开始从ql/config中检测") |
|
|
|
|
|
|
|
print("检测到toml格式配置文件\n") |
|
|
|
|
|
|
|
return toml_config_file |
|
|
|
|
|
|
|
elif os.path.exists(json_config_file): |
|
|
|
|
|
|
|
print("检测到json格式配置文件\n") |
|
|
|
|
|
|
|
return json_config_file |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
print("未检测到配置文件,程序退出") |
|
|
|
|
|
|
|
exit(1) |
|
|
|
|
|
|
|
else: |
|
|
|
else: |
|
|
|
print('失败 请检查环境') |
|
|
|
print('失败 请检查环境') |
|
|
|
exit(0) |
|
|
|
exit(0) |
|
|
|
return 0 |
|
|
|
|
|
|
|
|
|
|
|
def get_config_file(self): |
|
|
|
|
|
|
|
toml_file = f"{self.config_path}check.toml" |
|
|
|
|
|
|
|
json_file = f"{self.config_path}check.json" |
|
|
|
|
|
|
|
if os.path.exists(toml_file): |
|
|
|
|
|
|
|
print(f"启用了toml配置文件\n路径为{toml_file}\n") |
|
|
|
|
|
|
|
return toml_file |
|
|
|
|
|
|
|
elif os.path.exists(json_file): |
|
|
|
|
|
|
|
print(f"启用了json配置文件\n路径为{json_file}\n") |
|
|
|
|
|
|
|
return json_file |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
print("未找到配置文件") |
|
|
|
|
|
|
|
self.move_config_file() |
|
|
|
|
|
|
|
return toml_file |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def move_config_file(self): |
|
|
|
|
|
|
|
print("尝试移动配置文件到目录") |
|
|
|
|
|
|
|
if self.config_path == "/ql/config/": |
|
|
|
|
|
|
|
self.move_configuration_file_old() |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
self.move_configuration_file_new() |
|
|
|
|
|
|
|
|
|
|
|
def get_real_key(self, expression): |
|
|
|
def get_real_key(self, expression): |
|
|
|
""" |
|
|
|
""" |
|
|
@ -88,7 +103,7 @@ class config_get(object): |
|
|
|
for key in self.get_key_for_json(self.config_file): |
|
|
|
for key in self.get_key_for_json(self.config_file): |
|
|
|
if pattern.match(key) is not None: |
|
|
|
if pattern.match(key) is not None: |
|
|
|
real_key = key |
|
|
|
real_key = key |
|
|
|
if real_key is not '': |
|
|
|
if real_key != '': |
|
|
|
return real_key |
|
|
|
return real_key |
|
|
|
else: |
|
|
|
else: |
|
|
|
print("啊哦没有找到") |
|
|
|
print("啊哦没有找到") |
|
|
@ -101,6 +116,16 @@ class config_get(object): |
|
|
|
else: |
|
|
|
else: |
|
|
|
return self.get_value_for_json(self.config_file, real_key) |
|
|
|
return self.get_value_for_json(self.config_file, real_key) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
|
|
|
|
def move_configuration_file_old(): |
|
|
|
|
|
|
|
print("移动配置文件") |
|
|
|
|
|
|
|
os.system("cp /ql/repo/yuxian158_check/check.sample.toml /ql/config/check.toml") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
|
|
|
|
def move_configuration_file_new(): |
|
|
|
|
|
|
|
print("移动配置文件") |
|
|
|
|
|
|
|
os.system("cp /ql/data/repo/yuxian158_check/check.sample.toml /ql/data/config/check.toml") |
|
|
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
@staticmethod |
|
|
|
def get_value_for_toml(toml_path, key): |
|
|
|
def get_value_for_toml(toml_path, key): |
|
|
|
with open(toml_path, "rb") as f: |
|
|
|
with open(toml_path, "rb") as f: |
|
|
@ -166,7 +191,7 @@ class check(object): |
|
|
|
if not self.Configuration_flag: |
|
|
|
if not self.Configuration_flag: |
|
|
|
config = config_get() |
|
|
|
config = config_get() |
|
|
|
value_list = config.get_value(self.run_script_expression) |
|
|
|
value_list = config.get_value(self.run_script_expression) |
|
|
|
Push_message = "" |
|
|
|
push_message = "" |
|
|
|
num = 1 |
|
|
|
num = 1 |
|
|
|
for value in value_list: |
|
|
|
for value in value_list: |
|
|
|
print(f"<----------------账号【{num}】---------------->") |
|
|
|
print(f"<----------------账号【{num}】---------------->") |
|
|
@ -175,20 +200,20 @@ class check(object): |
|
|
|
try: |
|
|
|
try: |
|
|
|
result = func(value=value) + '\n\n' |
|
|
|
result = func(value=value) + '\n\n' |
|
|
|
print(f"执行结果:\n{result}") |
|
|
|
print(f"执行结果:\n{result}") |
|
|
|
Push_message += result |
|
|
|
push_message += result |
|
|
|
except IndexError: |
|
|
|
except IndexError: |
|
|
|
print("可能是示例格式被运行\n错误信息:") |
|
|
|
print("可能是示例格式被运行\n错误信息:") |
|
|
|
print(f"{traceback.format_exc()}") |
|
|
|
print(f"{traceback.format_exc()}") |
|
|
|
Push_message += '' |
|
|
|
push_message += '' |
|
|
|
except AttributeError: |
|
|
|
except AttributeError: |
|
|
|
print("可能是配置文件的键名出现问题\n" |
|
|
|
print("可能是配置文件的键名出现问题\n" |
|
|
|
"例如:在此次更新中什么值得买的键名从smzdm_cookie变成了cookie\n") |
|
|
|
"例如:在此次更新中什么值得买的键名从smzdm_cookie变成了cookie\n") |
|
|
|
print(f"{traceback.format_exc()}") |
|
|
|
print(f"{traceback.format_exc()}") |
|
|
|
Push_message += '' |
|
|
|
push_message += '' |
|
|
|
except TypeError: |
|
|
|
except TypeError: |
|
|
|
print(f"{traceback.format_exc()}") |
|
|
|
print(f"{traceback.format_exc()}") |
|
|
|
Push_message += '' |
|
|
|
push_message += '' |
|
|
|
send(self.run_script_name, Push_message) |
|
|
|
send(self.run_script_name, push_message) |
|
|
|
else: |
|
|
|
else: |
|
|
|
config = config_get() |
|
|
|
config = config_get() |
|
|
|
flag = config.get_value(self.run_script_expression) |
|
|
|
flag = config.get_value(self.run_script_expression) |
|
|
@ -201,7 +226,38 @@ class check(object): |
|
|
|
return wrapper |
|
|
|
return wrapper |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def change_cron(cron_file_path="/ql/db/crontab.db", repositories="yuxian158_check"): |
|
|
|
def pip_install(): |
|
|
|
|
|
|
|
print("正在安装依赖") |
|
|
|
|
|
|
|
os.system("pip3 install requests rsa tomli tomli_w beautifulsoup4") |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def change_cron_new(cron_file_path="/ql/data/db/database.sqlite", repositories="yuxian158_check"): |
|
|
|
|
|
|
|
print("尝试修改定时时间") |
|
|
|
|
|
|
|
os.system("cp /ql/data/db/database.sqlite /ql/data/db/database.sqlite.back") |
|
|
|
|
|
|
|
con = sqlite3.connect(cron_file_path) |
|
|
|
|
|
|
|
cur = con.cursor() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def change_time(time_str: str): |
|
|
|
|
|
|
|
words = re.sub("\\s+", " ", time_str).split() |
|
|
|
|
|
|
|
words[0] = str(random.randrange(60)) |
|
|
|
|
|
|
|
words[1] = str(random.randrange(22)) |
|
|
|
|
|
|
|
return " ".join(words) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
cur.execute("select id,name,command,schedule from Crontabs") |
|
|
|
|
|
|
|
res = cur.fetchall() |
|
|
|
|
|
|
|
for line in res: |
|
|
|
|
|
|
|
if line[2].find(repositories) != -1: |
|
|
|
|
|
|
|
sql = f" UPDATE Crontabs SET schedule = \"{change_time(line[3])}\" WHERE id = {line[0]}" |
|
|
|
|
|
|
|
print(f"任务名称 {line[1]} 修改为{sql}") |
|
|
|
|
|
|
|
cur.execute(sql) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
con.commit() |
|
|
|
|
|
|
|
con.close() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def change_cron_old(cron_file_path="/ql/db/crontab.db", repositories="yuxian158_check"): |
|
|
|
|
|
|
|
print("尝试修改定时时间") |
|
|
|
|
|
|
|
|
|
|
|
def change_time(time_str: str): |
|
|
|
def change_time(time_str: str): |
|
|
|
words = re.sub("\\s+", " ", time_str).split() |
|
|
|
words = re.sub("\\s+", " ", time_str).split() |
|
|
|
words[0] = str(random.randrange(60)) |
|
|
|
words[0] = str(random.randrange(60)) |
|
|
@ -226,4 +282,11 @@ def change_cron(cron_file_path="/ql/db/crontab.db", repositories="yuxian158_chec |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
|
if __name__ == "__main__": |
|
|
|
change_cron() |
|
|
|
pip_install() |
|
|
|
|
|
|
|
config = config_get() |
|
|
|
|
|
|
|
if config.config_path == "/ql/config/": |
|
|
|
|
|
|
|
change_cron_old() |
|
|
|
|
|
|
|
print("修改完成请重启容器") |
|
|
|
|
|
|
|
else: |
|
|
|
|
|
|
|
change_cron_new() |
|
|
|
|
|
|
|
print("修改完成请重启容器") |
|
|
|