tools.win_pg_dump_controller

Yohn Y. 2022-06-13 Parent:7c93b0305522

5:7de4fb4ef2f5 Go to Latest

tools.win_pg_dump_controller/win_pg_dump_controller/log_controller.py

. Не тот порт по умолчанию для postgres

Download raw file
View source Diff to previous Annotate
History
1 # coding: utf-8
2 """\
3 Модуль логирования
4
5 Метки в журнале о уровне сообщения:
6 "`": Debug
7 ".": Info
8 "*": Warning
9 "!": Error
10 "#": Alert
11 """
12
13 from time import time, ctime
14 from datetime import timedelta, datetime
15 from sys import exc_info
16 from os.path import join as p_join
17 from os import listdir, remove as file_remove, stat
18 from traceback import extract_tb, extract_stack
19 from typing import Optional
20
21 from .config import Config
22
23
24 class Timing(object):
25 def __init__(self, name: Optional[str] = None):
26 if name is None:
27 self.prefix = ''
28 else:
29 self.prefix = f'{name} :: '
30 self.tsAll = time()
31 self.ts = self.tsAll
32
33 def get_time(self):
34 return time() - self.ts
35
36 def reset(self):
37 self.ts = time()
38 self.tsAll = self.ts
39
40 def __str__(self):
41 ts = time()
42 return self.prefix + '%s(%.4f)' % (timedelta(seconds=(ts - self.tsAll)), ts - self.ts)
43
44 def __call__(self, msg):
45 _buf = f'{self} | {msg}'
46 self.ts = time()
47 return _buf
48
49
50 class BaseLogger(object):
51 def __init__(self, appname='main'):
52 self.appname = appname
53
54 @staticmethod
55 def _write(itr_content):
56 raise NotImplemented()
57
58 def __call__(self, msg):
59 self._write(map(
60 lambda x: '%3s | %s :: %s' % ('.', self.appname, x),
61 str(msg).splitlines()
62 ))
63
64 def err(self, msg):
65 self._write(map(
66 lambda x: '%3s | %s :: %s' % ('!', self.appname, x),
67 str(msg).splitlines()
68 ))
69
70 def warn(self, msg):
71 self._write(map(
72 lambda x: '%3s | %s :: %s' % ('*', self.appname, x),
73 str(msg).splitlines()
74 ))
75
76 def alert(self, msg):
77 self._write(map(
78 lambda x: '%3s | %s :: %s' % ('#', self.appname, x),
79 str(msg).splitlines()
80 ))
81
82 def debug(self, msg):
83 self._write(map(
84 lambda x: '%3s | %s :: %s' % ('`', self.appname, x),
85 str(msg).splitlines()
86 ))
87
88 @staticmethod
89 def get_timing(name: Optional[str] = None):
90 return Timing(name)
91
92 def sublog(self, name):
93 return self.__class__(f'{self.appname}/{name}')
94
95 def excpt(self, msg, e_class=None, e_obj=None, e_tb=None, stack_skip=0):
96 if e_class is None:
97 e_class, e_obj, e_tb = exc_info()
98
99 tb_data_tb = list(extract_tb(e_tb))[::-1]
100 tb_data_stack = list(extract_stack())[::-1][(2 + stack_skip):]
101 self.err(msg)
102 self.err('--- EXCEPTION ---')
103 self.err(' %s (%s)' % (e_class.__name__, e_obj))
104 self.err('--- TRACEBACK ---')
105 for _tbFile, _tbLine, _tbFunc, _tbText in tb_data_tb:
106 self.err('File: %s, line %s in %s' % (_tbFile, _tbLine, _tbFunc))
107 self.err(' %s' % _tbText)
108 self.err('>>> Exception Handler <<<')
109 for _tbFile, _tbLine, _tbFunc, _tbText in tb_data_stack:
110 self.err('File: %s, line %s in %s' % (_tbFile, _tbLine, _tbFunc))
111 self.err(' %s' % _tbText)
112 self.err('--- END EXCEPTION ---')
113
114
115 class NullLogger(BaseLogger):
116 @staticmethod
117 def _write(itr_content):
118 pass
119
120
121 class FileLogger(BaseLogger):
122 def __init__(self, filename: str, appname: str = 'main'):
123 super().__init__(appname)
124 self.fd = open(filename, 'a')
125
126 def _write(self, itr_content):
127 cur_time = ctime()
128 for i in itr_content:
129 self.fd.write(f'{cur_time}{i}\n')
130
131 self.fd.flush()
132
133 def __del__(self):
134 try:
135 self.fd.close()
136 except:
137 pass
138
139
140 class LogController(object):
141 def __init__(self, config: Config):
142 self.log_dir = config.log_dir
143 self.keep_logs_days = config.keep_logs_days
144 self.log_files = []
145 self.main_log = None
146
147 @staticmethod
148 def _get_timeprefix() -> str:
149 return datetime.now().strftime('%Y-%m-%d_%H-%M-%S')
150
151 def get_logger(self, name: str) -> BaseLogger:
152 if self.log_dir is not None:
153 _log_name = p_join(self.log_dir, f'{self._get_timeprefix()} - {name}.log')
154 if name.lower() != 'main':
155 self.log_files.append(_log_name)
156
157 else:
158 self.main_log = _log_name
159
160 return FileLogger(_log_name, appname=name)
161
162 else:
163 return NullLogger(appname=name)
164
165 def get_filename(self, name: str) -> Optional[str]:
166 if self.log_dir is not None:
167 return str(p_join(self.log_dir, f'{self._get_timeprefix()} - {name}.log'))
168
169 def clean(self):
170 if self.log_dir is not None:
171 now = time()
172 for f in listdir(self.log_dir):
173 f_path = p_join(self.log_dir, f)
174 f_stat = stat(f_path)
175
176 if divmod(now - f_stat.st_ctime, 86400)[0] > self.keep_logs_days:
177 file_remove(f_path)