From a127a6e871170bd33cc5f089babff47009e22207 Mon Sep 17 00:00:00 2001 From: Ken Tsang Date: Fri, 4 Aug 2017 19:09:05 +0100 Subject: [PATCH] Added active flag setting in inbound_number_dao --- app/dao/inbound_numbers_dao.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/app/dao/inbound_numbers_dao.py b/app/dao/inbound_numbers_dao.py index e9cdeaed2..f7b28e089 100644 --- a/app/dao/inbound_numbers_dao.py +++ b/app/dao/inbound_numbers_dao.py @@ -12,7 +12,7 @@ def dao_get_available_inbound_numbers(): def dao_get_inbound_number_for_service(service_id): - return InboundNumber.query.filter(InboundNumber.service_id == service_id).all() + return InboundNumber.query.filter(InboundNumber.service_id == service_id).first() @transactional @@ -20,3 +20,11 @@ def dao_set_inbound_number_to_service(service_id, inbound_number): inbound_number.service_id = service_id db.session.add(inbound_number) + + +@transactional +def dao_set_inbound_number_active_flag_for_service(service_id, active): + inbound_number = dao_get_inbound_number_for_service(service_id) + inbound_number.active = active + + db.session.add(inbound_number)