faker3/depend.py

117 lines
3.9 KiB
Python
Raw Normal View History

2024-06-28 15:38:28 +08:00
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("加载通知服务失败~")