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

pyinfra/operations/flatpak.py

@@ -0,0 +1,95 @@
+"""
+Manage flatpak packages. See https://www.flatpak.org/
+"""
+
+from __future__ import annotations
+
+from pyinfra import host
+from pyinfra.api import operation
+from pyinfra.facts.flatpak import FlatpakPackages
+
+
+@operation()
+def packages(
+    packages: str | list[str] | None = None,
+    remote: str | None = None,
+    present=True,
+):
+    """
+    Install/remove a flatpak package
+
+    + packages: List of packages
+    + remote: Source to install the application or runtime from
+    + present: whether the package should be installed
+
+    **Examples:**
+
+    .. code:: python
+
+        from pyinfra.operations import flatpak
+        # Install vlc flatpak
+        flatpak.package(
+            name="Install vlc",
+            packages="org.videolan.VLC",
+        )
+
+        # Install vlc flatpak from flathub
+        flatpak.package(
+            name="Install vlc",
+            packages="org.videolan.VLC",
+            remote="flathub",
+        )
+
+        # Install multiple flatpaks
+        flatpak.package(
+            name="Install vlc and kodi",
+            packages=["org.videolan.VLC", "tv.kodi.Kodi"],
+        )
+
+        # Remove vlc
+        flatpak.package(
+            name="Remove vlc",
+            packages="org.videolan.VLC",
+            present=False,
+        )
+    """
+
+    if packages is None:
+        return
+
+    if isinstance(packages, str):
+        packages = [packages]
+
+    flatpak_packages = host.get_fact(FlatpakPackages)
+
+    install_packages = []
+    remove_packages = []
+
+    if remote is None:
+        remote = ""
+    else:
+        # ensure we have a space between the remote and packages
+        remote = remote.strip() + " "
+
+    for package in packages:
+        # it's installed
+        if package in flatpak_packages:
+            if not present:
+                # we don't want it
+                remove_packages.append(package)
+
+        # it's not installed
+        if package not in flatpak_packages:
+            # we want it
+            if present:
+                install_packages.append(package)
+
+            # we don't want it
+            else:
+                host.noop(f"flatpak package {package} is not installed")
+
+    if install_packages:
+        yield f"flatpak install --noninteractive {remote}{' '.join(install_packages)}"
+
+    if remove_packages:
+        yield f"flatpak uninstall --noninteractive {' '.join(remove_packages)}"

pyinfra/operations/freebsd/__init__.py

@@ -0,0 +1,12 @@
+# This file only exists to support:
+# from pyinfra.operations import freebsd
+# freebsd.X.Y
+
+from glob import glob
+from os import path
+
+module_filenames = glob(path.join(path.dirname(__file__), "*.py"))
+module_names = [path.basename(name)[:-3] for name in module_filenames]
+__all__ = [name for name in module_names if name != "__init__"]
+
+from . import *  # noqa

pyinfra/operations/freebsd/freebsd_update.py

@@ -0,0 +1,70 @@
+"""
+Fetch and install binary updates to FreeBSD.
+"""
+
+from __future__ import annotations
+
+from typing_extensions import List, Optional, Union
+
+from pyinfra.api import QuoteString, StringCommand, operation
+
+
+@operation()
+def update(
+    force: bool = False,
+    basedir: Optional[str] = None,
+    workdir: Optional[str] = None,
+    conffile: Optional[str] = None,
+    jail: Optional[str] = None,
+    key: Optional[str] = None,
+    currently_running: Optional[str] = None,
+    server: Optional[str] = None,
+):
+    """
+    Based on the currently installed world and the configuration options set, fetch
+    all available binary updates and install them.
+
+    + force: See ``-F`` in ``freebsd-update(8)``.
+    + basedir: See ``-b`` in ``freebsd-update(8)``.
+    + workdir: See ``-d`` in ``freebsd-update(8)``.
+    + conffile: See ``-f`` in ``freebsd-update(8)``.
+    + jail: See ``-j`` in ``freebsd-update(8)``.
+    + key: See ``-k`` in ``freebsd-update(8)``.
+    + currently_running: See ``--currently-running`` in ``freebsd-update(8)``.
+    + server: See ``-s`` in ``freebsd-update(8)``.
+
+    **Example:**
+
+    .. code:: python
+
+        freebsd_update.update()
+    """
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.extend(["PAGER=cat", "freebsd-update", "--not-running-from-cron"])
+
+    if force:
+        args.append("-F")
+
+    if basedir is not None:
+        args.extend(["-b", QuoteString(basedir)])
+
+    if workdir is not None:
+        args.extend(["-d", QuoteString(workdir)])
+
+    if conffile is not None:
+        args.extend(["-f", QuoteString(conffile)])
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    if key is not None:
+        args.extend(["-k", QuoteString(key)])
+
+    if server is not None:
+        args.extend(["-s", QuoteString(server)])
+
+    args.extend(["fetch", "install"])
+
+    yield StringCommand(*args)

