76 lines
2.3 KiB
Python
76 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""Helm Chart 结构 sanity 检查（无 helm CLI 环境下的离线基本验证）。
|
||
|
||
检查项：
|
||
1. Chart.yaml / values.yaml 均为合法 YAML；
|
||
2. templates/ 下所有模板文件存在且非空；
|
||
3. 模板中引用的 `.Values.xxx` 键均能在 values.yaml 中找到（防拼写错误）；
|
||
4. 每个模板文件的 Go template 标记 `{{` / `}}` 数量配平。
|
||
|
||
用法：python _sanity_check.py
|
||
"""
|
||
import os
|
||
import re
|
||
import sys
|
||
|
||
HERE = os.path.dirname(os.path.abspath(__file__))
|
||
TEMPLATES = os.path.join(HERE, "templates")
|
||
|
||
import yaml # noqa: E402
|
||
|
||
|
||
def load(path):
|
||
with open(path, "r", encoding="utf-8") as fh:
|
||
return yaml.safe_load(fh)
|
||
|
||
|
||
def values_paths(text):
|
||
"""提取模板文本中的 .Values.<a>.<b> 路径集合。"""
|
||
return set(re.findall(r"\.Values\.([A-Za-z0-9_]+(?:\.[A-Za-z0-9_]+)*)", text))
|
||
|
||
|
||
def resolve(values, dotted):
|
||
node = values
|
||
for key in dotted.split("."):
|
||
if not isinstance(node, dict) or key not in node:
|
||
return False
|
||
node = node[key]
|
||
return True
|
||
|
||
|
||
def main():
|
||
failures = []
|
||
|
||
chart = load(os.path.join(HERE, "Chart.yaml"))
|
||
assert chart.get("apiVersion") == "v2", "Chart.yaml 应为 Helm v2 chart"
|
||
assert chart.get("name") == "iaop", "Chart.name 应为 iaop"
|
||
|
||
values = load(os.path.join(HERE, "values.yaml"))
|
||
assert values.get("inference", {}).get("backend") in ("gpu", "npu"), \
|
||
"values.inference.backend 应为 gpu|npu"
|
||
|
||
for name in sorted(os.listdir(TEMPLATES)):
|
||
path = os.path.join(TEMPLATES, name)
|
||
if not os.path.isfile(path):
|
||
continue
|
||
text = open(path, "r", encoding="utf-8").read()
|
||
if not text.strip():
|
||
failures.append(f"{name}: 模板文件为空")
|
||
continue
|
||
if text.count("{{") != text.count("}}"):
|
||
failures.append(f"{name}: Go template 标记 {{/}} 数量不配平")
|
||
for dotted in values_paths(text):
|
||
if not resolve(values, dotted):
|
||
failures.append(f"{name}: 引用了 values.yaml 中不存在的键 .Values.{dotted}")
|
||
|
||
if failures:
|
||
print("FAIL")
|
||
for f in failures:
|
||
print(" -", f)
|
||
sys.exit(1)
|
||
print(f"OK: Chart/values 合法，templates/{len(os.listdir(TEMPLATES))} 个模板校验通过")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|