Commit
·
20638d1
1
Parent(s):
97ee047
refactor to functor
Browse files- tools/helm_test_gen.py +1 -4
tools/helm_test_gen.py
CHANGED
@@ -1,6 +1,5 @@
|
|
1 |
import re
|
2 |
import yaml
|
3 |
-
import os
|
4 |
from typing import Optional
|
5 |
from smolagents import Tool
|
6 |
from jinja2 import Environment, FileSystemLoader
|
@@ -67,14 +66,12 @@ class GenerateHelmTest(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(
|
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
|
|
|
1 |
import re
|
2 |
import yaml
|
|
|
3 |
from typing import Optional
|
4 |
from smolagents import Tool
|
5 |
from jinja2 import Environment, FileSystemLoader
|
|
|
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('templates'))
|
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
|