mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-24 01:11:38 -05:00
a8adf4d7d7
When a service is allocated an inbound number and they only have one SMS sender, then update that SMS sender to the inbound number. That way they will not have more than one SMS sender and will not have to choose to use either one.
120 lines
3.9 KiB
Python
120 lines
3.9 KiB
Python
from sqlalchemy import desc
|
|
|
|
from app import db
|
|
from app.dao.dao_utils import transactional
|
|
from app.models import ServiceSmsSender
|
|
|
|
|
|
@transactional
|
|
def insert_or_update_service_sms_sender(service, sms_sender, inbound_number_id=None):
|
|
result = db.session.query(
|
|
ServiceSmsSender
|
|
).filter(
|
|
ServiceSmsSender.service_id == service.id
|
|
).update(
|
|
{'sms_sender': sms_sender,
|
|
'inbound_number_id': inbound_number_id
|
|
}
|
|
)
|
|
if result == 0:
|
|
new_sms_sender = ServiceSmsSender(sms_sender=sms_sender,
|
|
service=service,
|
|
is_default=True,
|
|
inbound_number_id=inbound_number_id
|
|
)
|
|
db.session.add(new_sms_sender)
|
|
|
|
|
|
def insert_service_sms_sender(service, sms_sender):
|
|
"""
|
|
This method is called from create_service which is wrapped in a transaction.
|
|
"""
|
|
new_sms_sender = ServiceSmsSender(sms_sender=sms_sender,
|
|
service=service,
|
|
is_default=True
|
|
)
|
|
db.session.add(new_sms_sender)
|
|
|
|
|
|
def dao_get_service_sms_senders_by_id(service_id, service_sms_sender_id):
|
|
return ServiceSmsSender.query.filter_by(
|
|
id=service_sms_sender_id,
|
|
service_id=service_id
|
|
).one()
|
|
|
|
|
|
def dao_get_sms_senders_by_service_id(service_id):
|
|
return ServiceSmsSender.query.filter_by(service_id=service_id).order_by(desc(ServiceSmsSender.is_default)).all()
|
|
|
|
|
|
@transactional
|
|
def dao_add_sms_sender_for_service(service_id, sms_sender, is_default, inbound_number_id=None):
|
|
old_default = _get_existing_default(service_id=service_id)
|
|
if is_default:
|
|
_reset_old_default_to_false(old_default)
|
|
else:
|
|
_raise_when_no_default(old_default)
|
|
|
|
new_sms_sender = ServiceSmsSender(
|
|
service_id=service_id,
|
|
sms_sender=sms_sender,
|
|
is_default=is_default,
|
|
inbound_number_id=inbound_number_id
|
|
)
|
|
|
|
db.session.add(new_sms_sender)
|
|
return new_sms_sender
|
|
|
|
|
|
@transactional
|
|
def dao_update_service_sms_sender(service_id, service_sms_sender_id, is_default, sms_sender=None):
|
|
old_default = _get_existing_default(service_id)
|
|
if is_default:
|
|
_reset_old_default_to_false(old_default)
|
|
else:
|
|
if old_default.id == service_sms_sender_id:
|
|
raise Exception("You must have at least one SMS sender as the default")
|
|
|
|
sms_sender_to_update = ServiceSmsSender.query.get(service_sms_sender_id)
|
|
sms_sender_to_update.is_default = is_default
|
|
if not sms_sender_to_update.inbound_number_id and sms_sender:
|
|
sms_sender_to_update.sms_sender = sms_sender
|
|
db.session.add(sms_sender_to_update)
|
|
return sms_sender_to_update
|
|
|
|
|
|
@transactional
|
|
def update_existing_sms_sender_with_inbound_number(service_sms_sender, sms_sender, inbound_number_id):
|
|
service_sms_sender.sms_sender = sms_sender
|
|
service_sms_sender.inbound_number_id = inbound_number_id
|
|
db.session.add(service_sms_sender)
|
|
return service_sms_sender
|
|
|
|
|
|
def _get_existing_default(service_id):
|
|
sms_senders = dao_get_sms_senders_by_service_id(service_id=service_id)
|
|
if sms_senders:
|
|
old_default = [x for x in sms_senders if x.is_default]
|
|
if len(old_default) == 1:
|
|
return old_default[0]
|
|
else:
|
|
raise Exception(
|
|
"There should only be one default sms sender for each service. Service {} has {}".format(
|
|
service_id,
|
|
len(old_default)
|
|
)
|
|
)
|
|
return None
|
|
|
|
|
|
def _reset_old_default_to_false(old_default):
|
|
if old_default:
|
|
old_default.is_default = False
|
|
db.session.add(old_default)
|
|
|
|
|
|
def _raise_when_no_default(old_default):
|
|
# check that the update is not updating the only default to false
|
|
if not old_default:
|
|
raise Exception("You must have at least one SMS sender as the default.", 400)
|