Added overlay for Cent OS.

4 years ago · b191df4209
parent ddaa8a4f87
commit b191df4209
3 changed files with 379 additions and 0 deletions
--- a/docs/generate_command_signatures.py
+++ b/docs/generate_command_signatures.py
@ -11,6 +11,7 @@ sys.path.append("../")

 # Import overlays
 from scripttease.library.overlays.common import COMMON_MAPPINGS
+from scripttease.library.overlays.centos import MAPPINGS as CENTOS_MAPPINGS
 from scripttease.library.overlays.django import DJANGO_MAPPINGS
 from scripttease.library.overlays.pgsql import PGSQL_MAPPINGS
 from scripttease.library.overlays.posix import POSIX_MAPPINGS
@ -110,6 +111,15 @@ print_heading("POSIX")
 print_description("Posix commands form the basis of overlays for nix platforms.")
 print_mapping(POSIX_MAPPINGS)

+exclude_from_centos = COMMON_MAPPINGS.copy()
+exclude_from_centos.update(DJANGO_MAPPINGS)
+exclude_from_centos.update(PGSQL_MAPPINGS)
+exclude_from_centos.update(POSIX_MAPPINGS)
+print_heading("Cent OS")
+print_description("The Cent OS overlay incorporates commands specific to that platform as well as commands from "
+                  "common, Django, Postgres, and POSIX.")
+print_mapping(CENTOS_MAPPINGS, excludes=exclude_from_centos)
+
 exclude_from_ubuntu = COMMON_MAPPINGS.copy()
 exclude_from_ubuntu.update(DJANGO_MAPPINGS)
 exclude_from_ubuntu.update(PGSQL_MAPPINGS)
