Commit
·
0993244
1
Parent(s):
b4c1b1a
fix
Browse files- tools/helm_test_gen.py +4 -1
tools/helm_test_gen.py
CHANGED
@@ -1,4 +1,5 @@
|
|
1 |
import re
|
|
|
2 |
import yaml
|
3 |
from typing import Optional
|
4 |
from smolagents import Tool
|
@@ -66,12 +67,14 @@ class GenerateHelmTest(Tool):
|
|
66 |
"""
|
67 |
doc = [*yaml.safe_load_all(document)]
|
68 |
result = ''
|
|
|
|
|
69 |
|
70 |
for item in doc:
|
71 |
name = item['metadata']['name']
|
72 |
kind = item['kind']
|
73 |
kindDesc = (re.sub(r'((?<=[a-z])[A-Z]|(?<!\A)[A-Z](?=[a-z]))', r' \1', kind)).lower()
|
74 |
-
env = Environment(loader=FileSystemLoader(
|
75 |
env.globals['gyv'] = GenerateHelmTest.get_yaml_value
|
76 |
env.globals['bool_to_str'] = GenerateHelmTest.bool_to_str
|
77 |
env.globals['get_object_by_kind'] = GenerateHelmTest.get_object_by_kind
|
|
|
1 |
import re
|
2 |
+
import os
|
3 |
import yaml
|
4 |
from typing import Optional
|
5 |
from smolagents import Tool
|
|
|
67 |
"""
|
68 |
doc = [*yaml.safe_load_all(document)]
|
69 |
result = ''
|
70 |
+
current_dir = os.path.dirname(os.path.abspath(__file__))
|
71 |
+
template_dir = os.path.join(current_dir, 'templates')
|
72 |
|
73 |
for item in doc:
|
74 |
name = item['metadata']['name']
|
75 |
kind = item['kind']
|
76 |
kindDesc = (re.sub(r'((?<=[a-z])[A-Z]|(?<!\A)[A-Z](?=[a-z]))', r' \1', kind)).lower()
|
77 |
+
env = Environment(loader=FileSystemLoader(template_dir))
|
78 |
env.globals['gyv'] = GenerateHelmTest.get_yaml_value
|
79 |
env.globals['bool_to_str'] = GenerateHelmTest.bool_to_str
|
80 |
env.globals['get_object_by_kind'] = GenerateHelmTest.get_object_by_kind
|