python-scripttease/scripttease/library/overlays/ubuntu.py

# Imports

from ..commands import Command, Template
from .common import COMMON_MAPPINGS
from .django import DJANGO_MAPPINGS
from .pgsql import PGSQL_MAPPINGS
from .posix import POSIX_MAPPINGS, Function

# Exports

__all__ = (
    "MAPPINGS",
    "apache",
    "apache_disable_module",
    "apache_disable_site",
    "apache_enable_module",
    "apache_enable_site",
    "apache_reload",
    "apache_restart",
    "apache_start",
    "apache_stop",
    "apache_test",
    "command_exists",
    "service_reload",
    "service_restart",
    "service_start",
    "service_stop",
    "system",
    "system_install",
    "system_reboot",
    "system_update",
    "system_upgrade",
    "system_uninstall",
    "template",
    "Function",
)


def command_exists(name):
    return name in MAPPINGS


def apache(op, **kwargs):
    if op == "reload":
        return apache_reload(**kwargs)
    elif op == "restart":
        return apache_restart(**kwargs)
    elif op == "start":
        return apache_start(**kwargs)
    elif op == "stop":
        return apache_stop(**kwargs)
    elif op == "test":
        return apache_test(**kwargs)
    else:
        raise NameError("Unrecognized or unsupported apache operation: %s" % op)


def apache_disable_module(name, **kwargs):
    kwargs.setdefault("comment", "disable %s apache module" % name)

    return Command("a2dismod %s" % name, **kwargs)


def apache_disable_site(name, **kwargs):
    kwargs.setdefault("comment", "disable %s apache site" % name)

    return Command("a2dissite %s" % name, **kwargs)


def apache_enable_module(name, **kwargs):
    kwargs.setdefault("comment", "enable %s apache module" % name)

    return Command("a2enmod %s" % name, **kwargs)


def apache_enable_site(name, **kwargs):
    kwargs.setdefault("comment", "enable %s apache module" % name)

    return Command("a2ensite %s" % name, **kwargs)


def apache_reload(**kwargs):
    kwargs.setdefault("comment", "reload apache")
    kwargs.setdefault("register", "apache_reloaded")

    return Command("service apache2 reload", **kwargs)


def apache_restart(**kwargs):
    kwargs.setdefault("comment", "restart apache")
    kwargs.setdefault("register", "apache_restarted")

    return Command("service apache2 restart", **kwargs)


def apache_start(**kwargs):
    kwargs.setdefault("comment", "start apache")
    kwargs.setdefault("register", "apache_started")

    return Command("service apache2 start", **kwargs)


def apache_stop(**kwargs):
    kwargs.setdefault("comment", "stop apache")

    return Command("service apache2 stop", **kwargs)


def apache_test(**kwargs):
    kwargs.setdefault("comment", "check apache configuration")
    kwargs.setdefault("register", "apache_checks_out")

    return Command("apachectl configtest", **kwargs)


def service_reload(name, **kwargs):
    kwargs.setdefault("comment", "reload %s service" % name)
    kwargs.setdefault("register", "%s_reloaded" % name)

    return Command("service %s reload" % name, **kwargs)


def service_restart(name, **kwargs):
    kwargs.setdefault("comment", "restart %s service" % name)
    kwargs.setdefault("register", "%s_restarted" % name)

    return Command("service %s restart" % name, **kwargs)


def service_start(name, **kwargs):
    kwargs.setdefault("comment", "start %s service" % name)
    kwargs.setdefault("register", "%s_started" % name)

    return Command("service %s start" % name, **kwargs)


def service_stop(name, **kwargs):
    kwargs.setdefault("comment", "stop %s service" % name)
    kwargs.setdefault("register", "%s_stopped" % name)

    return Command("service %s stop" % name, **kwargs)


def system(op, **kwargs):
    if op == "reboot":
        return system_reboot(**kwargs)
    elif op == "update":
        return system_update(**kwargs)
    elif op == "upgrade":
        return system_upgrade(**kwargs)
    else:
        raise NameError("Unrecognized or unsupported system operation: %s" % op)


def system_install(name, **kwargs):
    kwargs.setdefault("comment", "install system package %s" % name)

    return Command("apt-get install -y %s" % name, **kwargs)


def system_reboot(**kwargs):
    kwargs.setdefault("comment", "reboot the system")

    return Command("reboot", **kwargs)


def system_uninstall(name, **kwargs):
    kwargs.setdefault("comment", "remove system package %s" % name)

    return Command("apt-get uninstall -y %s" % name, **kwargs)


def system_update(**kwargs):
    kwargs.setdefault("comment", "update system package info")

    return Command("apt-get update -y", **kwargs)


def system_upgrade(**kwargs):
    kwargs.setdefault("comment", "upgrade the system")

    return Command("apt-get upgrade -y", **kwargs)


def template(source, target, backup=True, parser=None, **kwargs):
    return Template(source, target, backup=backup, parser=parser, **kwargs)


MAPPINGS = {
    'apache': apache,
    'apache.disable_module': apache_disable_module,
    'apache.disable_site': apache_disable_site,
    'apache.enable_module': apache_enable_module,
    'apache.enable_site': apache_enable_site,
    'apache.reload': apache_reload,
    'apache.restart': apache_restart,
    'apache.start': apache_start,
    'apache.stop': apache_stop,
    'apache.test': apache_test,
    'install': system_install,
    'reboot': system_reboot,
    'reload': service_reload,
    'restart': service_restart,
    'start': service_start,
    'stop': service_stop,
    'system': system,
    'update': system_update,
    'uninstall': system_uninstall,
    'upgrade': system_upgrade,
    'template': template,
}

MAPPINGS.update(COMMON_MAPPINGS)
MAPPINGS.update(DJANGO_MAPPINGS)
MAPPINGS.update(PGSQL_MAPPINGS)
MAPPINGS.update(POSIX_MAPPINGS)