pyinfra 0.11.dev3__py3-none-any.whl → 3.5.1__py3-none-any.whl

pyinfra/modules/pacman.py

@@ -1,60 +0,0 @@
-'''
-Mange pacman packages.
-'''
-
-from pyinfra.api import operation
-
-from .util.packaging import ensure_packages
-
-
-@operation
-def upgrade(state, host):
-    '''
-    Upgrades all pacman packages.
-    '''
-
-    yield 'pacman --noconfirm -Su'
-
-_upgrade = upgrade  # noqa: E305
-
-
-@operation
-def update(state, host):
-    '''
-    Updates pacman repositories.
-    '''
-
-    yield 'pacman -Sy'
-
-_update = update  # noqa: E305
-
-
-@operation
-def packages(
-    state, host,
-    packages=None, present=True,
-    update=False, upgrade=False,
-):
-    '''
-    Add/remove pacman packages.
-
-    + packages: list of packages to ensure
-    + present: whether the packages should be installed
-    + update: run pacman -Sy before installing packages
-    + upgrade: run pacman -Syu before installing packages
-
-    Versions:
-        Package versions can be pinned like pacman: ``<pkg>=<version>``.
-    '''
-
-    if update:
-        yield _update(state, host)
-
-    if upgrade:
-        yield _upgrade(state, host)
-
-    yield ensure_packages(
-        packages, host.fact.pacman_packages, present,
-        install_command='pacman --noconfirm -S',
-        uninstall_command='pacman --noconfirm -R',
-    )

pyinfra/modules/pip.py

@@ -1,99 +0,0 @@
-'''
-Manage pip packages. Compatible globally or inside a virtualenv.
-'''
-
-from pyinfra.api import operation
-from pyinfra.modules import files
-
-from .util.packaging import ensure_packages
-
-
-@operation
-def virtualenv(
-    state, host, path,
-    python=None, site_packages=False, always_copy=False, present=True,
-):
-    '''
-    Add/remove Python virtualenvs.
-
-    + python: python interpreter to use
-    + site_packages: give access to the global site-packages
-    + always_copy: always copy files rather than symlinking
-    + present: whether the virtualenv should exist
-    '''
-
-    if present is False and host.fact.directory(path):
-        yield files.directory(state, host, path, present=False)
-
-    elif present and not host.fact.directory(path):
-        # Create missing virtualenv
-        command = ['virtualenv']
-
-        if python:
-            command.append('-p {0}'.format(python))
-
-        if site_packages:
-            command.append('--system-site-packages')
-
-        if always_copy:
-            command.append('--always-copy')
-
-        command.append(path)
-
-        yield ' '.join(command)
-
-_virtualenv = virtualenv  # noqa
-
-
-@operation
-def packages(
-    state, host,
-    packages=None, present=True, latest=False,
-    requirements=None, pip='pip', virtualenv=None, virtualenv_kwargs=None,
-):
-    '''
-    Install/remove/update pip packages.
-
-    + packages: list of packages to ensure
-    + present: whether the packages should be installed
-    + latest: whether to upgrade packages without a specified version
-    + requirements: location of requirements file to install
-    + pip: name or path of the pip directory to use
-    + virtualenv: root directory of virtualenv to work in
-    + virtualenv_kwargs: dictionary of arguments to pass to ``pip.virtualenv``
-
-    Virtualenv:
-        This will be created if it does not exist already. ``virtualenv_kwargs``
-        will be passed to ``pip.virtualenv`` which can be used to control how
-        the env is created.
-
-    Versions:
-        Package versions can be pinned like pip: ``<pkg>==<version>``.
-    '''
-
-    virtualenv_kwargs = virtualenv_kwargs or {}
-
-    # Ensure any virutalenv
-    if virtualenv:
-        yield _virtualenv(state, host, virtualenv, **virtualenv_kwargs)
-
-        # And update pip path
-        virtualenv = virtualenv.rstrip('/')
-        pip = '{0}/bin/{1}'.format(virtualenv, pip)
-
-    # Install requirements
-    if requirements is not None:
-        yield '{0} install -r {1}'.format(pip, requirements)
-
-    # Handle passed in packages
-    if packages:
-        current_packages = host.fact.pip_packages(pip)
-
-        yield ensure_packages(
-            packages, current_packages, present,
-            install_command='{0} install'.format(pip),
-            uninstall_command='{0} uninstall'.format(pip),
-            upgrade_command='{0} install --upgrade'.format(pip),
-            version_join='==',
-            latest=latest,
-        )

pyinfra/modules/pkg.py

