| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| import json |
| import os |
| import time |
|
|
| import db |
| import public |
|
|
|
|
| class panelBase: |
| history_data_dir = None |
| data_dir = None |
| __plugin_path = '/www/server/total' |
| __frontend_path = '/www/server/panel/plugin/total' |
|
|
| def __init__(self): |
| pass |
|
|
| @staticmethod |
| def get_time_interval(local_time): |
| time_key_format = "%Y%m%d00" |
| start = int(time.strftime(time_key_format, local_time)) |
| time_key_format = "%Y%m%d23" |
| end = int(time.strftime(time_key_format, local_time)) |
| return start, end |
|
|
| @staticmethod |
| def get_time_interval_min(local_time): |
| time_key_format = "%Y%m%d0000" |
| start = int(time.strftime(time_key_format, local_time)) |
| time_key_format = "%Y%m%d2359" |
| end = int(time.strftime(time_key_format, local_time)) |
| return start, end |
|
|
| def get_time_interval_hour(self, local_time, second): |
| start = int(time.mktime(local_time) - second) |
| start_str = time.strftime("%Y%m%d%H", time.localtime(start)) |
| end_str = time.strftime("%Y%m%d%H", local_time) |
| return start_str, end_str |
|
|
| def get_last_days(self, day): |
| now = time.localtime() |
| if day == 30: |
| last_month = now.tm_mon - 1 |
| if last_month <= 0: |
| last_month = 12 |
| import calendar |
| _, last_month_days = calendar.monthrange(now.tm_year, last_month) |
| day = last_month_days |
| else: |
| day += 1 |
|
|
| t1 = time.mktime( |
| (now.tm_year, now.tm_mon, now.tm_mday - day, 0, 0, 0, 0, 0, 0)) |
| t2 = time.localtime(t1) |
| start, _ = self.get_time_interval(t2) |
| _, end = self.get_time_interval(now) |
| start_7_day_hour, end_7_day_hour = self.get_time_interval_hour(now, 604800) |
| start_one_day_hour, end_one_day_hour = self.get_time_interval_hour(now, 86400) |
| start_hour, end_hour = self.get_time_interval_hour(now, 3600) |
| return [start, end, |
| start_one_day_hour, end_one_day_hour, |
| start_hour, end_hour, |
| start_7_day_hour, end_7_day_hour] |
|
|
| @staticmethod |
| def get_time_key(date=None): |
| if date is None: |
| date = time.localtime() |
| time_key = 0 |
| time_key_format = "%Y%m%d%H" |
| if type(date) == time.struct_time: |
| time_key = int(time.strftime(time_key_format, date)) |
| if type(date) == str: |
| time_key = int(time.strptime(date, time_key_format)) |
| return time_key |
|
|
| def get_query_date(self, query_date): |
| """ |
| 获取查询日期的时间区间 |
| 查询日期的表示形式是以区间的格式表示，这里也考虑到了数据库里面实际存储的方式。 |
| 在数据库里面是以小时为单位统计日志数据，所以这里的区间也是也小时来表示。 |
| 比如表示今天，假设今天是2021/3/30，查询日期的格式是2021033000-2021033023。 |
| @param query_date: |
| @return: 2021033000-2021033023 |
| """ |
| start_date = None |
| end_date = None |
| if query_date == "today": |
| start_date, end_date = self.get_time_interval(time.localtime()) |
| elif query_date == "yesterday": |
| |
| now = time.localtime() |
| yes_i = time.mktime((now.tm_year, now.tm_mon, now.tm_mday - 1, 0, 0, 0, 0, 0, 0)) |
| yes = time.localtime(yes_i) |
| start_date, end_date = self.get_time_interval(yes) |
| elif query_date.startswith("l"): |
| days = int(query_date[1:]) |
| start_date, end_date = self.get_last_days(days) |
| elif query_date.startswith("this_month"): |
| now = time.localtime() |
| start_date = time.localtime(time.mktime((now.tm_year, now.tm_mon, 1, 0, 0, 0, 0, 0, 0))) |
| start_date = self.get_time_key(start_date) |
| end_date = self.get_time_key(now) + 1 |
| elif query_date.startswith("h1"): |
| |
| |
| |
| now = time.localtime() |
| if now.tm_min >= 30: |
| start_date = time.mktime((now.tm_year, now.tm_mon, now.tm_mday, now.tm_hour, 0, 0, 0, 0, 0)) |
| start_date = self.get_time_key(start_date) |
| end_date = start_date |
| else: |
| s = time.localtime(time.mktime((now.tm_year, now.tm_mon, now.tm_mday, now.tm_hour - 1, 0, 0, 0, 0, 0))) |
| start_date = self.get_time_key(s) |
| e = time.localtime(time.mktime((now.tm_year, now.tm_mon, now.tm_mday, now.tm_hour, 0, 0, 0, 0, 0))) |
| end_date = self.get_time_key(e) |
| else: |
| if query_date.find("-") >= 0: |
| s, e = query_date.split("-") |
| start_time = time.strptime(s.strip(), "%Y%m%d") |
| start_date, _ = self.get_time_interval(start_time) |
| end_time = time.strptime(e.strip(), "%Y%m%d") |
| _, end_date = self.get_time_interval(end_time) |
| else: |
| if query_date.isdigit(): |
| s_time = time.strptime(query_date, "%Y%m%d") |
| start_date, end_date = self.get_time_interval(s_time) |
|
|
| return start_date, end_date |
|
|
| def __get_file_json(self, filename, defaultv={}): |
| try: |
| if not os.path.exists(filename): return defaultv |
| return json.loads(public.readFile(filename)) |
| except: |
| os.remove(filename) |
| return defaultv |
|
|
| def __read_frontend_config(self): |
| config_json = self.__frontend_path + "/config.json" |
| data = {} |
| if os.path.exists(config_json): |
| data = json.loads(public.readFile(config_json)) |
| return data |
|
|
| def get_default_site(self): |
| config = self.__read_frontend_config() |
| default = None |
| if "default_site" in config: |
| default = config["default_site"] |
| if not default: |
| site = public.M('sites').field('name').order("addtime").find() |
| default = site["name"] |
| return default |
|
|
| def get_site_settings(self, site): |
| """获取站点配置""" |
|
|
| config_path = "/www/server/total/config.json" |
| config = self.__get_file_json(config_path) |
| if not config: |
| return {} |
|
|
| if site not in config.keys(): |
| res_config = config["global"] |
| res_config["push_report"] = False |
| else: |
| res_config = config[site] |
|
|
| for k, v in config["global"].items(): |
| if k not in res_config.keys(): |
| if k == "push_report": |
| res_config[k] = False |
| else: |
| res_config[k] = v |
| res_config["default_site"] = self.get_default_site() |
| return res_config |
|
|
| def get_data_dir(self): |
| if self.data_dir is None: |
| default_data_dir = os.path.join(self.__plugin_path, "logs") |
| settings = self.get_site_settings("global") |
| if "data_dir" in settings.keys(): |
| config_data_dir = settings["data_dir"] |
| else: |
| config_data_dir = default_data_dir |
| self.data_dir = default_data_dir if not config_data_dir else config_data_dir |
| return self.data_dir |
|
|
| def get_log_db_path(self, site, db_name="logs.db", history=False): |
| site = site.replace('_', '.') |
| if not history: |
| data_dir = self.get_data_dir() |
| db_path = os.path.join(data_dir, site, db_name) |
| else: |
| data_dir = self.get_history_data_dir() |
| db_name = "history_logs.db" |
| db_path = os.path.join(data_dir, site, db_name) |
| return db_path |
|
|
| def get_history_data_dir(self): |
| if self.history_data_dir is None: |
| default_history_data_dir = os.path.join(self.__plugin_path, "logs") |
| settings = self.get_site_settings("global") |
| if "history_data_dir" in settings.keys(): |
| config_data_dir = settings["history_data_dir"] |
| else: |
| config_data_dir = default_history_data_dir |
| self.history_data_dir = default_history_data_dir if not config_data_dir else config_data_dir |
| return self.history_data_dir |
|
|
| def _get_ts(self, get): |
| """ |
| @获取数据库连接对象 |
| @return: |
| """ |
| db_name = get.db_name if "db_name" in get else "total.db" |
| db_path = self.get_log_db_path(get.site_name, db_name=db_name) |
| if not os.path.isdir(os.path.dirname(db_path)): |
| os.makedirs(os.path.dirname(db_path)) |
| public.set_mode(os.path.dirname(db_path), "755") |
| public.set_own(os.path.dirname(db_path), "www") |
|
|
| ts = db.Sql() |
| ts.dbfile(db_path) |
| if not ts: |
| return False |
| return ts |
|
|
