in source/backend/create-collection-lambda/function.py [0:0]
def lambda_handler(event, context):
collection_id = event['ResourceProperties']['RekognitionCollectionName']
if event['RequestType'] == 'Delete':
try:
ret = rekognition.delete_collection(CollectionId=collection_id)
if ret['ResponseMetadata']['HTTPStatusCode'] == 200:
print('Resource deleted')
cfnresponse.send(event, context, "SUCCESS", {})
return
except:
logger.exception('Failed to delete Rekognition collection')
cfnresponse.send(event, context, "FAILED", {})
else:
try:
ret = rekognition.create_collection(CollectionId=collection_id)
if ret['ResponseMetadata']['HTTPStatusCode'] == 200:
print('Resource created')
cfnresponse.send(event, context, "SUCCESS", {})
except:
logger.exception('Failed to create Rekognition collection')
cfnresponse.send(event, context, "FAILED", {})