def create_viewer(self, data):
v = Viewer(self.master, self.viewers.__len__(), data)
self.master.rowconfigure(self.viewers.__len__() + 1, weight=1)
v.grid(row=self.viewers.__len__() + 1, columns=1, sticky='nswe')
v.draw()
self.viewers.append(v)
class Controller:
def __init__(self):
self.model = Model()
self.viewer = Viewer()
self.viewer.draw(self.model.levels, self.model.people_count,
self.model.position)
self.viewer.functions()
self.check_function()
def check_function(self):
exit = ''
while exit != 'x':
self.ask_motion()
self.viewer.draw(self.model.levels, self.model.people_count,
self.model.position)
self.viewer.functions()
def move(self):
try:
moveput = int(input('Which level would you like to go???'))
except ValueError:
print('Please insert only number.')
if moveput > 0 and moveput < self.model.levels:
self.model.level = moveput
self.model.position = self.model.levels
def ask_motion(self):
motion = input().lower()
if motion in 'arx':
if motion == 'a':
self.model.add_people()
elif motion == 'r':
self.model.remove_people()
else:
self.exit = 'x'
elif isinstance(int(motion), int):
self.move()
