def add_face_index()

in usecases/image-processing/dags/1.10/image_processing.py [0:0]

• 16 lines of code
• 1 McCabe index (conditional complexity)

def add_face_index(ds, **kwargs): 
    client = boto3.client('rekognition')
    arg = kwargs['dag_run'].conf

    
    response = client.index_faces(
        CollectionId=arg['RekognitionCollectionId'],
        DetectionAttributes=['ALL'],
        ExternalImageId=arg['userId'],
        Image={
            "S3Object": {
                "Bucket": arg['s3Bucket'],
                "Name": arg['s3Key']
            }
        }

    )
    print(response['FaceRecords'][0])
    kwargs['ti'].xcom_push(key="FaceIndexDetails", value=response['FaceRecords'][0]['Face'])