@@ -1,43 +0,0 @@
-'''
-Manage BSD packages and repositories. Note that BSD package names are case-sensitive.
-'''
-
-from pyinfra.api import operation
-
-from .util.packaging import ensure_packages
-
-
-@operation
-def packages(state, host, packages=None, present=True, pkg_path=None):
-    '''
-    Install/remove/update pkg_* packages.
-
-    + packages: list of packages to ensure
-    + present: whether the packages should be installed
-    + pkg_path: the PKG_PATH environment variable to set
-
-    pkg_path:
-        By default this is autogenerated as follows (tested/working for OpenBSD):
-        ``http://ftp.<OS>.org/pub/<OS>/<VERSION>/packages/<ARCH>/``. Note that OpenBSD's
-        official mirrors only hold the latest two versions packages.
-
-        NetBSD/FreeBSD helpfully use their own directory structures, so the default won't
-        work.
-    '''
-
-    if present is True:
-        # Autogenerate package path
-        if not pkg_path:
-            host_os = host.fact.os or ''
-            pkg_path = 'http://ftp.{http}.org/pub/{os}/{version}/packages/{arch}/'.format(
-                http=host_os.lower(),
-                os=host_os,
-                version=host.fact.os_version,
-                arch=host.fact.arch,
-            )
-
-    yield ensure_packages(
-        packages, host.fact.pkg_packages, present,
-        install_command='PKG_PATH={0} pkg_add'.format(pkg_path),
-        uninstall_command='pkg_delete',
-    )

pyinfra/modules/postgresql.py

@@ -1,245 +0,0 @@
-'''
-The PostgreSQL modules manage PostgreSQL databases, users and privileges.
-
-Requires the ``psql`` CLI executable on the target host(s).
-
-All operations in this module take four optional global arguments:
-    + ``postgresql_user``: the username to connect to postgresql to
-    + ``postgresql_password``: the password for the connecting user
-    + ``postgresql_host``: the hostname of the server to connect to
-    + ``postgresql_port``: the port of the server to connect to
-'''
-
-from pyinfra.api import operation
-from pyinfra.facts.postgresql import make_execute_psql_command, make_psql_command
-
-
-@operation
-def sql(
-    state, host, sql,
-    database=None,
-    # Details for speaking to PostgreSQL via `psql` CLI
-    postgresql_user=None, postgresql_password=None,
-    postgresql_host=None, postgresql_port=None,
-):
-    '''
-    Execute arbitrary SQL against PostgreSQL.
-
-    + sql: SQL command(s) to execute
-    + database: optional database to execute against
-    + postgresql_*: global module arguments, see above
-    '''
-
-    yield make_execute_psql_command(
-        sql,
-        database=database,
-        user=postgresql_user,
-        password=postgresql_password,
-        host=postgresql_host,
-        port=postgresql_port,
-    )
-
-
-@operation
-def role(
-    state, host, name,
-    present=True,
-    password=None, login=True, superuser=False, inherit=False,
-    createdb=False, createrole=False, replication=False, connection_limit=None,
-    # Details for speaking to PostgreSQL via `psql` CLI
-    postgresql_user=None, postgresql_password=None,
-    postgresql_host=None, postgresql_port=None,
-):
-    '''
-    Add/remove PostgreSQL roles.
-
-    + name: name of the role
-    + present: whether the role should be present or absent
-    + password: the password for the role
-    + login: whether the role can login
-    + superuser: whether role will be a superuser
-    + inherit: whether the role inherits from other roles
-    + createdb: whether the role is allowed to create databases
-    + createrole: whether the role is allowed to create new roles
-    + replication: whether this role is allowed to replicate
-    + connection_limit: the connection limit for the role
-    + postgresql_*: global module arguments, see above
-
-    Updates:
-        pyinfra will not attempt to change existing roles - it will either
-        create or drop roles, but not alter them (if the role exists this
-        operation will make no changes).
-    '''
-
-    roles = host.fact.postgresql_roles(
-        postgresql_user, postgresql_password,
-        postgresql_host, postgresql_port,
-    )
-
-    is_present = name in roles
-
-    # User not wanted?
-    if not present:
-        if is_present:
-            yield make_execute_psql_command(
-                'DROP ROLE {0}'.format(name),
-                user=postgresql_user,
-                password=postgresql_password,
-                host=postgresql_host,
-                port=postgresql_port,
-            )
-        return
-
-    # If we want the user and they don't exist
-    if not is_present:
-        sql_bits = ['CREATE ROLE {0}'.format(name)]
-
-        for key, value in (
-            ('LOGIN', login),
-            ('SUPERUSER', superuser),
-            ('INHERIT', inherit),
-            ('CREATEDB', createdb),
-            ('CREATEROLE', createrole),
-            ('REPLICATION', replication),
-        ):
-            if value:
-                sql_bits.append(key)
-
-        if connection_limit:
-            sql_bits.append('CONNECTION LIMIT {0}'.format(connection_limit))
-
-        if password:
-            sql_bits.append("PASSWORD '{0}'".format(password))
-
-        yield make_execute_psql_command(
-            ' '.join(sql_bits),
-            user=postgresql_user,
-            password=postgresql_password,
-            host=postgresql_host,
-            port=postgresql_port,
-        )
-
-
-@operation
-def database(
-    state, host, name,
-    present=True, owner=None,
-    template=None, encoding=None,
-    lc_collate=None, lc_ctype=None, tablespace=None,
-    connection_limit=None,
-    # Details for speaking to PostgreSQL via `psql` CLI
-    postgresql_user=None, postgresql_password=None,
-    postgresql_host=None, postgresql_port=None,
-):
-    '''
-    Add/remove PostgreSQL databases.
-
-    + name: name of the database
-    + present: whether the database should exist or not
-    + owner: the PostgreSQL role that owns the database
-    + template: name of the PostgreSQL template to use
-    + encoding: encoding of the database
-    + lc_collate: lc_collate of the database
-    + lc_ctype: lc_ctype of the database
-    + tablespace: the tablespace to use for the template
-    + connection_limit: the connection limit to apply to the database
-    + postgresql_*: global module arguments, see above
-
-    Updates:
-        pyinfra will not attempt to change existing databases - it will either
-        create or drop databases, but not alter them (if the db exists this
-        operation will make no changes).
-    '''
-
-    current_databases = host.fact.postgresql_databases(
-        postgresql_user, postgresql_password,
-        postgresql_host, postgresql_port,
-    )
-
-    is_present = name in current_databases
-
-    if not present:
-        if is_present:
-            yield make_execute_psql_command(
-                'DROP DATABASE {0}'.format(name),
-                user=postgresql_user,
-                password=postgresql_password,
-                host=postgresql_host,
-                port=postgresql_port,
-            )
-        return
-
-    # We want the database but it doesn't exist
-    if present and not is_present:
-        sql_bits = ['CREATE DATABASE {0}'.format(name)]
-
-        for key, value in (
-            ('OWNER', owner),
-            ('TEMPLATE', template),
-            ('ENCODING', encoding),
-            ('LC_COLLATE', lc_collate),
-            ('LC_CTYPE', lc_ctype),
-            ('TABLESPACE', tablespace),
-            ('CONNECTION LIMIT', connection_limit),
-        ):
-            if value:
-                sql_bits.append('{0} {1}'.format(key, value))
-
-        yield make_execute_psql_command(
-            ' '.join(sql_bits),
-            user=postgresql_user,
-            password=postgresql_password,
-            host=postgresql_host,
-            port=postgresql_port,
-        )
-
-
-@operation
-def dump(
-    state, host,
-    remote_filename, database=None,
-    # Details for speaking to PostgreSQL via `psql` CLI
-    postgresql_user=None, postgresql_password=None,
-    postgresql_host=None, postgresql_port=None,
-):
-    '''
-    Dump a PostgreSQL database into a ``.sql`` file. Requires ``mysqldump``.
-
-    + database: name of the database to dump
-    + remote_filename: name of the file to dump the SQL to
-    + postgresql_*: global module arguments, see above
-    '''
-
-    yield '{0} > {1}'.format(make_psql_command(
-        executable='pg_dump',
-        database=database,
-        user=postgresql_user,
-        password=postgresql_password,
-        host=postgresql_host,
-        port=postgresql_port,
-    ), remote_filename)
-
-
-@operation
-def load(
-    state, host,
-    remote_filename, database=None,
-    # Details for speaking to PostgreSQL via `psql` CLI
-    postgresql_user=None, postgresql_password=None,
-    postgresql_host=None, postgresql_port=None,
-):
-    '''
-    Load ``.sql`` file into a database.
-
-    + database: name of the database to import into
-    + remote_filename: the filename to read from
-    + postgresql_*: global module arguments, see above
-    '''
-
-    yield '{0} < {1}'.format(make_psql_command(
-        database=database,
-        user=postgresql_user,
-        password=postgresql_password,
-        host=postgresql_host,
-        port=postgresql_port,
-    ), remote_filename)

