mirror of
https://github.com/ansible-collections/community.general.git
synced 2026-02-04 07:51:50 +00:00
c7f6a28d89
* Add basic typing for module_utils.
* Apply some suggestions.
Co-authored-by: Alexei Znamensky <103110+russoz@users.noreply.github.com>
* Make pass again.
* Add more types as suggested.
* Normalize extra imports.
* Add more type hints.
* Improve typing.
* Add changelog fragment.
* Reduce changelog.
* Apply suggestions from code review.
Co-authored-by: Alexei Znamensky <103110+russoz@users.noreply.github.com>
* Fix typo.
* Cleanup.
* Improve types and make type checking happy.
* Let's see whether older Pythons barf on this.
* Revert "Let's see whether older Pythons barf on this."
This reverts commit 9973af3dbe.
* Add noqa.
---------
Co-authored-by: Alexei Znamensky <103110+russoz@users.noreply.github.com>
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
# Copyright (c) 2024, Alexei Znamensky <russoz@gmail.com>
|
|
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
# SPDX-License-Identifier: GPL-3.0-or-later
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
import typing as t
|
|
|
|
from ansible_collections.community.general.plugins.module_utils.cmd_runner import CmdRunner, _ensure_list
|
|
|
|
if t.TYPE_CHECKING:
|
|
from ansible.module_utils.basic import AnsibleModule
|
|
|
|
|
|
class PythonRunner(CmdRunner):
|
|
def __init__(
|
|
self,
|
|
module: AnsibleModule,
|
|
command,
|
|
arg_formats=None,
|
|
default_args_order=(),
|
|
check_rc: bool = False,
|
|
force_lang: str = "C",
|
|
path_prefix: list[str] | None = None,
|
|
environ_update: dict[str, str] | None = None,
|
|
python: str = "python",
|
|
venv: str | None = None,
|
|
) -> None:
|
|
self.python = python
|
|
self.venv = venv
|
|
self.has_venv = venv is not None
|
|
|
|
if os.path.isabs(python) or "/" in python:
|
|
self.python = python
|
|
elif venv is not None:
|
|
if path_prefix is None:
|
|
path_prefix = []
|
|
path_prefix.append(os.path.join(venv, "bin"))
|
|
if environ_update is None:
|
|
environ_update = {}
|
|
environ_update["PATH"] = f"{':'.join(path_prefix)}:{os.environ['PATH']}"
|
|
environ_update["VIRTUAL_ENV"] = venv
|
|
|
|
python_cmd = [self.python] + _ensure_list(command)
|
|
|
|
super().__init__(
|
|
module, python_cmd, arg_formats, default_args_order, check_rc, force_lang, path_prefix, environ_update
|
|
)
|