def output_pdf_temp_file_to_s3()

in src/pre_human_task_lambda.py [0:0]

• 12 lines of code
• 2 McCabe index (conditional complexity)

def output_pdf_temp_file_to_s3(s3, source_ref: str, content: Union[str, list], page_num: int, job_id: str):
    """Write a temporary file to a created S3 location and return the file key."""
    bucket, temp_folder_key = get_temp_folder_bucket_key_from_s3_uri(source_ref, job_id)
    print(f'uploading data to bucket = {bucket}, path = {temp_folder_key}')
    _, pdf_key = S3Client.bucket_key_from_s3_uri(source_ref)
    pdf_key_file_name = S3Client.remove_extension(pdf_key).split('/')[-1]
    if type(content) == bytes:
        file_key = f'{temp_folder_key}/{pdf_key_file_name}_{page_num}_base64'
    else:
        file_key = f'{temp_folder_key}/{pdf_key_file_name}_{page_num}_blocks.json'
        content = json.dumps(content, default=JSONHandler)
    s3.put_object(Body=content, Bucket=bucket, Key=file_key)
    return f's3://{bucket}/{file_key}'