utils.py

import shutil
import requests
import uuid

def load_file(path):
    ''' 
    Returns path of input file. If given parameter 'path' is an url,
    first downloads the file
    '''

    if path.startswith(('https://', 'http://')):
        fileName = path.split('/')[-1]
        filePath = 'data/input/'
        data = requests.get(path)
        open(filePath+fileName, 'wb') as f: f.write(data)
        return filePath+fileName
    else:
        path = path.replace('file://', '')
        return path

def expose_file(path):
    ''' 
    Saves file to make it accessible to API.
    '''

    fileExtension = path.split('.')[-1]
    filePath = 'data/ouput/'
    fileName = uuid.uuid4().hex + fileExtension
    new_path = filePath+fileName
    shutil.move(path, new_path)
    return new_path