anta/docs/scripts/generate_examples_tests.py
Daniel Baumann afeccccd6a
Merging upstream version 1.2.0.
Signed-off-by: Daniel Baumann <daniel@debian.org>
2025-02-05 11:55:22 +01:00

29 lines
842 B
Python
Executable file

#!/usr/bin/env python
# Copyright (c) 2024 Arista Networks, Inc.
# Use of this source code is governed by the Apache License 2.0
# that can be found in the LICENSE file.
"""Generates examples/tests.py."""
import os
from contextlib import redirect_stdout
from pathlib import Path
from sys import path
# Override global path to load anta from pwd instead of any installed version.
path.insert(0, str(Path(__file__).parents[2]))
examples_tests_path = Path(__file__).parents[2] / "examples" / "tests.yaml"
prev = os.environ.get("TERM", "")
os.environ["TERM"] = "dumb"
# imported after TERM is set to act upon rich console.
from anta.cli.get.commands import tests # noqa: E402
with examples_tests_path.open("w") as f:
f.write("---\n")
with redirect_stdout(f):
# removing the style
tests()
os.environ["TERM"] = prev