mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
106 lines
2.9 KiB
Python
106 lines
2.9 KiB
Python
from sqlalchemy import desc
|
|
|
|
from app import db
|
|
from app.dao.dao_utils import autocommit
|
|
from app.models import ServiceLetterContact, Template
|
|
|
|
|
|
def dao_get_letter_contacts_by_service_id(service_id):
|
|
letter_contacts = db.session.query(
|
|
ServiceLetterContact
|
|
).filter(
|
|
ServiceLetterContact.service_id == service_id,
|
|
ServiceLetterContact.archived == False # noqa
|
|
).order_by(
|
|
desc(ServiceLetterContact.is_default),
|
|
desc(ServiceLetterContact.created_at)
|
|
).all()
|
|
|
|
return letter_contacts
|
|
|
|
|
|
def dao_get_letter_contact_by_id(service_id, letter_contact_id):
|
|
letter_contact = db.session.query(
|
|
ServiceLetterContact
|
|
).filter(
|
|
ServiceLetterContact.service_id == service_id,
|
|
ServiceLetterContact.id == letter_contact_id,
|
|
ServiceLetterContact.archived == False # noqa
|
|
).one()
|
|
return letter_contact
|
|
|
|
|
|
@autocommit
|
|
def add_letter_contact_for_service(service_id, contact_block, is_default):
|
|
old_default = _get_existing_default(service_id)
|
|
if is_default:
|
|
_reset_old_default_to_false(old_default)
|
|
|
|
new_letter_contact = ServiceLetterContact(
|
|
service_id=service_id,
|
|
contact_block=contact_block,
|
|
is_default=is_default
|
|
)
|
|
db.session.add(new_letter_contact)
|
|
return new_letter_contact
|
|
|
|
|
|
@autocommit
|
|
def update_letter_contact(service_id, letter_contact_id, contact_block, is_default):
|
|
old_default = _get_existing_default(service_id)
|
|
# if we want to make this the default, ensure there are no other existing defaults
|
|
if is_default:
|
|
_reset_old_default_to_false(old_default)
|
|
|
|
letter_contact_update = ServiceLetterContact.query.get(letter_contact_id)
|
|
letter_contact_update.contact_block = contact_block
|
|
letter_contact_update.is_default = is_default
|
|
db.session.add(letter_contact_update)
|
|
return letter_contact_update
|
|
|
|
|
|
@autocommit
|
|
def archive_letter_contact(service_id, letter_contact_id):
|
|
letter_contact_to_archive = ServiceLetterContact.query.filter_by(
|
|
id=letter_contact_id,
|
|
service_id=service_id
|
|
).one()
|
|
|
|
Template.query.filter_by(
|
|
service_letter_contact_id=letter_contact_id
|
|
).update({
|
|
'service_letter_contact_id': None
|
|
})
|
|
|
|
letter_contact_to_archive.archived = True
|
|
|
|
db.session.add(letter_contact_to_archive)
|
|
return letter_contact_to_archive
|
|
|
|
|
|
def _get_existing_default(service_id):
|
|
old_defaults = [
|
|
x for x
|
|
in dao_get_letter_contacts_by_service_id(service_id=service_id)
|
|
if x.is_default
|
|
]
|
|
|
|
if len(old_defaults) == 0:
|
|
return None
|
|
|
|
if len(old_defaults) == 1:
|
|
return old_defaults[0]
|
|
|
|
raise Exception(
|
|
"There should only be one default letter contact for each service. Service {} has {}".format(
|
|
service_id,
|
|
len(old_defaults)
|
|
)
|
|
)
|
|
|
|
|
|
def _reset_old_default_to_false(old_default):
|
|
if old_default:
|
|
old_default.is_default = False
|
|
db.session.add(old_default)
|