import logging import subprocess from borgmatic.execute import execute_command from borgmatic.logger import get_logger logger = get_logger(__name__) def initialize_repository( repository, encryption_mode, append_only=None, storage_quota=None, local_path='borg', remote_path=None, ): ''' Given a local or remote repository path, a Borg encryption mode, whether the repository should be append-only, and the storage quota to use, initialize the repository. If the repository already exists, then log and skip initialization. ''' info_command = (local_path, 'info', repository) logger.debug(' '.join(info_command)) try: execute_command(info_command, output_log_level=None) logger.info('Repository already exists. Skipping initialization.') return except subprocess.CalledProcessError as error: if error.returncode != 2: raise init_command = ( (local_path, 'init', repository) + (('--encryption', encryption_mode) if encryption_mode else ()) + (('--append-only',) if append_only else ()) + (('--storage-quota', storage_quota) if storage_quota else ()) + (('--info',) if logger.getEffectiveLevel() == logging.INFO else ()) + (('--debug',) if logger.isEnabledFor(logging.DEBUG) else ()) + (('--remote-path', remote_path) if remote_path else ()) ) # Don't use execute_command() here because it doesn't support interactive prompts. subprocess.check_call(init_command)