55# snippet-start:[neptune.python.scenario.main]
66import boto3
77import time
8- import botocore .exceptions
8+ from botocore .exceptions import ClientError
99
1010# Constants used in this scenario
1111POLL_INTERVAL_SECONDS = 10
@@ -32,7 +32,7 @@ def delete_db_cluster(neptune_client, cluster_id: str):
3232 print (f"Deleting DB Cluster: { cluster_id } " )
3333 neptune_client .delete_db_cluster (** request )
3434
35- except botocore . ClientError as err :
35+ except ClientError as err :
3636 code = err .response ["Error" ]["Code" ]
3737 message = err .response ["Error" ]["Message" ]
3838
@@ -76,7 +76,7 @@ def delete_db_instance(neptune_client, instance_id: str):
7676
7777 print (f"DB Instance '{ instance_id } ' successfully deleted." )
7878
79- except botocore . ClientError as err :
79+ except ClientError as err :
8080 code = err .response ["Error" ]["Code" ]
8181 message = err .response ["Error" ]["Message" ]
8282
@@ -109,7 +109,7 @@ def delete_db_subnet_group(neptune_client, subnet_group_name):
109109 neptune_client .delete_db_subnet_group (** delete_group_request )
110110 print (f"🗑️ Deleting Subnet Group: { subnet_group_name } " )
111111
112- except botocore . ClientError as err :
112+ except ClientError as err :
113113 code = err .response ["Error" ]["Code" ]
114114 message = err .response ["Error" ]["Message" ]
115115
@@ -192,7 +192,7 @@ def start_db_cluster(neptune_client, cluster_identifier: str):
192192 # Initial wait in case the cluster was just stopped
193193 time .sleep (30 )
194194 neptune_client .start_db_cluster (DBClusterIdentifier = cluster_identifier )
195- except botocore . ClientError as err :
195+ except ClientError as err :
196196 code = err .response ["Error" ]["Code" ]
197197 message = err .response ["Error" ]["Message" ]
198198
@@ -211,7 +211,7 @@ def start_db_cluster(neptune_client, cluster_identifier: str):
211211 clusters = []
212212 for page in pages :
213213 clusters .extend (page .get ('DBClusters' , []))
214- except botocore . ClientError as err :
214+ except ClientError as err :
215215 code = err .response ["Error" ]["Code" ]
216216 message = err .response ["Error" ]["Message" ]
217217
@@ -252,7 +252,7 @@ def stop_db_cluster(neptune_client, cluster_identifier: str):
252252 """
253253 try :
254254 neptune_client .stop_db_cluster (DBClusterIdentifier = cluster_identifier )
255- except botocore . ClientError as err :
255+ except ClientError as err :
256256 code = err .response ["Error" ]["Code" ]
257257 message = err .response ["Error" ]["Message" ]
258258
@@ -271,7 +271,7 @@ def stop_db_cluster(neptune_client, cluster_identifier: str):
271271 clusters = []
272272 for page in pages :
273273 clusters .extend (page .get ('DBClusters' , []))
274- except botocore . ClientError as err :
274+ except ClientError as err :
275275 code = err .response ["Error" ]["Code" ]
276276 message = err .response ["Error" ]["Message" ]
277277
@@ -339,12 +339,12 @@ def describe_db_clusters(neptune_client, cluster_id: str):
339339
340340 if not found :
341341 # Treat empty response as cluster not found
342- raise botocore . ClientError (
342+ raise ClientError (
343343 {"Error" : {"Code" : "DBClusterNotFound" , "Message" : f"No cluster found with ID '{ cluster_id } '" }},
344344 "DescribeDBClusters"
345345 )
346346
347- except botocore . ClientError as err :
347+ except ClientError as err :
348348 code = err .response ["Error" ]["Code" ]
349349 message = err .response ["Error" ]["Message" ]
350350
@@ -377,7 +377,7 @@ def check_instance_status(neptune_client, instance_id: str, desired_status: str)
377377 for page in pages :
378378 instances .extend (page .get ('DBInstances' , []))
379379
380- except botocore . ClientError as err :
380+ except ClientError as err :
381381 code = err .response ["Error" ]["Code" ]
382382 message = err .response ["Error" ]["Message" ]
383383
@@ -430,7 +430,7 @@ def create_db_instance(neptune_client, db_instance_id: str, db_cluster_id: str)
430430 print (f"DB Instance '{ db_instance_id } ' is now available." )
431431 return instance ['DBInstanceIdentifier' ]
432432
433- except botocore . ClientError as err :
433+ except ClientError as err :
434434 code = err .response ["Error" ]["Code" ]
435435 message = err .response ["Error" ]["Message" ]
436436
@@ -479,7 +479,7 @@ def create_db_cluster(neptune_client, db_name: str) -> str:
479479 print (f"DB Cluster created: { cluster_id } " )
480480 return cluster_id
481481
482- except botocore . ClientError as e :
482+ except ClientError as e :
483483 code = e .response ["Error" ]["Code" ]
484484 message = e .response ["Error" ]["Message" ]
485485
@@ -559,7 +559,7 @@ def create_subnet_group(neptune_client, group_name: str):
559559 print (f"ARN: { arn } " )
560560 return name , arn
561561
562- except botocore . ClientError as e :
562+ except ClientError as e :
563563 code = e .response ["Error" ]["Code" ]
564564 msg = e .response ["Error" ]["Message" ]
565565
@@ -658,7 +658,7 @@ def run_scenario(neptune_client, subnet_group_name: str, db_instance_id: str, cl
658658
659659 print ("Neptune resources deleted successfully" )
660660
661- except botocore . ClientError as ce :
661+ except ClientError as ce :
662662 code = ce .response ["Error" ]["Code" ]
663663
664664 if code in ("DBInstanceNotFound" , "DBInstanceNotFoundFault" , "ResourceNotFound" ):
@@ -701,4 +701,4 @@ def main():
701701
702702if __name__ == "__main__" :
703703 main ()
704- # snippet-end:[neptune.python.scenario.main]
704+ # snippet-end:[neptune.python.scenario.main]
0 commit comments