/
lunch_setup.py
29 lines (24 loc) · 933 Bytes
/
lunch_setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
import os.path
import restaurant
import restaurant_manager
import person_manager
def writeOutRestaurants(restaurant_manager):
restaurant_manager.addRestaurant(restaurant.Restaurant.fromName("Cape of Good Hope"))
restaurant_manager.addRestaurant(restaurant.Restaurant.fromName("Cocos"))
restaurant_manager.saveRestaurants()
def writeOutPeople(person_manager):
person_manager.addPerson('thk123')
person_manager.savePeople()
def runSetup(restaurant_manager, person_manager):
if not os.path.isfile(restaurant_manager.restaurantsFile):
print "Running first time setup for restaurants"
writeOutRestaurants(restaurant_manager)
else:
print "Loading restaurants"
restaurant_manager.loadRestaurants()
if not os.path.isfile(person_manager.person_file):
print "Running first time setup for people`"
writeOutPeople(person_manager)
else:
print "Loading people"
person_manager.loadPeople()