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/config.py

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

Download raw file
View source Diff to previous Annotate
History
1 # coding: utf-8
2 from configparser import ConfigParser, SectionProxy
3 from sys import argv
4 from os.path import isdir
5 from typing import Optional, Iterable
6
7
8 from .error import Error
9
10
11 class ConfigError(Error):
12 pass
13
14
15 COMMON_SECTION = 'common'
16 MAIN = 'main'
17 SMTP = 'smtp'
18 DEFAULT_PG_PORT = 5432
19
20
21 class DBTask(object):
22 def __init__(self, task_name: str, host_name: str, db_name: str, user_name: str, passwd: Optional[str],
23 dst_dir: str, port: int = DEFAULT_PG_PORT):
24 self.host_name = host_name
25 self.db_name = db_name
26 self.user_name = user_name
27 self.passwd = passwd
28 self.dst_dir = dst_dir
29 self.name = task_name
30 self.port = port
31
32 if not (self.db_name or self.user_name or self.dst_dir or self.host_name):
33 raise ConfigError(f'Some important config parameters not set: '
34 f'db_name="{db_name}" '
35 f'user_name="{user_name}" '
36 f'dst_dir="{dst_dir}" '
37 f'host_name="{host_name}"')
38
39
40 class CommonTaskDescription(object):
41 def __init__(self, host_name: Optional[str], user_name: Optional[str],
42 passwd: Optional[str], dst_dir: Optional[str], port: int = DEFAULT_PG_PORT):
43 self.user_name = user_name
44 self.passwd = passwd
45 self.dst_dir = dst_dir
46 self.host_name = host_name
47 self.port = port
48
49 def parse_section(self, config_section: SectionProxy) -> DBTask:
50 db_name = config_section.get('db_name')
51 user_name = config_section.get('user_name', self.user_name)
52 passwd = config_section.get('passwd', self.passwd)
53 dst_dir = config_section.get('dst_dir', self.dst_dir)
54 host_name = config_section.get('host_name', self.host_name)
55 port = config_section.getint('port', self.port)
56
57 if not isdir(dst_dir):
58 raise ConfigError(f'Destionation directory not exists for "{config_section.name}": {dst_dir}')
59
60 return DBTask(task_name=config_section.name, host_name=host_name, db_name=db_name,
61 user_name=user_name, passwd=passwd, port=port, dst_dir=dst_dir)
62
63 @classmethod
64 def parse_config(cls, config: ConfigParser) -> Iterable[DBTask]:
65 if not config.has_section(COMMON_SECTION):
66 tmpl = cls(host_name=None, user_name=None, passwd=None, dst_dir=None)
67
68 else:
69 _section = config['common']
70 host_name = _section.get('host_name', '127.0.0.1')
71 user_name = _section.get('user_name')
72 passwd = _section.get('passwd')
73 dst_dir = _section.get('dst_dir')
74 port = _section.getint('port', DEFAULT_PG_PORT)
75
76 tmpl = cls(host_name=host_name, user_name=user_name, passwd=passwd, dst_dir=dst_dir, port=port)
77
78 for _section in filter(lambda x: x not in (COMMON_SECTION, MAIN, SMTP), config.sections()):
79 yield tmpl.parse_section(config[_section])
80
81
82 class Config(object):
83 def __init__(self):
84 try:
85 config_file = argv[1]
86 except IndexError:
87 raise ConfigError(f'No config file specified')
88
89 _config = ConfigParser()
90 _config.read(config_file)
91
92 _main_section = _config[MAIN]
93 self.pg_bin_path = _main_section.get('pg_bin_path')
94 self.pg_dump_flags = _main_section.get('pg_dump_flags', '')
95 self.log_dir = _main_section.get('log_dir')
96 self.teir1_days = _main_section.getint('teir1_days', 7)
97 self.teir2_copies_interval = _main_section.getint('teir2_copies_interval', 7)
98 self.tier2_store_days = _main_section.getint('tier2_store_days', 30)
99 self.keep_logs_days = _main_section.getint('keep_logs_days', 30)
100
101 self.smtp_server = None
102 self.mail_from = None
103 self.mail_to = None
104 self.smtp_port = None
105
106 if _config.has_section(SMTP):
107 _smtp_section = _config[SMTP]
108 self.smtp_server = _smtp_section.get('smtp server')
109 self.mail_from = _smtp_section.get('mail from')
110 self.mail_to = _smtp_section.get('mail to', fallback='root')
111 self.smtp_port = _smtp_section.getint('smtp port', fallback=25)
112
113 if not isdir(self.pg_bin_path):
114 raise ConfigError(f'No valid directory with binnary files of PostgreSQL is set: {self.pg_bin_path}')
115
116 self.tasks = list(CommonTaskDescription.parse_config(_config))