@@ -21,6 +21,7 @@ from kubernetes.config.config_exception import ConfigException
2121import argparse
2222import logging
2323import urllib3
24+
2425urllib3 .disable_warnings ()
2526
2627
@@ -30,7 +31,12 @@ if __name__ == "__main__":
3031 # Primary Options
3132 parser .add_argument ("--mas-instance-id" , required = True )
3233 parser .add_argument ("--mas-workspace-id" , required = True )
33- parser .add_argument ("--log-level" , required = False , choices = ["DEBUG" , "INFO" , "WARNING" , "ERROR" , "CRITICAL" ], default = "INFO" )
34+ parser .add_argument (
35+ "--log-level" ,
36+ required = False ,
37+ choices = ["DEBUG" , "INFO" , "WARNING" , "ERROR" , "CRITICAL" ],
38+ default = "INFO" ,
39+ )
3440 parser .add_argument ("--coreapi-port" , required = False , default = 443 )
3541 parser .add_argument ("--admin-dashboard-port" , required = False , default = 443 )
3642 parser .add_argument ("--manage-api-port" , required = False , default = 443 )
@@ -86,31 +92,47 @@ if __name__ == "__main__":
8692 config .load_kube_config ()
8793 logger .debug ("Loaded kubeconfig file" )
8894
89- user_utils = MASUserUtils (mas_instance_id , mas_workspace_id , client .api_client .ApiClient (), mas_version , coreapi_port = coreapi_port , admin_dashboard_port = admin_dashboard_port , manage_api_port = manage_api_port )
95+ user_utils = MASUserUtils (
96+ mas_instance_id ,
97+ mas_workspace_id ,
98+ client .api_client .ApiClient (),
99+ mas_version ,
100+ coreapi_port = coreapi_port ,
101+ admin_dashboard_port = admin_dashboard_port ,
102+ manage_api_port = manage_api_port ,
103+ )
90104
91105 if initial_users_secret_name is not None :
92106
93- logger .info (f"Loading initial_users configuration from secret { initial_users_secret_name } " )
107+ logger .info (
108+ f"Loading initial_users configuration from secret { initial_users_secret_name } "
109+ )
94110
95111 session = boto3 .session .Session ()
96112 aws_sm_client = session .client (
97- service_name = ' secretsmanager' ,
113+ service_name = " secretsmanager" ,
98114 )
99115 try :
100- initial_users_secret = aws_sm_client .get_secret_value ( # pragma: allowlist secret
101- SecretId = initial_users_secret_name
116+ initial_users_secret = (
117+ aws_sm_client .get_secret_value ( # pragma: allowlist secret
118+ SecretId = initial_users_secret_name
119+ )
102120 )
103121 except ClientError as e :
104- if e .response ['Error' ]['Code' ] == 'ResourceNotFoundException' :
105- logger .info (f"Secret { initial_users_secret_name } was not found, nothing to do, exiting now." )
122+ if e .response ["Error" ]["Code" ] == "ResourceNotFoundException" :
123+ logger .info (
124+ f"Secret { initial_users_secret_name } was not found, nothing to do, exiting now."
125+ )
106126 sys .exit (0 )
107127
108- raise Exception (f"Failed to fetch secret { initial_users_secret_name } : { str (e )} " )
128+ raise Exception (
129+ f"Failed to fetch secret { initial_users_secret_name } : { str (e )} "
130+ )
109131
110- secret_json = json .loads (initial_users_secret [' SecretString' ])
132+ secret_json = json .loads (initial_users_secret [" SecretString" ])
111133 initial_users = user_utils .parse_initial_users_from_aws_secret_json (secret_json )
112134 elif initial_users_yaml_file is not None :
113- with open (initial_users_yaml_file , 'r' ) as file :
135+ with open (initial_users_yaml_file , "r" ) as file :
114136 initial_users = yaml .safe_load (file )
115137 else :
116138 raise Exception ("Something unexpected happened" )
@@ -122,7 +144,9 @@ if __name__ == "__main__":
122144 if initial_users_secret_name is not None :
123145 has_updates = False
124146 for completed_user in result ["completed" ]:
125- logger .info (f"Removing synced user { completed_user ['email' ]} from { initial_users_secret_name } secret" )
147+ logger .info (
148+ f"Removing synced user { completed_user ['email' ]} from { initial_users_secret_name } secret"
149+ )
126150 secret_json .pop (completed_user ["email" ])
127151 has_updates = True
128152
@@ -131,10 +155,12 @@ if __name__ == "__main__":
131155 try :
132156 aws_sm_client .update_secret ( # pragma: allowlist secret
133157 SecretId = initial_users_secret_name ,
134- SecretString = json .dumps (secret_json )
158+ SecretString = json .dumps (secret_json ),
135159 )
136160 except ClientError as e :
137- raise Exception (f"Failed to update secret { initial_users_secret_name } : { str (e )} " )
161+ raise Exception (
162+ f"Failed to update secret { initial_users_secret_name } : { str (e )} "
163+ )
138164
139165 if len (result ["failed" ]) > 0 :
140166 failed_user_ids = list (map (lambda u : u ["email" ], result ["failed" ]))
0 commit comments