darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2025-12-20 15:31:15 -05:00
Code Issues Packages Projects Releases Wiki Activity

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34100 additions and 23589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

50
app/dao/service_data_retention_dao.py
View File

@@ -6,51 +6,57 @@ from app.models import ServiceDataRetention
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()
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()
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
def fetch_service_data_retention_by_notification_type(service_id, notification_type):
data_retention_list = ServiceDataRetention.query.filter_by(
service_id=service_id,
notification_type=notification_type
service_id=service_id, notification_type=notification_type
).first()
return data_retention_list
@autocommit
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)
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
@autocommit
def update_service_data_retention(service_data_retention_id, service_id, days_of_retention):
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()
}
)
ServiceDataRetention.service_id == service_id,
).update({"days_of_retention": days_of_retention, "updated_at": datetime.utcnow()})
return updated_count
def fetch_service_data_retention_for_all_services_by_notification_type(notification_type):
return ServiceDataRetention.query.filter(ServiceDataRetention.notification_type == notification_type).all()
def fetch_service_data_retention_for_all_services_by_notification_type(
notification_type,
):
return ServiceDataRetention.query.filter(
ServiceDataRetention.notification_type == notification_type
).all()