def test_saving_and_retrieving_items(self): first_item = Bar() first_item.id = '123456789' first_item.save() second_item = Bar() second_item.id = '987654321' second_item.save() saved_items = Bar.objects.all() self.assertEqual(saved_items.count(), 2) first_saved_item = saved_items[0] second_saved_item = saved_items[1] self.assertEqual(first_saved_item.id, '123456789') self.assertEqual(second_saved_item.id, '987654321')
def create_bar_in_db(current_bar): b = Bar() b.google_id = current_bar.google_id b.name = current_bar.name b.stripped_name = current_bar.stripped_name b.latitude = current_bar.latitude b.longitude = current_bar.longitude if hasattr(current_bar, 'vicinity'): b.vicinity = current_bar.vicinity if hasattr(current_bar, 'price_level'): b.price_level = current_bar.price_level if hasattr(current_bar, 'rating'): b.rating = current_bar.rating b.creation_date = timezone.now() if hasattr(current_bar, 'open_at_update'): b.open_at_update = current_bar.open_at_update b.save()
def create_bar_in_db(current_bar): """takes in current_bar object, and creates an item in the database with the desired attributes""" b = Bar() b.google_id = current_bar.google_id b.name = current_bar.name b.stripped_name = current_bar.stripped_name b.latitude = current_bar.latitude b.longitude = current_bar.longitude if hasattr(current_bar, 'vicinity'): b.vicinity = current_bar.vicinity if hasattr(current_bar, 'price_level'): b.price_level = current_bar.price_level if hasattr(current_bar, 'rating'): b.rating = current_bar.rating b.creation_date = timezone.now() if hasattr(current_bar, 'open_at_update'): b.open_at_update = current_bar.open_at_update b.save()