h2oaimichalmarszalek commited on
Commit
5395a33
·
1 Parent(s): 6181de0

refactor to functor

Browse files
Files changed (1) hide show
  1. tools/helm_test_gen.py +4 -4
tools/helm_test_gen.py CHANGED
@@ -19,13 +19,13 @@ class GenerateHelmTest(Tool):
19
 
20
  '''
21
 
22
-
23
- def get_yaml_value(self, d: dict[str], xpath: str, default: Optional[str] = None) -> str:
24
  try:
25
  if '.' in xpath:
26
  h = xpath.split('.')[0]
27
  t = '.'.join(xpath.split('.')[1:])
28
- return self.get_yaml_value(d[h], t)
29
  else:
30
  return d[xpath]
31
  except Exception as ex:
@@ -74,7 +74,7 @@ class GenerateHelmTest(Tool):
74
  kind = item['kind']
75
  kindDesc = (re.sub(r'((?<=[a-z])[A-Z]|(?<!\A)[A-Z](?=[a-z]))', r' \1', kind)).lower()
76
  env = Environment(loader=FileSystemLoader(template_dir))
77
- env.globals['gyv'] = self.get_yaml_value
78
  env.globals['bool_to_str'] = GenerateHelmTest.bool_to_str
79
  env.globals['get_object_by_kind'] = GenerateHelmTest.get_object_by_kind
80
  env.globals['get_env_by_ref'] = GenerateHelmTest.get_envs_by_ref
 
19
 
20
  '''
21
 
22
+ @staticmethod
23
+ def get_yaml_value(d: dict[str], xpath: str, default: Optional[str] = None) -> str:
24
  try:
25
  if '.' in xpath:
26
  h = xpath.split('.')[0]
27
  t = '.'.join(xpath.split('.')[1:])
28
+ return GenerateHelmTest.get_yaml_value(d[h], t)
29
  else:
30
  return d[xpath]
31
  except Exception as ex:
 
74
  kind = item['kind']
75
  kindDesc = (re.sub(r'((?<=[a-z])[A-Z]|(?<!\A)[A-Z](?=[a-z]))', r' \1', kind)).lower()
76
  env = Environment(loader=FileSystemLoader(template_dir))
77
+ env.globals['gyv'] = GenerateHelmTest.get_yaml_value
78
  env.globals['bool_to_str'] = GenerateHelmTest.bool_to_str
79
  env.globals['get_object_by_kind'] = GenerateHelmTest.get_object_by_kind
80
  env.globals['get_env_by_ref'] = GenerateHelmTest.get_envs_by_ref