pyinfra/modules/puppet.py

@@ -1,20 +0,0 @@
-from pyinfra.api import operation
-
-
-@operation
-def agent(state, host, server=None, port=None):
-    '''
-    Run puppet agent
-
-    + server: master server URL
-    + port: puppet master port
-    '''
-
-    args = []
-
-    if server:
-        args.append('--server=%s' % server)
-    if port:
-        args.append('--masterport=%s' % port)
-
-    yield 'puppet agent -t %s' % ' '.join(args)

pyinfra/modules/python.py

@@ -1,37 +0,0 @@
-'''
-The Python module allows you to execute Python code within the context of a deploy.
-'''
-
-from pyinfra.api import operation
-
-
-@operation
-def call(state, host, func, *args, **kwargs):
-    '''
-    Execute a Python function within a deploy.
-
-    + func: the function to execute
-    + args: additional arguments to pass to the function
-    + kwargs: additional keyword arguments to pass to the function
-
-    Callback functions arge passed the state, host, and any args/kwargs passed
-    into the operation directly, eg:
-
-    .. code:: python
-
-        def my_callback(state, host, hello=None):
-            ...
-
-        python.call(my_callback, hello='world')
-
-    '''
-
-    yield (func, args, kwargs)
-
-
-@operation
-def raise_exception(state, host, exception_class, *args, **kwargs):
-    def raise_exc(*args, **kwargs):  # pragma: no cover
-        raise exception_class(*args, **kwargs)
-
-    yield (raise_exc, args, kwargs)