backup-tool/app/__main__.py

"""The main entrypoint of the program."""
import argparse
import sys
from parser import read_specs_file


def except_hook(ext_type, value, traceback):
    """Make errors not show the stracktrace to stdout.

    Todo:
        * Replace this with proper error handling
    """
    sys.stderr.write("{}: {}\n".format(ext_type.__name__, value))


# sys.excepthook = except_hook


if __name__ == "__main__":
    # Define parser
    parser = argparse.ArgumentParser(
        description="Backup directories and Docker volumes."
    )
    parser.add_argument(
        "-f",
        "--file",
        action="append",
        dest="file",
        required=True,
        help="File containing spec definitions.",
    )
    parser.add_argument(
        "-j",
        "--json",
        action="store_const",
        const=True,
        default=False,
        help="Print out the parsed specs as JSON and exit",
    )
    parser.add_argument(
        "spec", nargs="*", help="The specs to process. Defaults to all."
    )

    # Parse arguments
    args = parser.parse_args()
    specs = sum((read_specs_file(path) for path in args.file), [])

    # Filter specs if needed
    if args.spec:
        specs = list(filter(lambda s: s.name in args.spec, specs))

    # Dump parsed data as json
    if args.json:
        import json

        # TODO replace this with error handling system
        print(json.dumps([spec.to_dict() for spec in specs], indent=4))

    elif not specs:
        # TODO replace this with error handling system
        print("No specs, exiting.")
        sys.exit(0)

        for spec in specs:
            spec.backup()