def run(config):
borg = local['borg']
logging.info('Starting borg backup to {}'.format(borg_repository(config)))
archive_name = datetime.datetime.now().__format__(config['where']['archive-template'])
archive = '{}::{}'.format(borg_repository(config), archive_name)
with open('/tmp/borg-exclude-file', 'w') as exclude_file:
if config['what']['exclude']:
exclude_file.writelines(config['what']['exclude'])
exclude_file.write("\n")
if config['what']['exclude-files']:
for f in config['what']['exclude-files']:
with open(f) as input_file:
exclude_file.write(input_file.read())
exclude_file.write("\n")
command = borg['create', archive, '--exclude-from', '/tmp/borg-exclude-file', '--exclude-caches']
arguments = []
if config['where']['compression'] == 'fast':
arguments.append('-C')
arguments.append('lz4')
if config['where']['compression'] == 'slow':
arguments.append('-C')
arguments.append('lzma,8')
if config['what']['include']:
arguments.extend(config['what']['include'])
if config['what']['include-files']:
for f in config['what']['include-files']:
with open(f) as input_file:
arguments.extend(input_file.readlines())
try:
command(tuple(arguments))
except ProcessExecutionError as e:
logging.error('Backup failed, borg returned error')
result = e.stdout + e.stderr
common.failure(config, result)
logging.info('Borg backup complete')
cleanup(config)
def verify(config):
try:
local['borg']
except CommandNotFound as e:
common.failure(common, '"borg" command is not found in $PATH')
def main():
parser = argparse.ArgumentParser(description="Backup manager")
parser.add_argument('--config', '-c', help="Config file location", default="/etc/backup.yml")
parser.add_argument('--quiet', '-q', help="Raise loglevel to warning", action="store_true")
parser.add_argument('--debug', '-d', help="Set loglevel to debug", action="store_true", default=False)
parser.add_argument('command', choices=["init", "run", "info", "verify", "setup-systemd"])
args = parser.parse_args()
if not args.quiet:
logging.basicConfig(level=logging.INFO)
if args.debug:
logging.basicConfig(level=logging.DEBUG)
logging.debug("Debug logging enabled")
if not os.path.isfile(args.config) and args.command != 'init':
logging.error("Config file does not exist at {}".format(args.config))
logging.warning("You can create a config file scaffolding with 'backupmanager init'")
exit(1)
if args.command == "init":
if not os.path.isfile('/etc/backup.yml'):
logging.info('Creating /etc/backup.yml from template')
filename = pkg_resources.resource_filename('backupmanager', 'backup.yml.dist')
copyfile(filename, '/etc/backup.yml')
else:
logging.warning('/etc/backup.yml exists already, skipping')
exit(0)
with open(args.config) as config_file:
config = yaml.load(config_file)
tool = config['tool']
if tool == "borg":
tool = borg
if tool == "duplicity":
tool = duplicity
else:
logging.error("The tool '{}' is not supported".format(tool))
if args.command == "info":
tool.info(config)
elif args.command == "run":
common.verify_config(config)
if config['hooks']['pre-backup']:
logging.info('Running pre-backup scripts')
for hook in config['hooks']['pre-backup']:
logging.info('Running {}'.format(hook))
try:
subprocess.check_output(hook, shell=True)
except subprocess.CalledProcessError as e:
common.failure(config, e.output)
tool.run(config)
if config['hooks']['post-backup']:
logging.info('Running post-backup scripts')
for hook in config['hooks']['post-backup']:
logging.info('Running {}'.format(hook))
try:
subprocess.check_output(hook, shell=True)
except subprocess.CalledProcessError as e:
common.failure(config, e.output)
elif args.command == "verify":
common.verify_config(config)
tool.verify(config)
elif args.command == "setup-systemd":
if not os.path.isfile('/etc/systemd/system/backup.service'):
logging.info('Creating /etc/systemd/system/backup.service from template')
filename = pkg_resources.resource_filename('backupmanager', 'systemd/backup.service')
copyfile(filename, '/etc/systemd/system/backup.service')
else:
logging.warning('/etc/systemd/system/backup.service exists already, skipping')
if not os.path.isfile('/etc/systemd/system/backup.timer'):
logging.info('Creating /etc/systemd/system/backup.timer from template')
filename = pkg_resources.resource_filename('backupmanager', 'systemd/backup.timer')
copyfile(filename, '/etc/systemd/system/backup.timer')
else:
logging.warning('/etc/systemd/system/backup.timer exists already, skipping')
systemctl = local['systemctl']
systemctl('daemon-reload')
systemctl('enable', 'backup.timer')
systemctl('start', 'backup.timer')
