faker3/depend.py
2024-06-28 15:38:28 +08:00

117 lines
3.9 KiB
Python
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

import json
import re
import sys
import os
class Depend:
@staticmethod
def get_ql_path():
if re.search('/ql/data/', sys.path[0]):
return '/ql/data/'
else:
return '/ql/'
@staticmethod
def get_env(env, default=None):
"""
青龙环境变量读取支持将整数bool类型变量转化为正常的值
Args:
env: 字符串,被读取的青龙环境变量
default: 字符串,如果找不到这个环境变量,返回的默认变量
Returns:
result 被格式化的变量
"""
if env in os.environ and os.environ[env]:
if os.environ[env] in ["True", "False"]:
return False if os.environ[env] == "False" else True
elif os.environ[env].isdigit():
return int(os.environ[env])
else:
return os.environ[env]
else:
if default:
if default in ["True", "False"]:
return False if default == "False" else True
elif default.isdigit():
return int(default)
else:
return default
else:
return None
@staticmethod
def str2list(string):
if string and string != "":
if re.search(",", string):
return string.split(",")
else:
return [string]
else:
return []
@staticmethod
def not2append(addlist, appended):
for i in addlist:
if i not in appended:
appended.append(i)
return appended
@staticmethod
def re_filter_list(string, filter_list):
for i in filter_list:
if re.search(i, string):
return True
return False
def only_check(self, pyname, pyabspath,osenv=None):
only_path = self.get_ql_path() + pyname + '_by_keven1024'
result = "☺当前脚本目录为: " + str(pyabspath) + "\n"
j_data = {
"py_path": None
}
if osenv and self.get_env(osenv):
result += "😏检测到环境变量:" + str(osenv) + " = " + self.get_env(osenv) + " 将按照该路径为准\n"
pyabspath = self.get_env(osenv)
if os.path.exists(only_path):
with open(only_path, 'r+') as f:
py_data = f.read(2097152)
if py_data == pyabspath:
# 对旧版转换为json格式
j_data["py_path"] = pyabspath
f.seek(0, os.SEEK_SET)
result += "😏检测到旧版检测文件,自动转换新版\n"
f.writelines(json.dumps(j_data))
try:
j_data = json.loads(py_data)
except:
pass
if j_data["py_path"] and j_data["py_path"] == pyabspath:
result += "😁脚本唯一性检测通过,继续运行!\n"
else:
result += "🙄检测到其他同类型的青龙日志分析脚本存在,拒绝运行!\n"
load_send()
send(pyname, result)
exit(0)
else:
with open(only_path, "w") as f:
j_data["py_path"] = pyabspath
f.writelines(json.dumps(j_data))
result += "🙄检测到第一次运行,已写入唯一性检测文件,如无特殊情况请勿删除\n"
return result
def load_send():
global send
cur_path = os.path.abspath(os.path.dirname(__file__))
sys.path.append(cur_path)
if os.path.exists(cur_path + "/notify.py"):
try:
from notify import send
except:
send = False
print("加载通知服务失败~")
else:
send = False
print("加载通知服务失败~")