chickliu/quark-auto-save
1
0
Fork 0
mirror of https://github.com/Cp0204/quark-auto-save.git synced 2026-01-12 15:20:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🔨 全局变量大写

Browse Source
This commit is contained in:
Cp0204 2024-05-24 13:54:08 +08:00
parent a4425a1c19
commit 46bec2331c
1 changed files with 13 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
app/run.py
View File

@ -33,9 +33,9 @@ def get_app_ver():
# 文件路径 # 文件路径
python_path = "python3" if os.path.exists("/usr/bin/python3") else "python" PYTHON_PATH = "python3" if os.path.exists("/usr/bin/python3") else "python"
script_path = os.environ.get("SCRIPT_PATH", "./quark_auto_save.py") SCRIPT_PATH = os.environ.get("SCRIPT_PATH", "./quark_auto_save.py")
config_path = os.environ.get("CONFIG_PATH", "./config/quark_config.json") CONFIG_PATH = os.environ.get("CONFIG_PATH", "./config/quark_config.json")
DEBUG = os.environ.get("DEBUG", False) DEBUG = os.environ.get("DEBUG", False)
app = Flask(__name__) app = Flask(__name__)
@ -65,14 +65,14 @@ def gen_md5(string):
# 读取 JSON 文件内容 # 读取 JSON 文件内容
def read_json(): def read_json():
with open(config_path, "r", encoding="utf-8") as f: with open(CONFIG_PATH, "r", encoding="utf-8") as f:
data = json.load(f) data = json.load(f)
return data return data
# 将数据写入 JSON 文件 # 将数据写入 JSON 文件
def write_json(data): def write_json(data):
with open(config_path, "w", encoding="utf-8") as f: with open(CONFIG_PATH, "w", encoding="utf-8") as f:
json.dump(data, f, indent=4, ensure_ascii=False, sort_keys=False) json.dump(data, f, indent=4, ensure_ascii=False, sort_keys=False)
@ -167,7 +167,7 @@ def run_script_now():
if not is_login(): if not is_login():
return "未登录" return "未登录"
task_index = request.args.get("task_index", "") task_index = request.args.get("task_index", "")
command = [python_path, "-u", script_path, config_path, task_index] command = [PYTHON_PATH, "-u", SCRIPT_PATH, CONFIG_PATH, task_index]
logging.info(f">>> 手动运行任务{task_index}") logging.info(f">>> 手动运行任务{task_index}")
def generate_output(): def generate_output():
@ -196,7 +196,7 @@ def run_script_now():
# 定时任务执行的函数 # 定时任务执行的函数
def run_python(args): def run_python(args):
logging.info(f">>> 定时运行任务") logging.info(f">>> 定时运行任务")
os.system(f"{python_path} {args}") os.system(f"{PYTHON_PATH} {args}")
# 重新加载任务 # 重新加载任务
@ -213,8 +213,8 @@ def reload_tasks():
scheduler.add_job( scheduler.add_job(
run_python, run_python,
trigger=trigger, trigger=trigger,
args=[f"{script_path} {config_path}"], args=[f"{SCRIPT_PATH} {CONFIG_PATH}"],
id=script_path, id=SCRIPT_PATH,
) )
if scheduler.state == 0: if scheduler.state == 0:
scheduler.start() scheduler.start()
@ -234,10 +234,10 @@ def reload_tasks():
def init(): def init():
logging.info(f">>> 初始化配置") logging.info(f">>> 初始化配置")
# 检查配置文件是否存在 # 检查配置文件是否存在
if not os.path.exists(config_path): if not os.path.exists(CONFIG_PATH):
if not os.path.exists(os.path.dirname(config_path)): if not os.path.exists(os.path.dirname(CONFIG_PATH)):
os.makedirs(os.path.dirname(config_path)) os.makedirs(os.path.dirname(CONFIG_PATH))
with open("quark_config.json", "rb") as src, open(config_path, "wb") as dest: with open("quark_config.json", "rb") as src, open(CONFIG_PATH, "wb") as dest:
dest.write(src.read()) dest.write(src.read())
data = read_json() data = read_json()
# 默认管理账号 # 默认管理账号