mirror of
https://github.com/kevinveenbirkenbach/computer-playbook.git
synced 2026-11-06 19:01:54 +00:00
72 lines
2.1 KiB
Python
72 lines
2.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
Ansible lookup plugin: version
|
|
|
|
No parameters.
|
|
Reads pyproject.toml relative to this plugin file:
|
|
<repo>/plugins/lookup/version.py -> <repo>/pyproject.toml
|
|
|
|
Resolution order:
|
|
1) [project].version (PEP 621)
|
|
2) [tool.poetry].version (Poetry fallback)
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from ansible.errors import AnsibleError
|
|
from ansible.plugins.lookup import LookupBase
|
|
|
|
try:
|
|
import tomllib # Python 3.11+
|
|
except Exception:
|
|
try:
|
|
import tomli as tomllib # type: ignore
|
|
except Exception as exc:
|
|
raise AnsibleError(
|
|
"version lookup requires 'tomllib' (Python 3.11+) "
|
|
"or the 'tomli' package installed on the control node."
|
|
) from exc
|
|
|
|
|
|
def _get_nested(data: dict, path: list[str]):
|
|
cur = data
|
|
for key in path:
|
|
if not isinstance(cur, dict) or key not in cur:
|
|
return None
|
|
cur = cur[key]
|
|
return cur
|
|
|
|
|
|
class LookupModule(LookupBase):
|
|
def run(self, terms, variables=None, **kwargs):
|
|
# Intentionally ignore terms/kwargs — no parameters supported
|
|
|
|
plugin_dir = os.path.dirname(os.path.abspath(__file__))
|
|
pyproject_path = os.path.normpath(
|
|
os.path.join(plugin_dir, "..", "..", "pyproject.toml")
|
|
)
|
|
|
|
if not os.path.exists(pyproject_path):
|
|
raise AnsibleError(f"version lookup: file not found: {pyproject_path}")
|
|
|
|
try:
|
|
with open(pyproject_path, "rb") as f:
|
|
data = tomllib.load(f)
|
|
except Exception as exc:
|
|
raise AnsibleError(
|
|
f"version lookup: failed to parse TOML: {pyproject_path}: {exc}"
|
|
) from exc
|
|
|
|
version = _get_nested(data, ["project", "version"])
|
|
if not version:
|
|
version = _get_nested(data, ["tool", "poetry", "version"])
|
|
|
|
if not version or not isinstance(version, str) or not version.strip():
|
|
raise AnsibleError(
|
|
f"version lookup: could not extract version from {pyproject_path}. "
|
|
"Tried [project].version and [tool.poetry].version."
|
|
)
|
|
|
|
return [version.strip()]
|