pyinfra/operations/freebsd/pkg.py

@@ -0,0 +1,219 @@
+"""
+Manage FreeBSD packages.
+"""
+
+from __future__ import annotations
+
+from typing_extensions import List, Optional, Union
+
+from pyinfra import host
+from pyinfra.api import QuoteString, StringCommand, operation
+from pyinfra.facts.freebsd import PkgPackage
+
+
+@operation()
+def update(jail: Optional[str] = None, force: bool = False, reponame: Optional[str] = None):
+    """
+    Update the local catalogues of the enabled package repositories.
+
+    + jail: See ``-j`` in ``pkg(8)``.
+    + force: See ``-f`` in ``pkg-update(8)``.
+    + reponame: See ``-r`` in ``pkg-update(8)``
+
+    **Examples:**
+
+    .. code:: python
+
+        # host
+        pkg.update()
+
+        # jail
+        pkg.update(
+            jail="nginx"
+        )
+    """
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.append("pkg")
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    args.extend(["update"])
+
+    if force:
+        args.append("-f")
+
+    if reponame is not None:
+        args.extend(["-r", QuoteString(reponame)])
+
+    yield StringCommand(*args)
+
+
+@operation()
+def upgrade(jail: Optional[str] = None, force: bool = False, reponame: Optional[str] = None):
+    """
+    Perform upgrades of package software distributions.
+
+    + jail: See ``-j`` in ``pkg(8)``.
+    + force: See ``-f`` in ``pkg-upgrade(8)``.
+    + reponame: See ``-r`` in ``pkg-upgrade(8)``.
+
+    **Examples:**
+
+    .. code:: python
+
+        # host
+        pkg.upgrade()
+
+        # jail
+        pkg.upgrade(
+            jail="nginx"
+        )
+    """
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.append("pkg")
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    args.extend(["upgrade", "-y"])
+
+    if force:
+        args.append("-f")
+
+    if reponame is not None:
+        args.extend(["-r", QuoteString(reponame)])
+
+    yield StringCommand(*args)
+
+
+@operation()
+def install(package: str, jail: Optional[str] = None, reponame: Optional[str] = None):
+    """
+    Install packages from remote packages repositories or local archives.
+
+    + package: Package to install.
+    + jail: See ``-j`` in ``pkg(8)``.
+    + reponame: See ``-r`` in ``pkg-install(8)``.
+
+    **Example:**
+
+    .. code:: python
+
+        pkg.install("nginx")
+    """
+
+    if host.get_fact(PkgPackage, package=package, jail=jail):
+        host.noop(f"Package '{package}' already installed")
+        return
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.append("pkg")
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    args.extend(["install", "-y"])
+
+    if reponame is not None:
+        args.extend(["-r", QuoteString(reponame)])
+
+    args.extend(["--", QuoteString(package)])
+
+    yield StringCommand(*args)
+
+
+@operation()
+def remove(package: str, jail: Optional[str] = None):
+    """
+    Deletes packages from the database and the system.
+
+    + package: Package to remove.
+    + jail: See ``-j`` in ``pkg(8)``.
+
+    **Example:**
+
+    .. code:: python
+
+        pkg.remove("nginx")
+    """
+
+    if not host.get_fact(PkgPackage, package=package, jail=jail):
+        host.noop(f"Package '{package}' cannot be found")
+        return
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.append("pkg")
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    args.extend(["remove", "-y"])
+
+    args.extend(["--", QuoteString(package)])
+
+    yield StringCommand(*args)
+
+
+@operation()
+def autoremove(jail: Optional[str] = None):
+    """
+    Remove orphan packages.
+
+    + jail: See ``-j`` in ``pkg(8)``.
+
+    **Example:**
+
+    .. code:: python
+
+        pkg.autoremove()
+    """
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.append("pkg")
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    args.extend(["autoremove", "-y"])
+
+    yield StringCommand(*args)
+
+
+@operation()
+def clean(all_pkg: bool = False, jail: Optional[str] = None):
+    """
+    Clean the local cache of fetched remote packages.
+
+    + all_pkg: See ``-a`` in ``pkg-clean(8)``.
+    + jail: See ``-j`` in ``pkg(8)``.
+
+    **Example:**
+
+    .. code:: python
+
+        pkg.clean(
+            all_pkg=True
+        )
+    """
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.append("pkg")
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    args.extend(["clean", "-y"])
+
+    if all_pkg:
+        args.append("-a")
+
+    yield StringCommand(*args)

