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_staff_and_status_filter_by_date_time(
staff_id, status, start_time, end_time):
try:
client = MongoClient(Constant.DB_CONNECTION_URL)
document = client[Constant.DB_NAME][AppointmentDB.COLLECTION_NAME]
appointments = document.find({
'staffId': staff_id,
'status': status,
'dateTime': {
'$gte': start_time,
'$lte': end_time
}
})
appointments_for_staff = []
for appointment in appointments:
a = Appointment(str(appointment['_id']), staff_id,
appointment['visitorId'],
appointment['subject'],
appointment['dateTime'], appointment['status'])
visitor = VisitorDB.get_visitor_by_id(appointment['visitorId'])
a.set_visitor(visitor)
appointments_for_staff.append(a)
client.close()
return appointments_for_staff
except Exception as e:
print('unable to get appointments for staff')
raise Exception('Could not get appointments for staff')
def get_appointments_for_staff_in(staff_id, appointment_ids):
try:
client = MongoClient(Constant.DB_CONNECTION_URL)
document = client[Constant.DB_NAME][AppointmentDB.COLLECTION_NAME]
appointment_ids_obj = [ObjectId(item) for item in appointment_ids]
appointments = document.find({
'_id': {
'$in': appointment_ids_obj
},
'staffId': staff_id
})
appointments_for_staff = []
for appointment in appointments:
a = Appointment(str(appointment['_id']), staff_id,
appointment['visitorId'],
appointment['subject'],
appointment['dateTime'], appointment['status'])
visitor = VisitorDB.get_visitor_by_id(appointment['visitorId'])
a.set_visitor(visitor)
appointments_for_staff.append(a)
client.close()
return appointments_for_staff
except Exception as e:
print('unable to get appointments for staff - ' + str(e))
raise Exception('Could not get appointments for staff')
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')
