1111 IaCTemplateGenerationEC2 ,
1212 IaCTemplateGenerationS3 ,
1313 IaCTemplateGenerationIAM ,
14- IaCTemplateGenerationArgoCD
14+ IaCTemplateGenerationArgoCD ,
15+ IaCTemplateGenerationELB ,
16+ IaCTemplateGenerationEFS
1517 )
1618
1719from fastapi import Response
2527from app .template_generators .terraform .aws .IAM import (IaC_template_generator_iam )
2628from app .template_generators .terraform .aws .s3 import (IaC_template_generator_s3 )
2729from app .template_generators .terraform .argocd import (IaC_template_generator_argocd )
30+ from app .template_generators .terraform .aws .ELB import (IaC_template_generator_elb )
31+ from app .template_generators .terraform .aws .EFS import (IaC_template_generator_efs )
2832import os
2933
3034@app .post ("/IaC-basic/" )
@@ -102,4 +106,26 @@ async def IaC_template_generation_argocd(request:IaCTemplateGenerationArgoCD) ->
102106 output = gpt_service (generated_prompt )
103107 edit_directory_generator ("terraform_generator" ,output )
104108 execute_pythonfile ("MyTerraform" ,"terraform_generator" )
105- return Output (output = 'output' )
109+ return Output (output = 'output' )
110+
111+
112+
113+ @app .post ("/IaC-template/aws/elb" )
114+ async def IaC_template_generation_aws_elb (request :IaCTemplateGenerationELB ) -> Output :
115+ if os .environ .get ("TEST" ):
116+ return Output (output = 'output (nothing special)' )
117+ generated_prompt = IaC_template_generator_elb (request )
118+ output = gpt_service (generated_prompt )
119+ edit_directory_generator ("terraform_generator" ,output )
120+ execute_pythonfile ("MyTerraform" ,"terraform_generator" )
121+ return Output (output = 'output' )
122+
123+ @app .post ("/IaC-template/aws/efs" )
124+ async def IaC_template_generation_aws_efs (request :IaCTemplateGenerationEFS ) -> Output :
125+ if os .environ .get ("TEST" ):
126+ return Output (output = 'output (nothing special)' )
127+ generated_prompt = IaC_template_generator_efs (request )
128+ output = gpt_service (generated_prompt )
129+ edit_directory_generator ("terraform_generator" ,output )
130+ execute_pythonfile ("MyTerraform" ,"terraform_generator" )
131+ return Output (output = 'output' )
0 commit comments