Added new endpoints to get service data retention for a service.

This commit is contained in:
Rebecca Law
2018-07-13 15:18:27 +01:00
parent e2a1dfeb31
commit ab695d24d5
4 changed files with 138 additions and 4 deletions

View File

@@ -5,11 +5,21 @@ from app.dao.dao_utils import transactional
from app.models import ServiceDataRetention
def get_service_data_retention(service_id):
data_retention = ServiceDataRetention.query.filter_by(service_id=service_id).all()
def fetch_service_data_retention_by_id(service_id, data_retention_id):
data_retention = ServiceDataRetention.query.filter_by(service_id=service_id, id=data_retention_id).first()
return data_retention
def fetch_service_data_retention(service_id):
data_retention_list = ServiceDataRetention.query.filter_by(
service_id=service_id
).order_by(
# in the order that models.notification_types are created (email, sms, letter)
ServiceDataRetention.notification_type
).all()
return data_retention_list
@transactional
def insert_service_data_retention(service_id, notification_type, days_of_retention):
new_data_retention = ServiceDataRetention(service_id=service_id,