def setup(self):
"""
Sets up a test database
"""
campus1 = Campus(1, 'test campus')
rest1 = Restaurant(1, 'test r1', 'student centre', 1, time(hour=8),
time(hour=17), 750)
rest2 = Restaurant(2, 'test r2', 'ballymun road', 1, time(hour=9),
time(hour=18), 250)
campus1.save()
rest1.save()
rest2.save()
def part2_setup(self):
"""
Sets up a test database - for testing part two
"""
campus1 = Campus(1, 'test campus')
campus2 = Campus(2, 'another test campus')
rest1 = Restaurant(1, 'test r1', 'student centre', 1, time(hour=8),
time(hour=17), 750, False)
rest2 = Restaurant(2, 'test r2', 'ballymun road', 1, time(hour=9),
time(hour=18), 250, True)
rest3 = Restaurant(3, 'test r3', 'in library', 1, time(hour=10),
time(hour=17), 300, False)
rest4 = Restaurant(4, 'test r4', 'beside entrance', 2, time(hour=10),
time(hour=16), 200, True)
campus1.save()
campus2.save()
rest1.save()
rest2.save()
rest3.save()
rest4.save()
def setup(self):
"""
Sets up a test database
"""
campus1 = Campus(1, 'test campus')
campus2 = Campus(2, 'another test campus')
campus3 = Campus(3, 'yet another test campus')
rest1 = Restaurant(1, 'test r1', 'student centre', 1, time(hour=8),
time(hour=17), 750)
rest2 = Restaurant(2, 'test r2', 'ballymun road', 1, time(hour=9),
time(hour=18), 250)
rest3 = Restaurant(3, 'test r3', 'in library', 1, time(hour=10),
time(hour=17), 300)
rest4 = Restaurant(4, 'test r4', 'beside entrance', 3, time(hour=10),
time(hour=16), 200)
campus1.save()
campus2.save()
campus3.save()
rest1.save()
rest2.save()
rest3.save()
rest4.save()
def part3_setup(self):
"""
Sets up a test database - for testing part three
"""
campus1 = Campus(1, 'test campus')
campus2 = Campus(2, 'another test campus')
campus3 = Campus(3, 'yet another test campus')
rest1 = Restaurant(1, 'test r1', 'student centre', 1, time(hour=8),
time(hour=16), 750, False, True,
time(hour=7, minute=15), time(hour=17, minute=34))
rest2 = Restaurant(2, 'test r2', 'ballymun road', 1, time(hour=9),
time(hour=18), 250, False, True,
time(hour=6, minute=12), time(hour=19, minute=42))
rest3 = Restaurant(3, 'test r3', 'in library', 2, time(hour=10),
time(hour=17), 300)
rest4 = Restaurant(4, 'test r4', 'beside entrance', 3, time(hour=10),
time(hour=16), 200, True, False)
campus1.save()
campus2.save()
campus3.save()
rest1.save()
rest2.save()
rest3.save()
rest4.save()
import csv
from datetime import time
from eatatdcu.models import Campus, Restaurant
with open('../data/campus.csv') as ca:
reader = csv.reader(ca)
for row in reader:
campus = Campus(int(row[0]), row[1], row[2])
campus.save()
with open('../data/restaurant.csv') as re:
reader = csv.reader(re)
for row in reader:
opening = row[4].split(':')
closing = row[5].split(':')
we_opening = row[9].split(':')
we_closing = row[10].split(':')
restaurant = Restaurant(
int(row[0]), row[1], row[2], int(row[3]),
time(hour=int(opening[0]), minute=int(opening[1])),
time(hour=int(closing[0]), minute=int(closing[1])), int(row[6]),
int(row[7]), int(row[8]),
time(hour=int(we_opening[0]), minute=int(we_opening[1])),
time(hour=int(we_closing[0]), minute=int(we_closing[1])))
restaurant.save()
def part3_setup_weekends(self):
campus1 = Campus(1, 'test campus')
campus2 = Campus(2, 'another test campus')
campus3 = Campus(3, 'yet another test campus')
rest1 = Restaurant(1, 'test r1', 'student centre', 1, time(hour=8),
time(hour=16), 750, 0, 1, 1, time(hour=7,
minute=15),
time(hour=17, minute=34))
rest2 = Restaurant(2, 'test r2', 'ballymun road', 1, time(hour=9),
time(hour=18), 250, 0, 0, 1, time(hour=6,
minute=12),
time(hour=19, minute=42))
rest3 = Restaurant(3, 'test r3', 'in library', 1, time(hour=10),
time(hour=17), 300)
rest4 = Restaurant(4, 'test r4', 'beside entrance', 3, time(hour=10),
time(hour=16), 200)
campus1.save()
campus2.save()
campus3.save()
rest1.save()
rest2.save()
rest3.save()
rest4.save()
