tools.win_pg_dump_controller: be791d354d2a win_pg_dump

1 # coding: utf-8

3 import subprocess as sp

4 from os.path import join as p_join

5 from os import environ

7 from .store_controller import StoreController

8 from .log_controller import LogController

9 from .config import Config, DBTask

10 from .error import Error

13 class ExecutorError(Error):

14 pass

17 def backup(task: DBTask, config: Config, log_controller: LogController):

18 log = log_controller.get_logger(f'{task.name}.executor')

19 log_t = log.get_timing()

20 log(log_t(f'Начинаем копирование {task.name}'))

23 stor = StoreController(task)

24 backup_item = stor.new_item()

26 log(log_t(f'Копируется "{task.db_name}" -> "{backup_item.get_path()}" '

27 f'с параметрами pg_dump "{config.pg_dump_flags}"'))

29 pg_dump = p_join(config.pg_bin_path, 'pg_dump.exe')

30 pg_dump = (

31 f'"{pg_dump}" {config.pg_dump_flags} -F c -v --clean '

32 f'-U "{task.user_name}" -h "{task.host_name}" -p "{task.port}" '

33 f'-f "{backup_item.get_path()}" {task.db_name}'

36 environ['PGPASSWORD'] = task.passwd

38 pglog_name = log_controller.get_filename(task.name)

39 pglog_fd = sp.DEVNULL if pglog_name is None else open(pglog_name, 'w')

41 log(log_t(f'Пробуем запустить: {pg_dump}'))

43 try:

44 cmd = sp.Popen(pg_dump, stderr=sp.STDOUT, stdout=pglog_fd, shell=True)

45 cmd.wait()

47 log(log_t(f'Команда выполнена, статус возврата "{cmd.returncode}", журнал: "{pglog_name}"'))

49 if cmd.returncode != 0:

50 raise ExecutorError(f'Не нулевой код возврата, подробности в "{pglog_name}"')

52 except:

53 stor.remove(backup_item)

54 raise

56 else:

57 stor.add_item(backup_item)

59 log(log_t('Очистка старых копий'))

61 log(log_t('\n'.join(map(lambda x: f'- {x}', stor.clean(config.teir1_days, config.teir2_copies_interval,

62 config.tier2_store_days)))))

64 stor.save_index()

tools.win_pg_dump_controller