database.py 文件源码

def mongo_job_id_exists(job_id):
    """
    Check if a job_id exists in MongoDB.
    This uses the `mongo` context object from Flask.

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

    Returns
    -------
    bool
        True if job_id exists. False otherwise.
    """
    key = dict(_id=ObjectId(job_id))
    count = mongo.db.jobs.count(key)
    return count == 1