add heplers.tpl to prompt

Author: abolfazl8131

app/__pycache__/prompt_generators.cpython-311.pyc

@@ -1,86 +1,70 @@
 import os
-import yaml
 
-# Define directory and file structure
-project_name = 'MyHelm'
-base_path = os.path.join('app', 'media', project_name)
+# Define project structure
+project_name = 'app/media/MyHelm'
+directories = ['charts/', 'templates/']
+templates_dir = 'templates/web'
+files = ['Chart.yaml', 'values.yaml']
+template_subdirectories = ['web']
+template_files = ['service.yaml', 'helpers.tpl']
 
-directories = [
-    os.path.join(base_path, 'charts'),
-    os.path.join(base_path, 'templates', 'web')
-]
+# Chart.yaml content
+chart_yaml_content = """apiVersion: v2
+name: my-helm-chart
+description: A Helm chart for Kubernetes
+version: 0.1.0
+"""
 
-files = {
-    'Chart.yaml': {
-        'apiVersion': 'v2',
-        'name': project_name,
-        'description': 'A Helm chart for Kubernetes',
-        'version': '0.1.0',
-        'appVersion': '1.0.0'
-    },
-    'values.yaml': {
-        'web': {
-            'image': {
-                'repository': 'nginx',
-                'tag': 'latest'
-            },
-            'service': {
-                'enabled': True,
-                'port': 80
-            },
-            'replicaCount': 1,
-            'persistence': {
-                'enabled': True,
-                'size': '1Gi',
-                'accessModes': ['ReadWriteOnce']
-            },
-            'env': [{
-                'name': 'ENV1',
-                'value': 'Hi'
-            }],
-            'ingress': {
-                'enabled': False,
-                'host': 'www.example.com'
-            }
-        }
-    }
-}
+# values.yaml content
+values_yaml_content = """web:
+  image: nginx
+  replicas: 1
+  service:
+    targetPort: 80
+  persistence:
+    enabled: true
+    size: 1Gi
+    accessModes:
+      - ReadWriteOnce
+  env:
+    - name: ENV1
+      value: Hi
+  ingress:
+    enabled: false
+    host: www.example.com
+"""
 
-# Create directories
+# Create project structure
+os.makedirs(project_name, exist_ok=True)
 for directory in directories:
-    os.makedirs(directory, exist_ok=True)
+    os.makedirs(os.path.join(project_name, directory), exist_ok=True)
 
-# Create files with initial content
-for filename, content in files.items():
-    with open(os.path.join(base_path, filename), 'w') as file:
-        yaml.dump(content, file)
+# Create templates/web directory
+os.makedirs(os.path.join(project_name, templates_dir), exist_ok=True)
 
-# Create service.yaml template
-service_yaml_content = """apiVersion: v1
-kind: Service
-metadata:
-  name: {{ include "{project_name}.fullname" . }}
-spec:
-  type: ClusterIP
-  ports:
-    - port: {{ .Values.web.service.port }}
-  selector:
-    app: {{ include "{project_name}.name" . }}
-"""
+# Create files
+with open(os.path.join(project_name, 'Chart.yaml'), 'w') as chart_file:
+    chart_file.write(chart_yaml_content)
 
-with open(os.path.join(base_path, 'templates', 'web', 'service.yaml'), 'w') as service_file:
-    service_file.write(service_yaml_content.format(project_name=project_name))
+with open(os.path.join(project_name, 'values.yaml'), 'w') as values_file:
+    values_file.write(values_yaml_content)
 
-# Create secret.yaml template if environment variables exist
-if files['values.yaml']['web'].get('env'):
-    secret_yaml_content = """apiVersion: v1
-kind: Secret
-metadata:
-  name: {{ include "{project_name}.fullname" . }}-env
-type: Opaque
-data:
-  ENV1: {{ .Values.web.env[0].value | b64enc | quote }}
-"""
+# Create template files and directories
+for template in template_subdirectories:
+    template_path = os.path.join(project_name, 'templates', template)
+    os.makedirs(template_path, exist_ok=True)
+    for template_file in template_files:
+        with open(os.path.join(template_path, template_file), 'w') as file:
+            file.write(f"# {template_file} for {template}\n")
+
+# Create secret.yaml if env variable is considered
+with open(os.path.join(template_path, 'secret.yaml'), 'w') as secret_file:
+    secret_file.write("# secret.yaml for environment variables\n")
 
-    with open(os.path.join(base_path, 'templates', 'web', 'secret.yaml'), 'w') as secret_file:
-        secret_file.write(secret_yaml_content.format(project_name=project_name))
+# Create helpers.tpl
+with open(os.path.join(template_path, 'helpers.tpl'), 'w') as helpers_file:
+    helpers_file.write(f"""{{- define "{template}.labels" -}}
+name: {{ .Release.Name }}
+{{- end -}}
+// Add additional helper functions if needed
+""")

app/directory_generators/helm_generator.py

@@ -1,5 +1,4 @@
 apiVersion: v2
-appVersion: 1.0.0
+name: my-helm-chart
 description: A Helm chart for Kubernetes
-name: MyHelm
 version: 0.1.0

app/media/MyHelm/Chart.yaml

@@ -0,0 +1,4 @@
+{- define "web.labels" -}
+name: { .Release.Name }
+{- end -}
+// Add additional helper functions if needed

app/media/MyHelm/templates/web/helpers.tpl

@@ -1,7 +1 @@
-apiVersion: v1
-kind: Secret
-metadata:
-  name: { include "MyHelm.fullname" . }-env
-type: Opaque
-data:
-  ENV1: { .Values.web.env[0].value | b64enc | quote }
+# secret.yaml for environment variables

app/media/MyHelm/templates/web/secret.yaml

@@ -1,10 +1 @@
-apiVersion: v1
-kind: Service
-metadata:
-  name: { include "MyHelm.fullname" . }
-spec:
-  type: ClusterIP
-  ports:
-    - port: { .Values.web.service.port }
-  selector:
-    app: { include "MyHelm.name" . }
+# service.yaml for web

app/media/MyHelm/templates/web/service.yaml

@@ -1,19 +1,16 @@
 web:
+  image: nginx
+  replicas: 1
+  service:
+    targetPort: 80
+  persistence:
+    enabled: true
+    size: 1Gi
+    accessModes:
+      - ReadWriteOnce
   env:
-  - name: ENV1
-    value: Hi
-  image:
-    repository: nginx
-    tag: latest
+    - name: ENV1
+      value: Hi
   ingress:
     enabled: false
     host: www.example.com
-  persistence:
-    accessModes:
-    - ReadWriteOnce
-    enabled: true
-    size: 1Gi
-  replicaCount: 1
-  service:
-    enabled: true
-    port: 80

app/media/MyHelm/values.yaml

@@ -88,7 +88,10 @@ def helm_template_generator(input : HelmTemplateGeneration) -> str:
             if environment variable is considered for pod, then create secret.yaml in the related template.
 
             please set a something default in chart.yaml and values.yaml based on the requirement.
+            please generate helpers.tpl for each template and initialize resources and set a label for reach resource.
 
             just Generate a python code without any additional notes or ```python3 entry
             """
-    return prompt
+    return prompt
+
+