def _get_people(self): self._population = [] for i in range(1, self._population_sheet.nrows): if self._population_sheet.cell_type(i, 0) is 0: break else: p = Worker() p.name = self._population_sheet.cell_value(i, 0) p.phone_number = self._population_sheet.cell_value(i, 1) p.email = self._population_sheet.cell_value(i, 2) p.address = self._population_sheet.cell_value(i, 3) self._get_qualifications(i, p) self._get_dates(i, p) if p.name in map(name, self._population): raise ExcellImportExportError('There were people with the same name : ' + p.name) else: self._population.append(p)
def _loadInPopulationTables(self): _c = self._connection.cursor() self._institution.people = [] personCacheDict = {'Name': 'PersonObject'} for row in _c.execute('SELECT * FROM Population ORDER BY name'): p = Worker() p.name = row[0] p.address = row[1] p.email = row[2] p.phoneNumber = row[3] self._institution.people.append(p) personCacheDict[p.name] = p for row in _c.execute('SELECT * FROM PopulationAvailability'): personCacheDict[row[0]].blacklist_date(row[1]) for row in _c.execute('SELECT * FROM PopulationQualifications'): r = GlobalRoleList.role_from_desc(row[1]) personCacheDict[row[0]].roles().add(r) self._personCacheDict = personCacheDict
def main(): GlobalRoleList.add_role(Role("Baker", "B", 2)) GlobalRoleList.add_role(Role("Steward", "S", 9)) GlobalRoleList.add_role(Role("Fisherman", "F", 7)) bob = Worker() bob.name = "Bob" bob.email = "*****@*****.**" bob.phoneNumber = "0116" testdate = date(2012, 12, 31) bob.blacklist_date(testdate) app = QtGui.QApplication(sys.argv) w = PersonWidget(None) w.setWindowTitle("Person") w.person(bob) w.show() sys.exit(app.exec_())
def newWorker(): bob = Worker() bob.name = ('Bob') bob.does_nothing() bob.add_role('S') bob.add_role('B') testDate = date(2012, 12, 31) testDate2 = date(2011, 12, 31) bob.clear_blacklist() bob.blacklist_date(testDate) bob.blacklist_date(testDate2) bob.phone_number = '0115' bob.address = 'a' bob.email = 'b' return bob
def setUp(self): self.bob = Worker() self.bob.name = 'Bob' Roles.GlobalRoleList.add_role(Roles.Role('Plumber', 'P', 1)) Roles.GlobalRoleList.add_role(Roles.Role('Sailor', 'S', 1)) Roles.GlobalRoleList.add_role(Roles.Role('Treasurer', 'T', 9))
class WorkerTest(unittest.TestCase): def setUp(self): self.bob = Worker() self.bob.name = 'Bob' Roles.GlobalRoleList.add_role(Roles.Role('Plumber', 'P', 1)) Roles.GlobalRoleList.add_role(Roles.Role('Sailor', 'S', 1)) Roles.GlobalRoleList.add_role(Roles.Role('Treasurer', 'T', 9)) def tearDown(self): self.bob = None Roles.GlobalRoleList.clear() def testName(self): self.assertEqual(self.bob.name, 'Bob') def testNoRole(self): self.bob.does_nothing() self.assertEqual(self.bob.number_of_roles(), 0) def testAddRole(self): self.bob.does_nothing() self.bob.add_role('S') self.assertEqual(self.bob.number_of_roles(), 1) self.bob.add_role('P') self.assertEqual(self.bob.number_of_roles(), 2) self.bob.add_role('S') self.assertEqual(self.bob.number_of_roles(), 2) def testRemoveRole(self): self.bob.does_nothing() self.bob.add_roles('S P T P') self.assertEqual(self.bob.number_of_roles(), 3) self.bob.remove_role('P') self.assertEqual(self.bob.number_of_roles(), 2) self.bob.remove_role('S') self.assertEqual(self.bob.number_of_roles(), 1) def testSuitableForRole(self): self.bob.does_nothing() self.bob.add_roles('S P T P') self.assertTrue(self.bob.suitable_for_role('P')) self.assertTrue(self.bob.suitable_for_role('S')) self.assertTrue(self.bob.suitable_for_role('T')) self.bob.remove_role('P') self.assertFalse(self.bob.suitable_for_role('P')) self.assertTrue(self.bob.suitable_for_role('S')) self.assertTrue(self.bob.suitable_for_role('T')) self.bob.remove_role('S') self.assertFalse(self.bob.suitable_for_role('P')) self.assertFalse(self.bob.suitable_for_role('S')) self.assertTrue(self.bob.suitable_for_role('T'))