def add_place(name, phone, lat, lon, address, fs_id=None, yelp_id=None, wifi=None, plugs=None, exp=None): # ensure it has a fs_id or a yelp_id if not (fs_id or yelp_id): return ErrorResponse.POST_INVALID_ARGS_FS_ID_YELP_ID # Create a review dict review = make_review_dict(wifi, plugs, exp) # Check if place already exists in the database if fs_id: spec = {Place.A_FS_ID: fs_id} place_dict = Place.mdbc().find_one(spec) # If place exists add a review to it if place_dict: place_id = place_dict.get(Place.A_FS_ID) return Place.review_place(place_id=place_id, review=review) # If place doesn't already exist in the database add it place_data = { Place.A_NAME: name, Place.A_FS_ID: fs_id, Place.A_YELP_ID: yelp_id, Place.A_ADDRESS: address, Place.A_LOCATION: [float(lat), float(lon)] } # Only add phone if it's not null if phone: place_data[Place.A_PHONE] = phone return Place.add_place(place_data=place_data, review=review)
def test_post(): if not Place.mdbc().count() == 0: Place.mdbc().remove() # Add new place url = "http://localhost/places" payload = { "fs_id": "aslfkj13409182491k1we1-12e801", "name": "Ninth Street Espresso", "phone": "2129201880", "lat": "42.881111", "lon": "78.191111", "address": "180 Crosby Street", "wifi": "2", "plugs": "1", } r = requests.post(url=url, data=payload) spec = r.content spec_js = json.loads(spec) place_id = spec_js.get("place_id") # Add another review to previous place payload = {"place_id": place_id, "wifi": "0", "plugs": "0"} r = requests.post(url=url, data=payload) # Add another review to previous place payload = {"place_id": place_id, "wifi": "0", "plugs": "0"} r = requests.post(url=url, data=payload) # Add new place payload = { "fs_id": "12e801", "name": "The Bean", "phone": "2129201881", "lat": "42.881111", "lon": "78.191111", "address": "180 Crosby Street", "wifi": "2", "plugs": "1", } r = requests.post(url=url, data=payload) r.content # Add new place payload = { "fs_id": "12801", "name": "The Native Bean", "phone": "2129202880", "lat": "42.881111", "lon": "78.191111", "address": "180 Crosby Street", "wifi": "1", "plugs": "1", } r = requests.post(url=url, data=payload) r.content # Add new place payload = { "fs_id": "12801", "name": "Gimme Coffee", "phone": "2129201980", "lat": "42.881111", "lon": "78.191111", "address": "180 Crosby Street", "wifi": "0", "plugs": "1", } r = requests.post(url=url, data=payload) r.content # Add new place payload = { "fs_id": "12801", "name": "Gimme Coffee Now", "phone": "3129201980", "lat": "43.881111", "lon": "79.191111", "address": "180 Crosby Street", "wifi": "0", "plugs": "1", } r = requests.post(url=url, data=payload) r.content print "Database_count = " + str(Place.mdbc().count())