def test_restaurant(self):
r = models.Restaurant(name='test')
db.session.add(r)
db.session.commit()
rs = models.Restaurant.query.all()
self.assertEqual(len(rs), 1)
self.assertEqual(rs[0].name, 'test')
def add_restaurant(restaurant_name):
restaurant = models.Restaurant.query.filter_by(
name=restaurant_name).first()
if not restaurant:
restaurant = models.Restaurant(name=restaurant_name)
db.session.add(restaurant)
db.session.commit()
return restaurant.id
def test_get_user_restaurants(self):
db.create_all()
user1 = models.User(username='******', password='******', location_id=1)
db.session.add(user1)
db.session.commit()
res1 = models.Restaurant(name='Qdoba')
res2 = models.Restaurant(name='Salween Thai')
db.session.add(res1)
db.session.add(res2)
db.session.commit()
user_res1 = models.UserRestaurant(user_id=user1.id,
restaurant_id=res1.id)
user_res2 = models.UserRestaurant(user_id=user1.id,
restaurant_id=res2.id)
db.session.add(user_res1)
db.session.add(user_res2)
db.session.commit()
actual = utils.get_user_restaurants(user1.id)
self.assertEqual(len(actual), 2)
def submitRestaurant(request):
if request.method == 'POST':
restaurant = m.Restaurant()
restaurant.name = request.POST['restaurant-name']
restaurant.address = request.POST['restaurant-address']
restaurant.phoneNumber = request.POST['restaurant-phone']
restaurant.city = 'Seattle'
restaurant.isLive = True
if Restaurant.objects.filter(name=restaurant.name):
messages.add_message(
request,
messages.INFO,
"Looks like this place is popular. It's already on here.",
fail_silently=True)
return render(
request, 'app/submitRestaurant.html', {
'title':
'Submit Restaurant',
'message':
"Looks like this place is popular. It's already on here."
})
else:
restaurant.save()
messages.add_message(request,
messages.INFO,
"Thanks for submitting this restaurant",
fail_silently=True)
return render(
request, 'app/submitRestaurant.html', {
'title': 'Submit Restaurant',
'message': "Thanks for submitting this restaurant"
})
else:
messages
return render(request, 'app/submitRestaurant.html', {
'title': 'Submit Restaurant',
'year': datetime.now().year,
'message': ''
})
def test_user_restaurant(self):
l = models.Location(city='New York City',
region='New York State',
country='United States')
db.session.add(l)
db.session.commit()
u = models.User(username='******', password='******', location_id=l.id)
db.session.add(u)
db.session.commit()
r = models.Restaurant(name='test')
db.session.add(r)
db.session.commit()
user_restaurant = models.UserRestaurant(user_id=u.id,
restaurant_id=r.id)
db.session.add(user_restaurant)
db.session.commit()
user_restaurants = models.UserRestaurant.query.all()
self.assertEqual(len(user_restaurants), 1)
self.assertEqual(user_restaurants[0].user_id, u.id)
self.assertEqual(user_restaurants[0].restaurant_id, r.id)
from app import db
from app import models
db.create_all()
# Restaurants
rest1 = models.Restaurant(name="GoodPlace",
description="Important restaurant with stuff")
rest2 = models.Restaurant(name="Bad Place",
description="Come get some bad food ")
# Menu Items
mi11 = models.Menu_Item(name="Big Dish",
description="Very big and tasty",
price="23.0",
restaurant_id=rest1.id)
mi12 = models.Menu_Item(name="Small Dish",
description="Very small and smelly",
price="3.0",
restaurant_id=rest1.id)
mi21 = models.Menu_Item(name="Bad Dish",
description="Very bad and nasty",
price="23.0",
restaurant_id=rest2.id)
mi22 = models.Menu_Item(name="Small Baddy",
description="Tiny portion",
def main():
with open("./data/businesses_plus.csv") as csvfile:
reader = csv.reader(csvfile, delimiter=',')
i = 0
for data in reader:
i += 1
if i == 1:
continue
if data[0] in rep:
#print rep[data[0]], data
continue
rep[data[0]] = i
rec = models.Restaurant(
id=int(data[0]),
name=preprocess(data[1]),
address=preprocess(data[2]),
postal_code=int(data[4]) if data[4].isdigit() else None,
latitude=float(data[5]) if isFloat(data[5]) else None,
longitude=float(data[6]) if isFloat(data[6]) else None,
phone_number=data[7][1:4] + '-' + data[7][4:7] + '-' +
data[7][7:11] if data[7].isdigit() else None)
db.session.add(rec)
if i % 1000 == 0:
db.session.flush()
db.session.commit()
with open("./data/inspections_plus.csv") as csvfile:
reader = csv.reader(csvfile, delimiter=',')
i = 0
for data in reader:
i += 1
if i == 1:
continue
rec = models.Inspections(
business_id=int(data[0]),
date=datetime.strptime(data[2], '%Y%m%d')
if data[2] != '' else None,
score=int(data[1]) if data[1].isdigit() else None,
insp_type=data[3])
db.session.add(rec)
if i % 1000 == 0:
db.session.flush()
db.session.commit()
with open("./data/violations_plus.csv") as csvfile:
reader = csv.reader(csvfile, delimiter=',')
i = 0
for data in reader:
i += 1
if i == 1:
continue
rec = models.Violations(
business_id=int(data[0]),
date=datetime.strptime(data[1], '%Y%m%d')
if data[1] != '' else None,
typeID=int(data[2]) if data[2].isdigit() else -1,
risk_category=data[3],
description=data[4])
db.session.add(rec)
if i % 1000 == 0:
db.session.flush()
db.session.commit()