pyinfra/operations/freebsd/service.py

@@ -0,0 +1,116 @@
+"""
+Manage FreeBSD services.
+"""
+
+from __future__ import annotations
+
+from typing_extensions import List, Optional, Union
+
+from pyinfra import host
+from pyinfra.api import QuoteString, StringCommand, operation
+from pyinfra.api.exceptions import OperationValueError
+from pyinfra.facts.freebsd import ServiceScript, ServiceStatus
+
+SRV_STARTED: str = "started"
+SRV_STOPPED: str = "stopped"
+SRV_RESTARTED: str = "restarted"
+SRV_RELOADED: str = "reloaded"
+SRV_CUSTOM: str = "custom"
+
+
+@operation()
+def service(
+    srvname: str,
+    jail: Optional[str] = None,
+    srvstate: str = SRV_STARTED,
+    command: Optional[Union[str, List[str]]] = None,
+    environment: Optional[List[str]] = None,
+    verbose: bool = False,
+):
+    """
+    Control (start/stop/etc.) ``rc(8)`` scripts.
+
+    + srvname: Service.
+    + jail: See ``-j`` in ``service(8)``.
+    + srvstate: Desire state of the service.
+    + command: When ``srvstate`` is ``custom``, the command to execute.
+    + environment: See ``-E`` in ``service(8)``.
+    + verbose: See ``-v`` in ``service(8)``.
+
+    States:
+        There are a few states you can use to manipulate the service:
+
+        - started: The service must be started.
+        - stopped: The service must be stopped.
+        - restarted: The service must be restarted.
+        - reloaded: The service must be reloaded.
+        - custom: Run a custom command for this service.
+
+    **Examples:**
+
+    .. code:: python
+
+        # Start a service.
+        service.service(
+            "beanstalkd",
+            srvstate="started"
+        )
+
+        # Execute a custom command.
+        service.service(
+            "sopel",
+            srvstate="custom",
+            command="configure"
+        )
+    """
+
+    if not host.get_fact(ServiceScript, srvname=srvname, jail=jail):
+        host.noop(f"Cannot find rc(8) script '{srvname}'")
+        return
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.append("service")
+
+    if verbose:
+        args.append("-v")
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    if environment is not None:
+        for env_var in environment:
+            args.extend(["-E", QuoteString(env_var)])
+
+    if srvstate == SRV_STARTED:
+        if host.get_fact(ServiceStatus, srvname=srvname, jail=jail):
+            host.noop(f"Service '{srvname}' already started")
+            return
+
+        args.extend([QuoteString(srvname), "start"])
+
+    elif srvstate == SRV_STOPPED:
+        if not host.get_fact(ServiceStatus, srvname=srvname, jail=jail):
+            host.noop(f"Service '{srvname}' already stopped")
+            return
+
+        args.extend([QuoteString(srvname), "stop"])
+    elif srvstate == SRV_RESTARTED:
+        args.extend([QuoteString(srvname), "restart"])
+
+    elif srvstate == SRV_RELOADED:
+        args.extend([QuoteString(srvname), "reload"])
+
+    elif srvstate == SRV_CUSTOM:
+        args.append(QuoteString(srvname))
+
+        if command is not None:
+            if isinstance(command, str):
+                command = [command]
+
+            args.extend([QuoteString(c) for c in command])
+
+    else:
+        raise OperationValueError(f"Invalid service command: {srvstate}")
+
+    yield StringCommand(*args)

