manage.py

#!/usr/bin/python3
import argparse
import os
import sys
from sys import exit as x

from py.colours import Colours
from py.dcompose import DCompose
from py.profile import Profile
from py.runner import Runner
from py.snapshots import Snapshots
from py.util import pprint

parser = argparse.ArgumentParser(
    description=os.path.basename(__file__).title()
    + ": "
    + "dq-dev, daiquiri docker compose dev setup",
    epilog="If used without arg, profile list is displayed\n",
    formatter_class=argparse.RawDescriptionHelpFormatter,
)
parser.add_argument(
    "-b",
    "--build",
    type=str,
    nargs="*",
    default=None,
    help="build a profile's containers, exit when done",
)
parser.add_argument(
    "-bnc",
    "--build_no_cache",
    type=str,
    nargs="*",
    default=None,
    help="build a profile's containers without using cache, exit when done",
)
parser.add_argument(
    "-r",
    "--run",
    type=str,
    nargs="*",
    default=None,
    help="run a profile's containers, build if necessary",
)
parser.add_argument(
    "-p",
    "--stop",
    type=str,
    nargs="*",
    default=None,
    help="stop profile's running containers",
)
parser.add_argument(
    "-d",
    "--down",
    type=str,
    nargs="*",
    default=None,
    help=(
        "stop and remove profile's running containers, "
        + "remove docker's volumes, keep folders containing the "
        + "volume data, they can be reused on next run"
    ),
)
parser.add_argument(
    "-rmi",
    "--remove_images",
    action="store_true",
    default=False,
    help='remove images when running "down", use in combination "-d -rmi"\n\n',
)
parser.add_argument(
    "-rmn",
    "--remove_network",
    action="store_true",
    default=False,
    help="remove daiquiri containers' network",
)
parser.add_argument(
    "-g",
    "--tail_logs",
    type=str,
    nargs="*",
    default=None,
    help="tail docker compose logs\n\n",
)
parser.add_argument(
    "-c",
    "--create_profile",
    type=str,
    default=None,
    help="create a new profile with the default settings",
)
parser.add_argument(
    "-s",
    "--set_profile",
    type=str,
    default=None,
    help="set profile to active"
)
parser.add_argument(
    "-e",
    "--render",
    type=str,
    nargs="*",
    default=None,
    help="only render docker-compose.yaml for profile",
)
parser.add_argument(
    "-a",
    "--display_profile",
    type=str,
    nargs="*",
    default=None,
    help="display currently active profile",
)
parser.add_argument(
    "--list_snapshots",
    action="store_true",
    default=False,
    help="list currently saved snapshots",
)
parser.add_argument(
    "--save_snapshot",
    type=str,
    nargs="*",
    default=None,
    help="save current db and config to snapshot",
)
parser.add_argument(
    "--restore_snapshot",
    type=str,
    nargs="*",
    default=None,
    help="restore saved snapshot",
)
parser.add_argument(
    "-n",
    "--dry_run",
    action="store_true",
    default=False,
    help=(
        "do not run any docker-compose commands nor "
        + "save rendered docker-compose.yaml, just print them"
    ),
)
args = parser.parse_args()


if __name__ == "__main__":
    col = Colours()
    prof = Profile(args)
    conf = prof.conf
    dco = DCompose(conf, prof)
    snap = Snapshots(conf)

    if args.create_profile is not None:
        prof.create(args.create_profile)
        x()

    if args.set_profile is not None:
        prof.set(args.set_profile)
        x()

    if not prof.exists:
        print(
            col.red("There is no active profile!\n\n") +
            col.gre("Please activate a profile with \n") +
            col.gre("    python manage.py -s <profile_name>\n") +
            col.gre("or, first, create a new profile with\n") +
            col.gre("    python manage.py -c <profile_name>\n") +
            col.gre("and then activate it.")
        )

    if len(sys.argv) <=1 or not prof.is_active():
        prof.list()
        x()

    if args.display_profile is not None:
        c = prof.read_profile_config(conf["args"]["display_profile"])
        print("Currently set profile", col.yel(c["name"]))
        pprint(c)

    if args.render is not None:
        dco.render_dc_yaml(conf["args"]["render"])
        dco.render_dockerfile_templates()

    if args.build is not None:
        dco.render_dc_yaml(conf["args"]["run"])
        dco.render_dockerfile_templates()
        run = Runner(conf)
        run.build()

    if args.build_no_cache is not None:
        dco.render_dc_yaml(conf["args"]["run"])
        dco.render_dockerfile_templates()
        run = Runner(conf)
        run.build_no_cache()

    if args.run is not None:
        dco.render_dc_yaml(conf["args"]["run"])
        dco.render_dockerfile_templates()
        run = Runner(conf)
        run.start()

    if args.stop is not None:
        run = Runner(conf)
        run.stop()

    if args.down is not None:
        run = Runner(conf)
        run.down()

    if conf["args"]["tail_logs"] is True:
        run = Runner(conf)
        run.tail_logs()

    if args.remove_images is True:
        run = Runner(conf)
        run.remove_images()

    if args.remove_network is True:
        run = Runner(conf)
        run.remove_network()

    if args.list_snapshots is True:
        snap.list_snapshots()

    if conf["args"]["save_snapshot"] is not None:
        snap.save_snapshot()

    if conf["args"]["restore_snapshot"] is not None:
        snap.restore_snapshot()