--- a/scripttease/library/overlays/centos.py
+++ b/scripttease/library/overlays/centos.py
@ -0,0 +1,279 @@
+# Imports
+
+from superpython.utils import split_csv
+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_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",
+    "user",
+    "Function",
+)
+
+
+def command_exists(name):
+    """Indicates whether a given command exists in this overaly.
+
+    :param name: The name of the command.
+    :type name: str
+
+    :rtype: bool
+
+    """
+    return name in MAPPINGS
+
+
+def apache(op, **kwargs):
+    """Execute an Apache-related command.
+
+    - op (str): The operation to perform; reload, restart, start, stop, test.
+
+    """
+    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_reload(**kwargs):
+    kwargs.setdefault("comment", "reload apache")
+    kwargs.setdefault("register", "apache_reloaded")
+
+    return Command("apachectl –k reload", **kwargs)
+
+
+def apache_restart(**kwargs):
+    kwargs.setdefault("comment", "restart apache")
+    kwargs.setdefault("register", "apache_restarted")
+
+    return Command("apachectl –k restart", **kwargs)
+
+
+def apache_start(**kwargs):
+    kwargs.setdefault("comment", "start apache")
+    kwargs.setdefault("register", "apache_started")
+
+    return Command("apachectl –k start", **kwargs)
+
+
+def apache_stop(**kwargs):
+    kwargs.setdefault("comment", "stop apache")
+
+    return Command("apachectl –k 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):
+    """Reload a service.
+
+    - name (str): The service name.
+
+    """
+    kwargs.setdefault("comment", "reload %s service" % name)
+    kwargs.setdefault("register", "%s_reloaded" % name)
+
+    return Command("systemctl reload %s" % name, **kwargs)
+
+
+def service_restart(name, **kwargs):
+    """Restart a service.
+
+    - name (str): The service name.
+
+    """
+    kwargs.setdefault("comment", "restart %s service" % name)
+    kwargs.setdefault("register", "%s_restarted" % name)
+
+    return Command("ssystemctl restart %s" % name, **kwargs)
+
+
+def service_start(name, **kwargs):
+    """Start a service.
+
+    - name (str): The service name.
+
+    """
+    kwargs.setdefault("comment", "start %s service" % name)
+    kwargs.setdefault("register", "%s_started" % name)
+
+    return Command("systemctl start %s" % name, **kwargs)
+
+
+def service_stop(name, **kwargs):
+    """Stop a service.
+
+    - name (str): The service name.
+
+    """
+    kwargs.setdefault("comment", "stop %s service" % name)
+    kwargs.setdefault("register", "%s_stopped" % name)
+
+    return Command("systemctl stop %s" % name, **kwargs)
+
+
+def system(op, **kwargs):
+    """Perform a system operation.
+
+    - op (str): The operation to perform; reboot, update, upgrade.
+
+    """
+    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):
+    """Install a system-level package.
+
+    - name (str): The name of the package to install.
+
+    """
+    kwargs.setdefault("comment", "install system package %s" % name)
+
+    return Command("yum install -y %s" % name, **kwargs)
+
+
+def system_reboot(**kwargs):
+    kwargs.setdefault("comment", "reboot the system")
+
+    return Command("reboot", **kwargs)
+
+
+def system_uninstall(name, **kwargs):
+    """Uninstall a system-level package.
+
+    - name (str): The name of the package to uninstall.
+
+    """
+    kwargs.setdefault("comment", "remove system package %s" % name)
+
+    return Command("yum remove -y %s" % name, **kwargs)
+
+
+def system_update(**kwargs):
+    kwargs.setdefault("comment", "update system package info")
+
+    return Command("yum check-update", **kwargs)
+
+
+def system_upgrade(**kwargs):
+    kwargs.setdefault("comment", "upgrade the system")
+
+    return Command("yum update -y", **kwargs)
+
+
+def template(source, target, backup=True, parser=None, **kwargs):
+    """Create a file from a template.
+
+    - source (str): The path to the template file.
+    - target (str): The path to where the new file should be created.
+    - backup (bool): Indicates whether a backup should be made if the target file already exists.
+    - parser (str): The parser to use ``jinja`` (the default) or ``simple``.
+
+    """
+    return Template(source, target, backup=backup, parser=parser, **kwargs)
+
+
+def user(name, groups=None, home=None, op="add", password=None, **kwargs):
+    """Create or remove a user.
+
+    - name (str): The user name.
+    - groups (str | list): A list of groups to which the user should belong.
+    - home (str): The path to the user's home directory.
+    - op (str); The operation to perform; ``add`` or ``remove``.
+    - password (str): The user's password. (NOT IMPLEMENTED)
+
+    """
+    if op == "add":
+        kwargs.setdefault("comment", "create a user named %s" % name)
+
+        commands = list()
+
+        a = list()
+        a.append('adduser %s' % name)
+        if home is not None:
+            a.append("--home %s" % home)
+
+        commands.append(Command(" ".join(a), **kwargs))
+
+        if type(groups) is str:
+            groups = split_csv(groups, smart=False)
+
+        if type(groups) in [list, tuple]:
+            for group in groups:
+                commands.append(Command("gpasswd -a %s %s" % (name, group), **kwargs))
+
+        a = list()
+        for c in commands:
+            a.append(c.get_statement(suppress_comment=True))
+
+        return Command("\n".join(a), **kwargs)
+    elif op == "remove":
+        kwargs.setdefault("comment", "remove a user named %s" % name)
+        return Command("userdel -r %s" % name, **kwargs)
+    else:
+        raise NameError("Unsupported or unrecognized operation: %s" % op)
+
+
+MAPPINGS = {
+    'apache': apache,
+    'install': system_install,
+    'reboot': system_reboot,
+    'reload': service_reload,
+    'restart': service_restart,
+    'start': service_start,
+    'stop': service_stop,
+    'system': system,
+    'template': template,
+    'update': system_update,
+    'uninstall': system_uninstall,
+    'upgrade': system_upgrade,
+    'user': user,
+}
+
+MAPPINGS.update(COMMON_MAPPINGS)
+MAPPINGS.update(DJANGO_MAPPINGS)
+MAPPINGS.update(PGSQL_MAPPINGS)
+MAPPINGS.update(POSIX_MAPPINGS)
--- a/tests/test_library_overlays_centos.py
+++ b/tests/test_library_overlays_centos.py
@ -0,0 +1,90 @@
+import pytest
+from scripttease.library.commands.templates import Template
+from scripttease.library.overlays.centos import *
+
+
+def test_apache():
+    c = apache("reload")
+    assert "apachectl –k reload" in c.get_statement()
+
+    c = apache("restart")
+    assert "apachectl –k restart" in c.get_statement()
+
+    c = apache("start")
+    assert "apachectl –k start" in c.get_statement()
+
+    c = apache("stop")
+    assert "apachectl –k stop" in c.get_statement()
+
+    c = apache("test")
+    assert "apachectl configtest" in c.get_statement()
+
+    with pytest.raises(NameError):
+        apache("nonexistent")
+
+
+def test_command_exists():
+    assert command_exists("apache") is True
+    assert command_exists("nonexistent") is False
+
+
+def test_service_reload():
+    c = service_reload("postfix")
+    assert "systemctl reload postfix" in c.get_statement()
+
+
+def test_service_restart():
+    c = service_restart("postfix")
+    assert "systemctl restart postfix" in c.get_statement()
+
+
+def test_service_start():
+    c = service_start("postfix")
+    assert "systemctl start postfix" in c.get_statement()
+
+
+def test_service_stop():
+    c = service_stop("postfix")
+    assert "systemctl stop postfix" in c.get_statement()
+
+
+def test_system():
+    c = system("reboot")
+    assert "reboot" in c.get_statement()
+
+    c = system("update")
+    assert "yum check-update" in c.get_statement()
+
+    c = system("upgrade")
+    assert "yum update -y" in c.get_statement()
+
+    with pytest.raises(NameError):
+        system("nonexistent")
+
+
+def test_system_install():
+    c = system_install("vim")
+    assert "yum install -y vim" in c.get_statement()
+
+
+def test_system_uninstall():
+    c = system_uninstall("lftp")
+    assert "yum remove -y lftp" in c.get_statement()
+
+
+def test_template():
+    t = template("/path/to/source.txt", "/path/to/target.txt")
+    assert isinstance(t, Template)
+
+
+def test_user():
+    statement = user("deploy", groups="sudo", home="/path/to/deploy/root").get_statement()
+    assert "adduser deploy" in statement
+    assert "--home" in statement
+    assert "gpasswd -a deploy sudo" in statement
+
+    statement = user("deploy", op="remove").get_statement()
+    assert "userdel -r deploy" in statement
+
+    with pytest.raises(NameError):
+        user("deploy", op="unsupported")