def send_message(recipients, virtual_tn, msg, session_id, is_system_msg=False):
"""
For each recipient, passes a Message to Flowroute's messaging controller.
The message will be sent from the 'virtual_tn' number. If this is a system
message, the message body will be prefixed with the org name for context.
If an exception is raised by the controller, an error is logged, and an
internal error is raised with the exception content.
"""
if is_system_msg:
msg = "[{}]: {}".format(ORG_NAME.upper(), msg)
for recipient in recipients:
message = Message(to=recipient, from_=virtual_tn, content=msg)
try:
app.sms_controller.create_message(message)
except Exception as e:
strerr = vars(e).get("response_body", None)
log.critical(
{
"message": "Raised an exception sending SMS",
"status": "failed",
"exc": e,
"strerr": vars(e).get("response_body", None),
}
)
raise InternalSMSDispatcherError(
"An error occured when requesting against Flowroute's API.",
payload={"strerr": strerr, "reason": "InternalSMSDispatcherError"},
)
else:
log.info(
{"message": "Message sent to {} for session {}".format(recipient, session_id), "status": "succeeded"}
)
def send_message(recipients, virtual_tn, msg, session_id, is_system_msg=False):
"""
For each recipient, passes a Message to Flowroute's messaging controller.
The message will be sent from the 'virtual_tn' number. If this is a system
message, the message body will be prefixed with the org name for context.
If an exception is raised by the controller, an error is logged, and an
internal error is raised with the exception content.
"""
if is_system_msg:
msg = "[{}]: {}".format(ORG_NAME.upper(), msg)
for recipient in recipients:
message = Message(to=recipient, from_=virtual_tn, content=msg)
try:
app.sms_controller.create_message(message)
except Exception as e:
strerr = vars(e).get('response_body', None)
log.critical({
"message": "Raised an exception sending SMS",
"status": "failed",
"exc": e,
"strerr": vars(e).get('response_body', None)
})
raise InternalSMSDispatcherError(
"An error occured when requesting against Flowroute's API.",
payload={
"strerr": strerr,
"reason": "InternalSMSDispatcherError"
})
else:
log.info({
"message":
"Message sent to {} for session {}".format(
recipient, session_id),
"status":
"succeeded"
})
def add_proxy_session():
"""
The ProxySession resource endpoint for adding a new ProxySession
to the pool.
"""
body = request.json
try:
participant_a = body['participant_a']
participant_b = body['participant_b']
assert len(participant_a) <= 18
assert len(participant_b) <= 18
except (AssertionError, KeyError):
raise InvalidAPIUsage(
("Required argument: 'participant_a' (str, length <= 18)"
", 'participant_b' (str, length <= 18)"),
payload={'reason': 'invalidAPIUsage'})
if 'expiry_window' in body:
expiry_window = body['expiry_window']
else:
expiry_window = None
# Release any VirtualTNs from expired ProxySessions back to the pool
ProxySession.clean_expired()
virtual_tn = VirtualTN.get_next_available()
if virtual_tn is None:
msg = "Could not create a new session -- No virtual TNs available."
log.critical({"message": msg, "status": "failed"})
return Response(json.dumps({
"message": msg,
"status": "failed"
}),
content_type="application/json",
status=400)
else:
session = ProxySession(virtual_tn.value, participant_a, participant_b,
expiry_window)
try:
virtual_tn.session_id = session.id
db_session.add(session)
db_session.add(virtual_tn)
db_session.commit()
except IntegrityError:
db_session.rollback()
msg = "There were two sessions attempting to reserve the same virtual tn. Please retry."
log.error({"message": msg, "status": "failed"})
return Response(json.dumps({
"message": msg,
"status": "failed"
}),
content_type="application/json",
status=500)
expiry_date = session.expiry_date.strftime(
'%Y-%m-%d %H:%M:%S') if session.expiry_date else None
recipients = [participant_a, participant_b]
try:
send_message(recipients,
virtual_tn.value,
SESSION_START_MSG,
session.id,
is_system_msg=True)
except InternalSMSDispatcherError as e:
db_session.delete(session)
virtual_tn.session_id = None
db_session.add(virtual_tn)
db_session.commit()
raise e
msg = "ProxySession {} started with participants {} and {}".format(
session.id, participant_a, participant_b)
log.info({"message": msg, "status": "succeeded"})
return Response(json.dumps({
"message": "Created new session",
"status": "succeeded",
"session_id": session.id,
"expiry_date": expiry_date,
"virtual_tn": virtual_tn.value,
"participant_a": participant_a,
"participant_b": participant_b
}),
content_type="application/json")
def add_proxy_session():
"""
The ProxySession resource endpoint for adding a new ProxySession
to the pool.
"""
body = request.json
try:
participant_a = body["participant_a"]
participant_b = body["participant_b"]
assert len(participant_a) <= 18
assert len(participant_b) <= 18
except (AssertionError, KeyError):
raise InvalidAPIUsage(
("Required argument: 'participant_a' (str, length <= 18)" ", 'participant_b' (str, length <= 18)"),
payload={"reason": "invalidAPIUsage"},
)
if "expiry_window" in body:
expiry_window = body["expiry_window"]
else:
expiry_window = None
# Release any VirtualTNs from expired ProxySessions back to the pool
ProxySession.clean_expired()
virtual_tn = VirtualTN.get_next_available()
if virtual_tn is None:
msg = "Could not create a new session -- No virtual TNs available."
log.critical({"message": msg, "status": "failed"})
return Response(json.dumps({"message": msg, "status": "failed"}), content_type="application/json", status=400)
else:
session = ProxySession(virtual_tn.value, participant_a, participant_b, expiry_window)
try:
virtual_tn.session_id = session.id
db_session.add(session)
db_session.add(virtual_tn)
db_session.commit()
except IntegrityError:
db_session.rollback()
msg = "There were two sessions attempting to reserve the same virtual tn. Please retry."
log.error({"message": msg, "status": "failed"})
return Response(
json.dumps({"message": msg, "status": "failed"}), content_type="application/json", status=500
)
expiry_date = session.expiry_date.strftime("%Y-%m-%d %H:%M:%S") if session.expiry_date else None
recipients = [participant_a, participant_b]
try:
send_message(recipients, virtual_tn.value, SESSION_START_MSG, session.id, is_system_msg=True)
except InternalSMSDispatcherError as e:
db_session.delete(session)
virtual_tn.session_id = None
db_session.add(virtual_tn)
db_session.commit()
raise e
msg = "ProxySession {} started with participants {} and {}".format(session.id, participant_a, participant_b)
log.info({"message": msg, "status": "succeeded"})
return Response(
json.dumps(
{
"message": "Created new session",
"status": "succeeded",
"session_id": session.id,
"expiry_date": expiry_date,
"virtual_tn": virtual_tn.value,
"participant_a": participant_a,
"participant_b": participant_b,
}
),
content_type="application/json",
)
