def get_s3_keys()

in functions/source/publish-call-metadata/lambda_function.py [0:0]

• 18 lines of code
• 5 McCabe index (conditional complexity)

def get_s3_keys(bucket, prefix):
    '''
    Get individual audio files of both caller and callee from s3
    '''
    keys = []

    kwargs = {'Bucket': bucket,'Prefix' : prefix}
    while True:
        resp = s3_client.list_objects_v2(**kwargs)
        for obj in resp['Contents']:
            key = obj['Key']
            extension = key.split('.')[-1]
            if extension == 'wav':
                keys.append(key)
        try:
            kwargs['ContinuationToken'] = resp['NextContinuationToken']
        except KeyError:
            break
    return keys