spadinastan/tasks.py

import invoke

import plex.tasks
import deluge.tasks
import kong.tasks

ns = invoke.Collection()

PYINFRA_COMMON_PREFIX = "pyinfra -vvv pyinfra/inventory.py"


@invoke.task()
def system_updates(ctx):
    ctx.run(f"{PYINFRA_COMMON_PREFIX} pyinfra/system_updates.py")


@invoke.task()
def system_reboot(ctx):
    ctx.run(f"{PYINFRA_COMMON_PREFIX} pyinfra/reboot.py")


server = invoke.Collection("server")

server.add_task(system_updates, name="update")
server.add_task(system_reboot, name="reboot")

ns.add_collection(plex.tasks.ns)
ns.add_collection(deluge.tasks.ns)
ns.add_collection(kong.tasks.ns)
ns.add_collection(server)