operations.py 文件源码

def getDetailsByZpid(zpid, get_prediction=False):
    db = mongodb_client.getDB()
    prop = json.loads(dumps(db[PROPERTY_TABLE_NAME].find_one({'zpid': zpid})))
    if prop == None:
        prop = zillow_web_scraper_client.get_property_by_zpid(zpid)

    # Get prediction
    if get_prediction:
        predicted_value = ml_prediction_client.predict(
            prop['zipcode'],
            prop['property_type'],
            prop['bedroom'],
            prop['bathroom'],
            prop['size'])
        prop['predicted_value'] = int(predicted_value)
    return prop