44
55from cloudshell .api .cloudshell_api import ResourceInfo
66
7- from cloudshell .iac .terraform .constants import AZURE2G_MODEL , ATTRIBUTE_NAMES , AWS2G_MODEL
7+ from cloudshell .iac .terraform .constants import AZURE2G_MODEL , ATTRIBUTE_NAMES , AWS2G_MODEL , CLP_PROVIDER_MODELS
88from cloudshell .iac .terraform .models .shell_helper import ShellHelperObject
99
1010
@@ -26,20 +26,11 @@ def initialize_provider(shell_helper: ShellHelperObject):
2626 raise ValueError (f"{ clpr_res_fam } currently not supported" )
2727
2828 try :
29- if clp_res_model in [ 'Microsoft Azure' , AZURE2G_MODEL ] :
29+ if clp_res_model in CLP_PROVIDER_MODELS :
3030 ProviderHandler ._set_cloud_env_vars (
3131 clp_details ,
3232 clp_res_model ,
33- shell_helper ,
34- ProviderHandler ._set_azure_env_vars_based_on_clp
35- )
36-
37- elif clp_res_model in ['AWS EC2' , AWS2G_MODEL ]:
38- ProviderHandler ._set_cloud_env_vars (
39- clp_details ,
40- clp_res_model ,
41- shell_helper ,
42- ProviderHandler ._set_aws_env_vars_based_on_clp
33+ shell_helper
4334 )
4435 else :
4536 shell_helper .logger .error (f"{ clp_res_model } currently not supported" )
@@ -54,7 +45,6 @@ def _set_cloud_env_vars(
5445 clp_details : ResourceInfo ,
5546 clp_res_model : str ,
5647 shell_helper : ShellHelperObject ,
57- set_cloud_env_vars_based_on_clp : Callable
5848 ):
5949 shell_helper .sandbox_messages .write_message ("initializing provider..." )
6050 shell_helper .logger .info ("Initializing Environment variables with CloudProvider details" )
@@ -64,7 +54,12 @@ def _set_cloud_env_vars(
6454 if clp_res_model in [AZURE2G_MODEL , AWS2G_MODEL ]:
6555 cloud_attr_name_prefix = clp_res_model + "."
6656
67- set_cloud_env_vars_based_on_clp (cloud_attr_name_prefix , clp_resource_attributes , shell_helper )
57+ if clp_res_model in ['AWS EC2' , AWS2G_MODEL ]:
58+ ProviderHandler ._set_aws_env_vars_based_on_clp (
59+ cloud_attr_name_prefix , clp_resource_attributes , shell_helper )
60+ elif clp_res_model in ['Microsoft Azure' , AZURE2G_MODEL ]:
61+ ProviderHandler ._set_azure_env_vars_based_on_clp (
62+ cloud_attr_name_prefix , clp_resource_attributes , shell_helper )
6863
6964 @staticmethod
7065 def _set_azure_env_vars_based_on_clp (azure_attr_name_prefix , clp_resource_attributes , shell_helper ):
0 commit comments