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)