11import os
22from abc import ABCMeta
3-
43from cloudshell .iac .terraform .models .shell_helper import ShellHelperObject
54
65
@@ -13,14 +12,14 @@ def set_env_vars_based_on_clp(self):
1312
1413 @staticmethod
1514 def does_attribute_match (clp_res_model , clp_attribute , attr_name_to_check ) -> bool :
16- if f"{ clp_res_model } .{ clp_attribute .Name } " == attr_name_to_check or clp_attribute .Name == attr_name_to_check :
15+ if f"{ clp_res_model } .{ clp_attribute .Name } " == attr_name_to_check or clp_attribute .Name == attr_name_to_check \
16+ or clp_attribute .Name == f"{ clp_res_model } .{ attr_name_to_check } " :
1717 return True
1818 return False
1919
2020
2121class AWSCloudProviderEnvVarHandler (BaseCloudProviderEnvVarHandler ):
22- def __init__ (self , clp_res_model : str , clp_resource_attributes : list ,
23- shell_helper : ShellHelperObject ):
22+ def __init__ (self , clp_res_model : str , clp_resource_attributes : list , shell_helper : ShellHelperObject ):
2423 BaseCloudProviderEnvVarHandler .__init__ (self )
2524 self ._clp_res_model = clp_res_model
2625 self ._clp_resource_attributes = clp_resource_attributes
@@ -49,7 +48,7 @@ def set_env_vars_based_on_clp(self):
4948
5049
5150class AzureCloudProviderEnvVarHandler (BaseCloudProviderEnvVarHandler ):
52- def __init__ (self , clp_res_model , clp_resource_attributes , shell_helper ):
51+ def __init__ (self , clp_res_model : str , clp_resource_attributes : list , shell_helper : ShellHelperObject ):
5352 BaseCloudProviderEnvVarHandler .__init__ (self )
5453 self ._clp_res_model = clp_res_model
5554 self ._clp_resource_attributes = clp_resource_attributes
@@ -68,7 +67,7 @@ def set_env_vars_based_on_clp(self):
6867
6968
7069class GCPCloudProviderEnvVarHandler (BaseCloudProviderEnvVarHandler ):
71- def __init__ (self , clp_res_model , clp_resource_attributes , shell_helper ):
70+ def __init__ (self , clp_res_model : str , clp_resource_attributes : list , shell_helper : ShellHelperObject ):
7271 BaseCloudProviderEnvVarHandler .__init__ (self )
7372 self ._clp_res_model = clp_res_model
7473 self ._clp_resource_attributes = clp_resource_attributes
0 commit comments