database.py 文件源码

def mongo_add_s3_file_key(job_id, s3_file_key):
    """
    Adds 's3_file_key' key-value to job object in MongoDB.
    This uses the `mongo` context object from Flask.

    Parameters
    ----------
    job_id: str
    s3_file_key: str

    Returns
    -------
    None

    """
    response = mongo.db.jobs.update_one({'_id': ObjectId(job_id)}, {'$set': {'s3_file_key': s3_file_key}})
    return response