def validate_model(self):
id = get_mdict_item_or_list(request.args, 'id')
if id is None:
return False
# Get Invited and Being Served states, to see if CSR has any open tickets.
period_state_invited = PeriodState.get_state_by_name("Invited")
period_state_being_served = PeriodState.get_state_by_name(
"Being Served")
# See if CSR has any open tickets.
citizen = Citizen.query \
.join(Citizen.service_reqs) \
.join(ServiceReq.periods) \
.filter(Period.time_end.is_(None)) \
.filter(Period.csr_id==id) \
.filter(or_(Period.ps_id==period_state_invited.ps_id, Period.ps_id==period_state_being_served.ps_id)) \
.all()
if len(citizen) != 0:
flash(gettext('CSR has an open ticket and cannot be edited.'),
'error')
return False
if not self.can_edit:
return False
model = self.get_one(id)
if model is None:
flash(gettext('Record does not exist.'), 'error')
return False
return model
def invite(self, csr, snowplow_event="use_period"):
active_period = self.get_active_period()
if active_period.ps.ps_name in ["Invited", "Being Served", "On hold"]:
raise TypeError("You cannot invite a citizen that has already been invited")
# Calculate what Snowplow event to call.
if (snowplow_event == "use_period"):
if (active_period.ps.ps_name == "Waiting"):
snowplow_event = "invitefromlist"
else:
snowplow_event = "invitefromhold"
active_period.time_end = datetime.now()
# db.session.add(active_period)
period_state_invite = PeriodState.get_state_by_name("Invited")
new_period = Period(
sr_id=self.sr_id,
csr_id=csr.csr_id,
reception_csr_ind=csr.receptionist_ind,
ps_id=period_state_invite.ps_id,
time_start=datetime.now()
)
self.periods.append(new_period)
SnowPlow.snowplow_event(self.citizen_id, csr, snowplow_event)
def post(self, id):
csr = CSR.find_by_username(g.oidc_token_info['username'])
service_request = ServiceReq.query.filter_by(sr_id=id) \
.join(ServiceReq.citizen, aliased=True) \
.filter_by(office_id=csr.office_id).first_or_404()
active_service_state = SRState.get_state_by_name("Active")
complete_service_state = SRState.get_state_by_name("Complete")
# Find the currently active service_request and close it
for req in service_request.citizen.service_reqs:
if req.sr_state_id == active_service_state.sr_state_id:
req.sr_state_id = complete_service_state.sr_state_id
req.finish_service(csr, clear_comments=False)
db.session.add(req)
# Then set the requested service to active
service_request.sr_state_id = active_service_state.sr_state_id
period_state_being_served = PeriodState.get_state_by_name(
"Being Served")
new_period = Period(sr_id=service_request.sr_id,
csr_id=csr.csr_id,
reception_csr_ind=csr.receptionist_ind,
ps_id=period_state_being_served.ps_id,
time_start=datetime.now())
db.session.add(new_period)
db.session.add(service_request)
citizen_obj = Citizen.query.get(service_request.citizen_id)
citizen_obj.service_count = citizen_obj.service_count + 1
db.session.commit()
SnowPlow.choose_service(service_request, csr, "additionalservice")
citizen_result = self.citizen_schema.dump(service_request.citizen)
socketio.emit('update_active_citizen',
citizen_result.data,
room=csr.office_id)
result = self.service_request_schema.dump(service_request)
return {'service_request': result.data, 'errors': result.errors}, 200
def add_to_queue(self, csr, snowplow_event):
active_period = self.get_active_period()
active_period.time_end = datetime.now()
#db.session.add(active_period)
period_state_waiting = PeriodState.get_state_by_name("Waiting")
new_period = Period(
sr_id=self.sr_id,
csr_id=csr.csr_id,
reception_csr_ind=csr.receptionist_ind,
ps_id=period_state_waiting.ps_id,
time_start=datetime.now()
)
self.periods.append(new_period)
SnowPlow.snowplow_event(self.citizen_id, csr, snowplow_event)
def place_on_hold(self, csr):
active_period = self.get_active_period()
active_period.time_end = datetime.now()
# db.session.add(active_period)
period_state_on_hold = PeriodState.get_state_by_name("On hold")
new_period = Period(
sr_id=self.sr_id,
csr_id=csr.csr_id,
reception_csr_ind=csr.receptionist_ind,
ps_id=period_state_on_hold.ps_id,
time_start=datetime.now()
)
self.periods.append(new_period)
SnowPlow.snowplow_event(self.citizen_id, csr, "hold")
def begin_service(self, csr, snowplow_event):
active_period = self.get_active_period()
if active_period.ps.ps_name in ["Being Served"]:
raise TypeError("You cannot begin serving a citizen that is already being served")
active_period.time_end = datetime.now()
# db.session.add(active_period)
period_state_being_served = PeriodState.get_state_by_name("Being Served")
new_period = Period(
sr_id=self.sr_id,
csr_id=csr.csr_id,
reception_csr_ind=csr.receptionist_ind,
ps_id=period_state_being_served.ps_id,
time_start=datetime.now()
)
self.periods.append(new_period)
# Calculate number of active periods, for Snowplow call.
period_count = len(self.periods)
SnowPlow.snowplow_event(self.citizen_id, csr, snowplow_event, period_count = period_count)
def post(self):
json_data = request.get_json()
if not json_data:
return {"message": "No input data received for creating service request"}, 400
csr = CSR.find_by_username(g.oidc_token_info['username'])
try:
service_request = self.service_request_schema.load(json_data['service_request']).data
except ValidationError as err:
return {"message": err.messages}, 422
except KeyError as err:
print (err)
return {"message": str(err)}
active_sr_state = SRState.get_state_by_name("Active")
complete_sr_state = SRState.get_state_by_name("Complete")
citizen_state = CitizenState.query.filter_by(cs_state_name="Active").first()
citizen = Citizen.query.get(service_request.citizen_id)
service = Service.query.get(service_request.service_id)
if citizen is None:
return {"message": "No matching citizen found for citizen_id"}, 400
if service is None:
return {"message": "No matching service found for service_id"}, 400
# Find the currently active service_request and close it (if it exists)
for req in citizen.service_reqs:
if req.sr_state_id == active_sr_state.sr_state_id:
req.sr_state_id = complete_sr_state.sr_state_id
req.finish_service(csr, clear_comments=False)
db.session.add(req)
service_request.sr_state_id = active_sr_state.sr_state_id
# Only add ticket creation period and ticket number if it's their first service_request
if len(citizen.service_reqs) == 0:
period_state_ticket_creation = PeriodState.get_state_by_name("Ticket Creation")
ticket_create_period = Period(
csr_id=csr.csr_id,
reception_csr_ind=csr.receptionist_ind,
ps_id=period_state_ticket_creation.ps_id,
time_start=citizen.get_service_start_time(),
time_end=datetime.now()
)
service_request.periods.append(ticket_create_period)
# Move start_time back 6 hours to account for DST and UTC offsets
# It's only important that the number carries over _around_ midnight
offset_start_time = citizen.start_time - timedelta(hours=6)
service_count = ServiceReq.query \
.join(ServiceReq.citizen, aliased=True) \
.filter(Citizen.start_time >= offset_start_time.strftime("%Y-%m-%d")) \
.filter_by(office_id=csr.office_id) \
.join(ServiceReq.service, aliased=True) \
.filter_by(prefix=service.prefix) \
.count()
citizen.ticket_number = service.prefix + str(service_count)
else:
period_state_being_served = PeriodState.get_state_by_name("Being Served")
ticket_create_period = Period(
csr_id=csr.csr_id,
reception_csr_ind=csr.receptionist_ind,
ps_id=period_state_being_served.ps_id,
time_start=datetime.now()
)
service_request.periods.append(ticket_create_period)
citizen.cs_id = citizen_state.cs_id
# See whether first service, or next service.
if len(citizen.service_reqs) == 0:
snowplow_event = "chooseservice"
citizen.service_count = 1
else:
snowplow_event = "additionalservice"
citizen.service_count = citizen.service_count + 1
db.session.add(service_request)
db.session.add(citizen)
db.session.commit()
SnowPlow.choose_service(service_request, csr, snowplow_event)
citizen_result = self.citizen_schema.dump(citizen)
socketio.emit('update_active_citizen', citizen_result.data, room=csr.office_id)
result = self.service_request_schema.dump(service_request)
return {'service_request': result.data,
'errors': result.errors}, 201
def post(self):
lock = FileLock("lock/invite_citizen.lock")
with lock:
csr = CSR.find_by_username(g.oidc_token_info['username'])
active_citizen_state = CitizenState.query.filter_by(
cs_state_name='Active').first()
waiting_period_state = PeriodState.get_state_by_name("Waiting")
citizen = None
try:
qt_xn_csr_ind = request.get_json().get('qt_xn_csr_ind')
except AttributeError:
qt_xn_csr_ind = csr.qt_xn_csr_ind
if qt_xn_csr_ind:
citizen = Citizen.query \
.filter_by(qt_xn_citizen_ind=1, cs_id=active_citizen_state.cs_id, office_id=csr.office_id) \
.join(Citizen.service_reqs) \
.join(ServiceReq.periods) \
.filter_by(ps_id=waiting_period_state.ps_id) \
.filter(Period.time_end.is_(None)) \
.order_by(Citizen.priority, Citizen.citizen_id) \
.first()
else:
citizen = Citizen.query \
.filter_by(qt_xn_citizen_ind=0, cs_id=active_citizen_state.cs_id, office_id=csr.office_id) \
.join(Citizen.service_reqs) \
.join(ServiceReq.periods) \
.filter_by(ps_id=waiting_period_state.ps_id) \
.filter(Period.time_end.is_(None)) \
.order_by(Citizen.priority, Citizen.citizen_id) \
.first()
# Either no quick txn citizens for the quick txn csr, or vice versa
if citizen is None:
citizen = Citizen.query \
.filter_by(cs_id=active_citizen_state.cs_id, office_id=csr.office_id) \
.join(Citizen.service_reqs) \
.join(ServiceReq.periods) \
.filter_by(ps_id=waiting_period_state.ps_id) \
.filter(Period.time_end.is_(None)) \
.order_by(Citizen.priority, Citizen.citizen_id) \
.first()
if citizen is None:
return {"message": "There is no citizen to invite"}, 400
db.session.refresh(citizen)
active_service_request = citizen.get_active_service_request()
try:
active_service_request.invite(csr, "invitecitizen")
except TypeError:
return {
"message": "Error inviting citizen. Please try again."
}, 400
active_service_state = SRState.get_state_by_name("Active")
active_service_request.sr_state_id = active_service_state.sr_state_id
db.session.add(citizen)
db.session.commit()
socketio.emit('update_customer_list', {}, room=csr.office_id)
socketio.emit('citizen_invited', {},
room='sb-%s' % csr.office.office_number)
result = self.citizen_schema.dump(citizen)
socketio.emit('update_active_citizen',
result.data,
room=csr.office_id)
return {'citizen': result.data, 'errors': result.errors}, 200