rbackup/bin/backup

#!/usr/bin/python3
"""
.. moduleauthor:: Eric Torres

Command-Line Arguments
======================

-c, --use-checksums   use rsync's checksum feature to detect file changes
--debug               show debug messages
-n, --name            name to give to the backup snapshot
-s, --run-post-sync   run sync syscall after backup
-u, --umask           umask value to use while running backup process

Run a local backup, creating a snapshot in the process.

On each run of this script, a new snapshot is made and any unchanged
files are hardlinked into the new snapshot.
"""
import argparse
import logging
import os

import rbackup.config.config_files as config
import rbackup.logging
import rbackup.script as script
from rbackup.struct.repository import Repository

# ========== Constants ==========
EXTRA_RSYNC_OPTS = {
    "delete": "--delete-after",
    "checksum": "--checksum",
    "update": "--update",
}

DESCRIPTION = "Run a backup on a repository within a local path"

# ----- Error Codes -----
E_PERMISSION = 13


# ========== Logging Setup ==========
syslog = logging.getLogger("rbackup")
syslog.setLevel(logging.DEBUG)


# ========== Functions ==========
def parse_cmdline_arguments(**kwargs):
    """Parse command line arguments passed to the script.
        All kwargs are passed to ArgumentParser.parse_args().

    :rtype: argparse.Namespace object
    """
    parser = argparse.ArgumentParser(description=DESCRIPTION)
    parser.add_argument(
        "-c",
        "--use-checksums",
        action="store_const",
        dest="extra_rsync_opts",
        const=EXTRA_RSYNC_OPTS["checksum"],
        help="use rsync's checksumming feature to look for changed files",
    )
    parser.add_argument("--debug", action="store_true", help="log debug messages")
    parser.add_argument(
        "-n", "--name", default=None, help="name to give to the snapshot"
    )
    parser.add_argument(
        "-s",
        "--run-post-sync",
        action="store_true",
        help="run sync operation after backup is complete",
    )
    parser.add_argument(
        "-u",
        "--umask",
        type=int,
        default=None,
        help="umask value to use while running backup process",
    )
    parser.add_argument(
        "-v", "--verbose", action="store_true", help="log info messages"
    )
    parser.add_argument("repository", help="repository to back up to", metavar="repo")

    return parser.parse_args(**kwargs)


# ========== Main Script ==========
if __name__ == "__main__":
    args = parse_cmdline_arguments()

    stdout_handler, stderr_handler = rbackup.logging.retrieve_console_handlers(
        debug=args.debug
    )
    syslog.addHandler(stdout_handler)
    syslog.addHandler(stderr_handler)

    dest = args.repository

    try:
        parsed_config = config.parse_configfile()
        repo = Repository(dest)
    except PermissionError as e:
        syslog.critical(e)
        exit(E_PERMISSION)
    else:
        script.do_backup(repo, parsed_config, args)

    if args.run_post_sync:
        syslog.info("Running sync operation")
        os.sync()