mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-03 01:41:05 -05:00
Add data models, dao for inbound_numbers
This commit is contained in:
28
app/dao/inbound_numbers_dao.py
Normal file
28
app/dao/inbound_numbers_dao.py
Normal file
@@ -0,0 +1,28 @@
|
||||
from app import db
|
||||
from app.dao.dao_utils import transactional
|
||||
from app.models import InboundNumber
|
||||
|
||||
|
||||
def dao_get_inbound_numbers():
|
||||
return InboundNumber.query.all()
|
||||
|
||||
|
||||
def dao_get_available_inbound_numbers():
|
||||
return InboundNumber.query.filter(InboundNumber.active, InboundNumber.service_id.is_(None)).all()
|
||||
|
||||
|
||||
def dao_get_inbound_number_for_service(service_id):
|
||||
return InboundNumber.query.filter(InboundNumber.service_id == service_id).all()
|
||||
|
||||
|
||||
@transactional
|
||||
def dao_allocate_inbound_number_to_service(service_id):
|
||||
available_numbers = InboundNumber.query.filter(
|
||||
InboundNumber.active, InboundNumber.service_id.is_(None)).all()
|
||||
|
||||
if len(available_numbers) > 0:
|
||||
available_numbers[0].service_id = service_id
|
||||
|
||||
db.session.add(available_numbers[0])
|
||||
else:
|
||||
raise IndexError('No inbound numbers available')
|
||||
Reference in New Issue
Block a user