def main(args=None):
"""
The main script entry point
:param list[str] args: the raw arguments list. When not provided
it defaults to sys.args[1:]
"""
config = parse_arguments(args)
configure_logging(config)
try:
cloud_interface = get_cloud_interface(config)
with closing(cloud_interface):
if not cloud_interface.test_connectivity():
raise NetworkErrorExit()
# If test is requested, just exit after connectivity test
elif config.test:
raise SystemExit(0)
if not cloud_interface.bucket_exists:
logging.error("Bucket %s does not exist",
cloud_interface.bucket_name)
raise OperationErrorExit()
catalog = CloudBackupCatalog(cloud_interface, config.server_name)
if config.release:
catalog.release_keep(config.backup_id)
elif config.status:
target = catalog.get_keep_target(config.backup_id)
if target:
print("Keep: %s" % target)
else:
print("Keep: nokeep")
else:
backup_info = catalog.get_backup_info(config.backup_id)
if backup_info.status == BackupInfo.DONE:
catalog.keep_backup(config.backup_id, config.target)
else:
logging.error(
"Cannot add keep to backup %s because it has status %s. "
"Only backups with status DONE can be kept.",
config.backup_id,
backup_info.status,
)
raise OperationErrorExit()
except Exception as exc:
logging.error("Barman cloud keep exception: %s", force_str(exc))
logging.debug("Exception details:", exc_info=exc)
raise GeneralErrorExit()
class CloudBackupDownloader(object):
"""
Cloud storage download client
"""
def __init__(self, cloud_interface, server_name):
"""
Object responsible for handling interactions with cloud storage
:param CloudInterface cloud_interface: The interface to use to
upload the backup
:param str server_name: The name of the server as configured in Barman
"""
self.cloud_interface = cloud_interface
self.server_name = server_name
self.catalog = CloudBackupCatalog(cloud_interface, server_name)
def download_backup(self, backup_id, destination_dir, tablespaces):
"""
Download a backup from cloud storage
:param str backup_id: The backup id to restore
:param str destination_dir: Path to the destination directory
"""
backup_info = self.catalog.get_backup_info(backup_id)
if not backup_info:
logging.error("Backup %s for server %s does not exists", backup_id,
self.server_name)
raise SystemExit(1)
backup_files = self.catalog.get_backup_files(backup_info)
# We must download and restore a bunch of .tar files that contain PGDATA
# and each tablespace. First, we determine a target directory to extract
# each tar file into and record these in copy_jobs. For each tablespace,
# the location may be overriden by `--tablespace name:/new/location` on
# the command-line; and we must also add an entry to link_jobs to create
# a symlink from $PGDATA/pg_tblspc/oid to the correct location after the
# downloads.
copy_jobs = []
link_jobs = []
for oid in backup_files:
file_info = backup_files[oid]
# PGDATA is restored where requested (destination_dir)
if oid is None:
target_dir = destination_dir
else:
for tblspc in backup_info.tablespaces:
if oid == tblspc.oid:
target_dir = tblspc.location
if tblspc.name in tablespaces:
target_dir = os.path.realpath(
tablespaces[tblspc.name])
logging.debug(
"Tablespace %s (oid=%s) will be located at %s",
tblspc.name,
oid,
target_dir,
)
link_jobs.append([
"%s/pg_tblspc/%s" % (destination_dir, oid),
target_dir
])
break
else:
raise AssertionError(
"The backup file oid '%s' must be present "
"in backupinfo.tablespaces list")
# Validate the destination directory before starting recovery
if os.path.exists(target_dir) and os.listdir(target_dir):
logging.error(
"Destination %s already exists and it is not empty",
target_dir)
raise SystemExit(1)
copy_jobs.append([file_info, target_dir])
for additional_file in file_info.additional_files:
copy_jobs.append([additional_file, target_dir])
# Now it's time to download the files
for file_info, target_dir in copy_jobs:
# Download the file
logging.debug(
"Extracting %s to %s (%s)",
file_info.path,
target_dir,
"decompressing " + file_info.compression
if file_info.compression else "no compression",
)
self.cloud_interface.extract_tar(file_info.path, target_dir)
for link, target in link_jobs:
os.symlink(target, link)
# If we did not restore the pg_wal directory from one of the uploaded
# backup files, we must recreate it here. (If pg_wal was originally a
# symlink, it would not have been uploaded.)
wal_path = os.path.join(destination_dir, backup_info.wal_directory())
if not os.path.exists(wal_path):
os.mkdir(wal_path)