def search_all_appointments_visitors_name(visitor_name_keyword):
try:
client = MongoClient(Constant.DB_CONNECTION_URL)
document = client[Constant.DB_NAME][AppointmentDB.COLLECTION_NAME]
appointments = document.find({})
appointments_for_staff = []
for appointment in appointments:
a = Appointment(str(appointment['_id']),
appointment['staffId'],
appointment['visitorId'],
appointment['subject'],
appointment['dateTime'], appointment['status'])
visitor = VisitorDB.get_visitor_by_id(appointment['visitorId'])
if visitor_name_keyword.lower() not in visitor.name.lower():
continue
a.set_visitor(visitor)
staff = StaffDB.get_staff_by_id(appointment['staffId'])
a.set_staff(staff.serialize)
appointments_for_staff.append(a)
client.close()
return appointments_for_staff
except Exception as e:
print('unable to get appointments')
raise Exception('Could not get appointments')
def get_appointments_for_status(status):
try:
client = MongoClient(Constant.DB_CONNECTION_URL)
document = client[Constant.DB_NAME][AppointmentDB.COLLECTION_NAME]
appointments = document.find({'status': status})
appointments_for_status = []
for appointment in appointments:
a = Appointment(str(appointment['_id']),
appointment['staffId'],
appointment['visitorId'],
appointment['subject'],
appointment['dateTime'], status)
visitor = VisitorDB.get_visitor_by_id(appointment['visitorId'])
a.set_visitor(visitor)
staff = StaffDB.get_staff_by_id(appointment['staffId'])
a.set_staff(staff.serialize)
appointments_for_status.append(a)
client.close()
return appointments_for_status
except Exception as e:
print('unable to get appointments for staff')
raise Exception('Could not get appointments for staff')
