entry.py

"""
HomeCU tool for managing hosting at AWS.
"""

import os
import shellish
from . import deploy, stack, cumanage, logs, sched, certificates, ecr
from shellish.command import contrib


class RootCommand(shellish.Command):
    """ Manage Odyssey AWS hosting. """

    name = 'hosting'

    def setup_args(self, parser):
        self.add_subcommand(deploy.DeployCommand)
        self.add_subcommand(ecr.ECRCommand)
        self.add_subcommand(stack.StackCommand)
        self.add_subcommand(cumanage.CUManageCommand)
        self.add_subcommand(sched.SchedCommand)
        self.add_subcommand(logs.LogsCommand)
        self.add_subcommand(certificates.CertificateCommand)
        self.add_subcommand(DebugCommand)

    def prerun(self, args):
        """ Add the interactive commands just before it goes to the prompt so
        they don't show up in the --help from the commands line. """
        self.add_subcommand(contrib.Exit)
        self.add_subcommand(contrib.Help)
        self.add_subcommand(contrib.Reset)
        self.add_subcommand(contrib.Pager)

    def run(self, args):
        """ Run interactive mode if we are called without args. """
        self.session.run_loop()


class DebugCommand(shellish.Command):
    """ Debug command shell. """

    name = 'debug'

    def run(self, args):
        os.system('bash')


RootCommand()()