pyinfra/operations/freebsd/sysrc.py

@@ -0,0 +1,92 @@
+"""
+Manipulate system rc files.
+"""
+
+from __future__ import annotations
+
+from typing_extensions import List, Optional, Union
+
+from pyinfra import host
+from pyinfra.api import QuoteString, StringCommand, operation
+from pyinfra.api.exceptions import OperationValueError
+from pyinfra.facts.freebsd import Sysrc
+
+SYSRC_ADD: str = "add"
+SYSRC_SUB: str = "sub"
+SYSRC_SET: str = "set"
+SYSRC_DEL: str = "del"
+
+
+@operation()
+def sysrc(
+    parameter: str,
+    value: str,
+    jail: Optional[str] = None,
+    command: str = SYSRC_SET,
+    overwrite: bool = False,
+):
+    """
+    Safely edit system rc files.
+
+    + parameter: Parameter to manipulate.
+    + value: Value, if the parameter requires it.
+    + jail: See ``-j`` in ``sysrc(8)``.
+    + command: Desire state of the parameter.
+    + overwrite: Overwrite the value of the parameter when ``command`` is set to ``set``.
+
+    Commands:
+        There are a few commands you can use to manipulate the rc file:
+
+        - add: Adds the value to the parameter.
+        - sub: Delete the parameter value.
+        - set: Change the parameter value. If the parameter already has a value
+               set, the changes will not be applied unless ``overwrite`` is set
+               to ``True``.
+        - del: Delete the parameter.
+
+    **Example:**
+
+    .. code:: python
+
+        sysrc.sysrc(
+            "beanstalkd_enable",
+            "YES",
+            command="set"
+        )
+    """
+
+    args: List[Union[str, "QuoteString"]] = []
+
+    args.extend(["sysrc", "-i"])
+
+    if command == SYSRC_DEL:
+        sign = "="
+
+        if not host.get_fact(Sysrc, parameter=parameter, jail=jail):
+            host.noop(f"Cannot find sysrc(8) parameter '{parameter}'")
+            return
+
+        args.append("-x")
+
+    elif command == SYSRC_SET:
+        sign = "="
+
+        if not overwrite and host.get_fact(Sysrc, parameter=parameter, jail=jail):
+            host.noop(f"sysrc(8) parameter '{parameter}' already set")
+            return
+
+    elif command == SYSRC_ADD:
+        sign = "+="
+
+    elif command == SYSRC_SUB:
+        sign = "-="
+
+    else:
+        raise OperationValueError(f"Invalid sysrc command: {command}")
+
+    if jail is not None:
+        args.extend(["-j", QuoteString(jail)])
+
+    args.extend(["--", QuoteString(f"{parameter}{sign}{value}")])
+
+    yield StringCommand(*args)

pyinfra/operations/gem.py

@@ -0,0 +1,48 @@
+"""
+Manage Ruby gem packages. (see https://rubygems.org/ )
+"""
+
+from __future__ import annotations
+
+from pyinfra import host
+from pyinfra.api import operation
+from pyinfra.facts.gem import GemPackages
+
+from .util.packaging import ensure_packages
+
+
+@operation()
+def packages(packages: str | list[str] | None = None, present=True, latest=False):
+    """
+    Add/remove/update gem packages.
+
+    + packages: list of packages to ensure
+    + present: whether the packages should be installed
+    + latest: whether to upgrade packages without a specified version
+
+    Versions:
+        Package versions can be pinned like gem: ``<pkg>:<version>``.
+
+    **Example:**
+
+    .. code:: python
+
+        from pyinfra.operations import gem
+        # Note: Assumes that 'gem' is installed.
+        gem.packages(
+            name="Install rspec",
+            packages=["rspec"],
+        )
+    """
+
+    yield from ensure_packages(
+        host,
+        packages,
+        host.get_fact(GemPackages),
+        present,
+        install_command="gem install",
+        uninstall_command="gem uninstall",
+        upgrade_command="gem update",
+        version_join=":",
+        latest=latest,
+    )