def export_excell(self):
filename = str(QtGui.QFileDialog.getSaveFileName(self, 'Export population Excell File', '.xls'))
if filename:
saver = PopulationSavingObject(self._institution.people, filename)
saver.create()
saver.populate()
filename = str(QtGui.QFileDialog.getSaveFileName(self, 'Export duration Excell File', '.xls'))
if filename:
saver = DurationSavingObject(self.durations_widget.singleDurationWidget.duration, filename)
saver.create()
saver.populate()
def import_excell(self):
self.model.clear_stack()
filename = str(QtGui.QFileDialog.getOpenFileName(self, 'Import population Excell File', '.xls'))
if filename:
saver = PopulationSavingObject([], filename)
people = saver.load()
self._institution.people += people
filename = str(QtGui.QFileDialog.getOpenFileName(self, 'Import duration Excell File', '.xls'))
if filename:
saver = DurationSavingObject(self.durations_widget.singleDurationWidget.duration, filename)
saver.load(self._institution.people)
self.institution(self._institution)
def testSaveLoadPopulation(self):
self.institution.people.append(new_worker())
saver = PopulationSavingObject(self.institution.people, 'temp_sheet.xls')
saver.create()
saver.populate()
loader = PopulationSavingObject([], 'temp_sheet.xls')
loaded_population = loader.load()
self.assertEqual(len(loaded_population), 1)
loaded_person = loaded_population[0]
self.assertEqual(loaded_person.name, 'Bob')
self.assertEqual(loaded_person.address, 'a')
self.assertEqual(loaded_person.email, 'b')
self.assertEqual(loaded_person.phone_number, '0115')
self.assertEqual(len(loaded_person.roles()), 2)
self.assertTrue(loaded_person.suitable_for_role('B'))
self.assertTrue(loaded_person.suitable_for_role('S'))
self.assertFalse(loaded_person.suitable_for_role('F'))
def testLoadPopulation(self):
GlobalRoleList.clear()
GlobalRoleList.add_role(Role('Doctor', 'D', 10))
GlobalRoleList.add_role(Role('GP', 'G', 8))
GlobalRoleList.add_role(Role('Nurse', 'N', 10))
GlobalRoleList.add_role(Role('Cook', 'C', 7))
GlobalRoleList.add_role(Role('Reception', 'R', 5))
loader = PopulationSavingObject([], 'test_population_sheet.xls')
loaded_population = loader.load()
self.assertEqual(len(loaded_population), 6)
loaded_person = loaded_population[0]
self.assertEqual(loaded_person.name, 'Mary')
self.assertEqual(loaded_person.address, 'A34')
self.assertEqual(loaded_person.email, 'b')
self.assertEqual(loaded_person.phone_number, '122')
self.assertEqual(len(loaded_person.roles()), 2)
self.assertTrue(loaded_person.suitable_for_role('D'))
self.assertTrue(loaded_person.suitable_for_role('N'))
self.assertFalse(loaded_person.suitable_for_role('C'))
loaded_person = loaded_population[1]
self.assertEqual(loaded_person.name, 'Jane')
self.assertEqual(loaded_person.address, 'B32')
self.assertEqual(loaded_person.email, 'g')
self.assertEqual(loaded_person.phone_number, '463')
self.assertEqual(len(loaded_person.roles()), 1)
self.assertFalse(loaded_person.suitable_for_role('D'))
self.assertTrue(loaded_person.suitable_for_role('G'))
self.assertFalse(loaded_person.suitable_for_role('C'))
loaded_person = loaded_population[2]
self.assertEqual(loaded_person.name, 'Mick')
self.assertEqual(len(loaded_person.roles()), 1)
self.assertFalse(loaded_person.suitable_for_role('D'))
self.assertTrue(loaded_person.suitable_for_role('C'))
self.assertFalse(loaded_person.suitable_for_role('G'))
loaded_person = loaded_population[3]
self.assertEqual(loaded_person.name, 'John')
self.assertEqual(len(loaded_person.roles()), 2)
self.assertTrue(loaded_person.suitable_for_role('R'))
self.assertTrue(loaded_person.suitable_for_role('N'))
self.assertFalse(loaded_person.suitable_for_role('D'))
loaded_person = loaded_population[4]
self.assertEqual(loaded_person.name, 'Gaz')
self.assertEqual(len(loaded_person.roles()), 1)
self.assertFalse(loaded_person.suitable_for_role('D'))
self.assertFalse(loaded_person.suitable_for_role('N'))
self.assertTrue(loaded_person.suitable_for_role('C'))
loaded_person = loaded_population[5]
self.assertEqual(loaded_person.name, 'Toby')
self.assertEqual(loaded_person.address, 'C16')
self.assertEqual(loaded_person.email, '')
self.assertEqual(loaded_person.phone_number, '400')
self.assertEqual(len(loaded_person.roles()), 2)
self.assertTrue(loaded_person.suitable_for_role('D'))
self.assertTrue(loaded_person.suitable_for_role('G'))
self.assertFalse(loaded_person.suitable_for_role('N'))
def testSavePopulation(self):
self.institution.people.append(new_worker())
saver = PopulationSavingObject(self.institution.people, 'temp_sheet.xls')
saver.create()
saver.populate()
