mirror of
https://github.com/GSA/notifications-api.git
synced 2026-08-20 14:29:25 -04:00
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
from sqlalchemy import select
|
|
|
|
from app import db
|
|
from app.dao.dao_utils import autocommit
|
|
from app.models import InboundNumber
|
|
|
|
|
|
def dao_get_inbound_numbers():
|
|
stmt = select(InboundNumber).order_by(InboundNumber.updated_at)
|
|
return db.session.execute(stmt).scalars().all()
|
|
|
|
|
|
def dao_get_available_inbound_numbers():
|
|
stmt = select(InboundNumber).filter(
|
|
InboundNumber.active, InboundNumber.service_id.is_(None)
|
|
)
|
|
return db.session.execute(stmt).scalars().all()
|
|
|
|
|
|
def dao_get_inbound_number_for_service(service_id):
|
|
stmt = select(InboundNumber).filter(InboundNumber.service_id == service_id)
|
|
return db.session.execute(stmt).scalars().first()
|
|
|
|
|
|
def dao_get_inbound_number(inbound_number_id):
|
|
stmt = select(InboundNumber).filter(InboundNumber.id == inbound_number_id)
|
|
return db.session.execute(stmt).scalars().first()
|
|
|
|
|
|
@autocommit
|
|
def dao_set_inbound_number_to_service(service_id, inbound_number):
|
|
inbound_number.service_id = service_id
|
|
db.session.add(inbound_number)
|
|
|
|
|
|
@autocommit
|
|
def dao_set_inbound_number_active_flag(service_id, active):
|
|
stmt = select(InboundNumber).filter(
|
|
InboundNumber.service_id == service_id
|
|
)
|
|
inbound_number = db.session.execute(stmt).scalars().first()
|
|
inbound_number.active = active
|
|
|
|
db.session.add(inbound_number)
|
|
|
|
|
|
@autocommit
|
|
def dao_allocate_number_for_service(service_id, inbound_number_id):
|
|
updated = InboundNumber.query.filter_by(
|
|
id=inbound_number_id, active=True, service_id=None
|
|
).update({"service_id": service_id})
|
|
if not updated:
|
|
raise Exception("Inbound number: {} is not available".format(inbound_number_id))
|
|
return InboundNumber.query.get(inbound_number_id)
|