mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 23:41:17 -05:00
93 lines
3.7 KiB
Python
93 lines
3.7 KiB
Python
from datetime import datetime
|
|
from flask import current_app
|
|
from sqlalchemy import and_, desc, func, literal, String
|
|
|
|
from app import db
|
|
from app.models import (
|
|
Job, Notification, Template, LETTER_TYPE, JOB_STATUS_CANCELLED, JOB_STATUS_SCHEDULED,
|
|
NOTIFICATION_CANCELLED, ServiceDataRetention
|
|
)
|
|
from app.utils import midnight_n_days_ago
|
|
|
|
|
|
def dao_get_uploads_by_service_id(service_id, limit_days=None, page=1, page_size=50):
|
|
# Hardcoded filter to exclude cancelled or scheduled jobs
|
|
# for the moment, but we may want to change this method take 'statuses' as a argument in the future
|
|
today = datetime.utcnow().date()
|
|
jobs_query_filter = [
|
|
Job.service_id == service_id,
|
|
Job.original_file_name != current_app.config['TEST_MESSAGE_FILENAME'],
|
|
Job.original_file_name != current_app.config['ONE_OFF_MESSAGE_FILENAME'],
|
|
Job.job_status.notin_([JOB_STATUS_CANCELLED, JOB_STATUS_SCHEDULED]),
|
|
func.coalesce(
|
|
Job.processing_started, Job.created_at
|
|
) >= today - func.coalesce(ServiceDataRetention.days_of_retention, 7)
|
|
]
|
|
if limit_days is not None:
|
|
jobs_query_filter.append(Job.created_at >= midnight_n_days_ago(limit_days))
|
|
|
|
jobs_query = db.session.query(
|
|
Job.id,
|
|
Job.original_file_name,
|
|
Job.notification_count,
|
|
Template.template_type,
|
|
func.coalesce(ServiceDataRetention.days_of_retention, 7).label('days_of_retention'),
|
|
Job.created_at.label("created_at"),
|
|
Job.scheduled_for.label("scheduled_for"),
|
|
Job.processing_started.label('processing_started'),
|
|
Job.job_status.label("status"),
|
|
literal('job').label('upload_type'),
|
|
literal(None).label('recipient'),
|
|
).join(
|
|
Template, Job.template_id == Template.id
|
|
).outerjoin(
|
|
ServiceDataRetention, and_(
|
|
Template.service_id == ServiceDataRetention.service_id,
|
|
func.cast(Template.template_type, String) == func.cast(ServiceDataRetention.notification_type, String)
|
|
)
|
|
).filter(
|
|
*jobs_query_filter
|
|
)
|
|
|
|
letters_query_filter = [
|
|
Notification.service_id == service_id,
|
|
Notification.notification_type == LETTER_TYPE,
|
|
Notification.api_key_id == None, # noqa
|
|
Notification.status != NOTIFICATION_CANCELLED,
|
|
Template.hidden == True,
|
|
Notification.created_at >= today - func.coalesce(ServiceDataRetention.days_of_retention, 7)
|
|
|
|
]
|
|
if limit_days is not None:
|
|
letters_query_filter.append(Notification.created_at >= midnight_n_days_ago(limit_days))
|
|
|
|
letters_query = db.session.query(
|
|
Notification.id,
|
|
Notification.client_reference.label('original_file_name'),
|
|
literal('1').label('notification_count'),
|
|
literal(None).label('template_type'),
|
|
func.coalesce(ServiceDataRetention.days_of_retention, 7).label('days_of_retention'),
|
|
Notification.created_at.label("created_at"),
|
|
literal(None).label('scheduled_for'),
|
|
# letters don't have a processing_started date but we want created_at to be used for sorting
|
|
Notification.created_at.label('processing_started'),
|
|
Notification.status,
|
|
literal('letter').label('upload_type'),
|
|
Notification.to.label('recipient'),
|
|
).join(
|
|
Template, Notification.template_id == Template.id
|
|
).outerjoin(
|
|
ServiceDataRetention, and_(
|
|
Template.service_id == ServiceDataRetention.service_id,
|
|
func.cast(Template.template_type, String) == func.cast(ServiceDataRetention.notification_type, String)
|
|
)
|
|
).filter(
|
|
*letters_query_filter
|
|
)
|
|
|
|
return jobs_query.union_all(
|
|
letters_query
|
|
).order_by(
|
|
desc("processing_started"), desc("created_at")
|
|
).paginate(page=page, per_page=page_size)
|