def processAddRestaurant(tweet, twitter): contents = tweet['text'] m = re.search('restaurant ([\w\s\',]+)$', contents) if m: print("adding restaurant: " + m.group(1)) new_restaurant = restaurant.Restaurant.fromName(m.group(1)) did_add_restaurant = restaurant_manager.addRestaurant(new_restaurant) if did_add_restaurant: restaurant_manager.saveRestaurants() replyToTweet(tweet, "Ooo, good choice, I'd be salvating if I was a human", twitter) else: replyToTweet(tweet, "So good you want to add it twice?", twitter) else: print("Error passsing match " + contents) replyToTweet(tweet, "Sorry, that isn't a valid restaurant command", twitter)
def writeOutRestaurants(restaurant_manager): restaurant_manager.addRestaurant(restaurant.Restaurant.fromName("Cape of Good Hope")) restaurant_manager.addRestaurant(restaurant.Restaurant.fromName("Cocos")) restaurant_manager.saveRestaurants()