mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
from datetime import datetime
|
|
|
|
from app import db
|
|
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()
|
|
return data_retention
|
|
|
|
|
|
@transactional
|
|
def insert_service_data_retention(service_id, notification_type, days_of_retention):
|
|
new_data_retention = ServiceDataRetention(service_id=service_id,
|
|
notification_type=notification_type,
|
|
days_of_retention=days_of_retention)
|
|
|
|
db.session.add(new_data_retention)
|
|
return new_data_retention
|
|
|
|
|
|
@transactional
|
|
def update_service_data_retention(service_data_retention_id, service_id, days_of_retention):
|
|
updated_count = ServiceDataRetention.query.filter(
|
|
ServiceDataRetention.id == service_data_retention_id,
|
|
ServiceDataRetention.service_id == service_id
|
|
).update(
|
|
{
|
|
"days_of_retention": days_of_retention,
|
|
"updated_at": datetime.utcnow()
|
|
}
|
|
)
|
|
return updated_count
|