How to run a singular operation then move on?

[Dexmachi]

when doing something like:

def handleGitRepo(users, sysUsers, dot, host, state):
    from pyinfra.api.operations import run_host_op

    user = dot.get('user')
    if user not in users:
        print(f"Skipping dotfile setup for user '{user}', as they do not exist on system.")
        return None, None, None, None
    if user in sysUsers:
        print(f"Skipping dotfile setup for system user '{user}', activity not allowed.")
        return None, None, None, None

    dotName = dot.get('url').split('/')[-1].replace('.git', '')
    dotLoc = f"/home/{user}/.dotfiles/chaos/{dotName}"
    dotLocEx = host.get_fact(Directory, path=dotLoc)

    branch = dot.get('branch', 'main')
    pull = dot.get('pull', False)
    should_run_git = not dotLocEx or pull
    if should_run_git:
        print(f"Cloning {dot.get('url')} to {dotLoc}.")
        dotfiles_op = add_op(
            state, git.repo,
            name=f"Ensuring dotfile repo state for '{user}'",
            src=dot.get('url'),
            dest=dotLoc,
            branch=branch,
            pull=pull,
            user=user,
        )
        dotfiles_op_host: OperationMeta = dotfiles_op[host]

        op_hash = dotfiles_op_host._hash
        run_host_op(state, host, op_hash)
    return dotLoc, dotName, dot, user

it spits out

Traceback (most recent call last):
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/.venv/bin/chaos", line 10, in <module>
    sys.exit(main())
             ^^^^^^
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/cli/src/chaos/cli.py", line 48, in main
    handleApply(args, Console())
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/cli/src/chaos/cli.py", line 160, in handleApply
    handleOrchestration(args, dry, ikwid, ROLES_DISPATCHER, ROLE_ALIASES)
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/cli/src/chaos/lib/handlers.py", line 417, in handleOrchestration
    _run_tags(
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/cli/src/chaos/lib/handlers.py", line 379, in _run_tags
    ROLES_DISPATCHER[normalized_tag](*commonArgs)
  File "/home/dexmachina/.local/share/chaos/plugins/chaos-dots.whl/chaos_dots/roles/dotfiles/dotfiles_new.py", line 134, in runDotfiles
  File "/home/dexmachina/.local/share/chaos/plugins/chaos-dots.whl/chaos_dots/roles/dotfiles/dotfiles_new.py", line 77, in handleGitRepo
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/.venv/lib/python3.12/site-packages/pyinfra/api/operations.py", line 54, in run_host_op
    return _run_host_op(state, host, op_hash)
           ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/.venv/lib/python3.12/site-packages/pyinfra/api/operations.py", line 95, in _run_host_op
    for command in op_data.command_generator():
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/.venv/lib/python3.12/site-packages/pyinfra/api/operation.py", line 332, in command_generator
    host.in_op = _set_in_op
    ^^^^^^^^^^
  File "/home/dexmachina/projetos/Ch-aOS/Ch-aOS/.venv/lib/python3.12/site-packages/pyinfra/context.py", line 64, in __setattr__
    raise TypeError("Cannot assign to context base module")
TypeError: Cannot assign to context base module

how can I handle this correctly?

[Dexmachi]

Got it to work with a simple

host.get_fact(Command, f"git clone '{dot.get('url')}' '{dotLoc}'", _sudo=True, _sudo_user=user)

Bit of a hack, sure, but oh well

[Fizzadar]

Ah so it errors because the host/state context haven't been set (error is poor), untested but roughly should work:

from pyinfra.context import ctx_state, ctx_host

with ctx_state.use(state):
    with ctx_host.use(host):
        handleGitRepo(users, sysUsers, dot, host, state)

[Dexmachi]

ooooh, nice, ty!