def post(self):
post_data = request.get_json() or request.form
session_id = post_data.get('sessionId')
phone_number = post_data.get('phoneNumber')
user_input = post_data.get('text')
service_code = post_data.get('serviceCode')
if phone_number:
user = get_user_by_phone(phone_number)
# api chains all inputs that came through with *
latest_input = user_input.split('*')[-1]
if None in [user, session_id]:
current_menu = UssdMenu.find_by_name('exit_not_registered')
text = menu_display_text_in_lang(current_menu, user)
else:
current_menu = UssdProcessor.process_request(session_id, latest_input, user)
ussd_session = create_or_update_session(session_id, user, current_menu, user_input, service_code)
text = UssdProcessor.custom_display_text(current_menu, ussd_session)
if "CON" not in text and "END" not in text:
raise Exception("no menu found. text={}, user={}, menu={}, session={}".format(text, user.id, current_menu.name, ussd_session.id))
if len(text) > 164:
print(f"Warning, text has length {len(text)}, display may be truncated")
db.session.commit()
else:
current_menu = UssdMenu.find_by_name('exit_invalid_request')
text = menu_display_text_in_lang(current_menu, None)
return make_response(text), 200
def test_create_or_update_session(test_client, init_database):
from flask import g
g.active_organisation = OrganisationFactory(country_code='AU')
user = UserFactory(phone="123")
# create a session in db
menu3 = UssdMenu(id=3, name='foo', display_key='foo')
db.session.add(menu3)
session = UssdSession(session_id="1",
user_id=user.id,
msisdn="123",
ussd_menu=menu3,
state="foo",
service_code="*123#")
db.session.add(session)
db.session.commit()
# test updating existing
create_or_update_session("1", user,
UssdMenu(id=4, name="bar", display_key='bar'),
"input", "*123#")
sessions = UssdSession.query.filter_by(session_id="1")
assert sessions.count() == 1
session = sessions.first()
assert session.state == "bar"
assert session.user_input == "input"
assert session.ussd_menu_id == 4
# test creating a new one
sessions = UssdSession.query.filter_by(session_id="2")
assert sessions.count() == 0
create_or_update_session("2", user,
UssdMenu(id=5, name="bat", display_key='bat'), "",
"*123#")
sessions = UssdSession.query.filter_by(session_id="2")
assert sessions.count() == 1
session = sessions.first()
assert session.state == "bat"
assert session.user_input == ""
assert session.ussd_menu_id == 5
def post(self):
post_data = request.get_json() or request.form
session_id = post_data.get('sessionId')
phone_number = post_data.get('phoneNumber')
user_input = post_data.get('text')
service_code = post_data.get('serviceCode')
# enforce only one single service code that can access the ussd state machine
# through the endpoint
if config.USSD_VALID_SERVICE_CODE != service_code:
response = 'END '
response += i18n.t(
'ussd.kenya.invalid_service_code',
valid_service_code=config.USSD_VALID_SERVICE_CODE,
locale='sw')
response += "\n"
response += i18n.t(
'ussd.kenya.invalid_service_code',
valid_service_code=config.USSD_VALID_SERVICE_CODE,
locale='en')
return make_response(response, 200)
elif phone_number:
user = get_user_by_phone(phone_number, 'KE')
# api chains all inputs that came through with *
latest_input = user_input.split('*')[-1]
if None in [user, session_id]:
user_without_transfer_account = create_user_without_transfer_account(
phone_number)
current_menu = UssdMenu.find_by_name(
'initial_language_selection')
ussd_session = create_or_update_session(
session_id=session_id,
user=user_without_transfer_account,
user_input=latest_input,
service_code=service_code,
current_menu=current_menu)
text = KenyaUssdProcessor.custom_display_text(
current_menu, ussd_session)
else:
current_menu = KenyaUssdProcessor.process_request(
session_id, latest_input, user)
ussd_session = create_or_update_session(
session_id, user, current_menu, user_input, service_code)
text = KenyaUssdProcessor.custom_display_text(
current_menu, ussd_session)
if "CON" not in text and "END" not in text:
raise Exception(
"no menu found. text={}, user={}, menu={}, session={}".
format(text, user.id, current_menu.name,
ussd_session.id))
if len(text) > 164:
print(
f"Warning, text has length {len(text)}, display may be truncated"
)
db.session.commit()
else:
current_menu = UssdMenu.find_by_name('exit_invalid_request')
text = menu_display_text_in_lang(current_menu, None)
return make_response(text), 200