def _insertShiftPerson(shiftPerson, db):
result = 'Unknown'
isDup = _isDuplicateAssignment(shiftPerson)
shift = shifts_controller._getShiftByID(shiftPerson.shiftFK)
if shift is None:
result = 'BadShift'
person = models.Person.query.filter_by(entityFK=shiftPerson.personFK).first()
if person is None:
result = 'BadPerson'
if(isDup == False and result == 'Unknown'):
db.session.add(shiftPerson)
db.session.commit()
if(shiftPerson.pk > 0):
result = shiftPerson.pk
if(result == 'Unknown'):
result = 'Duplicate'
return result
def getShiftsByPersonJSON(personFK):
list = _getShiftsByPerson(personFK)
resultJSON = '{"Shifts":'
counter = 0
if list.count() > 0:
resultJSON += '['
for shiftPerson in list:
shift = shifts_controller._getShiftByID(shiftPerson.shiftFK)
shiftJSON = shifts_controller.shiftToJSON(shift)
if counter > 0:
resultJSON += ','
resultJSON += shiftJSON
counter += 1
resultJSON += ']'
else:
resultJSON += '"None"'
resultJSON += '}'
return resultJSON
def test__getShiftByID_false(self):
key = 9999
shift = shifts_controller._getShiftByID(key)
self.assertIsNone(shift)
def test__getShiftByID_true(self):
key = 1
shift = shifts_controller._getShiftByID(key)
self.assertIsNotNone(shift)
control = models.Shift.query.filter_by(pk=shift.pk).first()
self.assertEqual(shift.__repr__(), control.__repr__())