39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import argparse
|
|
import importlib
|
|
import logging
|
|
|
|
from ..machines.machines import Machine
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
def check_facts(machine: Machine) -> bool:
|
|
facts_module = importlib.import_module(machine.facts_module)
|
|
fact_store = facts_module.FactStore(machine=machine)
|
|
|
|
existing_facts = fact_store.get_all()
|
|
missing_facts = []
|
|
for service in machine.secrets_data:
|
|
for fact in machine.secrets_data[service]["facts"]:
|
|
if fact not in existing_facts.get(service, {}):
|
|
log.info(f"Fact {fact} for service {service} is missing")
|
|
missing_facts.append((service, fact))
|
|
|
|
if missing_facts:
|
|
return False
|
|
return True
|
|
|
|
|
|
def check_command(args: argparse.Namespace) -> None:
|
|
machine = Machine(name=args.machine, flake=args.flake)
|
|
if check_facts(machine):
|
|
print("All facts are present")
|
|
|
|
|
|
def register_check_parser(parser: argparse.ArgumentParser) -> None:
|
|
parser.add_argument(
|
|
"machine",
|
|
help="The machine to check facts for",
|
|
)
|
|
parser.set_defaults(func=check_command)
|