FIRST commit

depend.py

@@ -0,0 +1,116 @@
+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("加载通知服务失败~")