def construct_df_from_result()

in src/utils.py [0:0]

• 15 lines of code
• 3 McCabe index (conditional complexity)

def construct_df_from_result(s3_uri):
    bucket, s3_key = get_bucket_key_from_s3_uri(s3_uri)
    
    list_resp = s3_client.list_objects_v2(Bucket=bucket, Prefix=s3_key)
    
    df_list = []
    
    # If no data is present
    if 'Contents' not in list_resp:
        return None
    
    for key in list_resp['Contents']:
        obj_resp = s3_client.get_object(Bucket=bucket, Key=key['Key'])  
        json_str = obj_resp['Body'].read().decode('utf-8')
        json_obj = json.loads(json_str)

        df = pd.read_json(json_obj['drift_df'], orient='columns')    
        df['Time'] = [json_obj['end_time'].split(':')[0]] * len(df)
        df['accuracy'] = [json_obj['accuracy']] * len(df)
        df_list.append(df)

    return pd.concat(df_list)