darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-01 15:46:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

exclude non-pdfs from collate task

Browse Source 
and fix celery kwargsg
This commit is contained in:
Leo Hemsted
2017-12-22 15:38:49 +00:00
committed by Ken Tsang
parent 309b4d7d33
commit 9debc96c8c
2 changed files with 27 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
app/celery/letters_pdf_tasks.py
View File

@@ -97,7 +97,7 @@ def collate_letter_pdfs_for_day(date):
)
notify_celery.send_task(
name=TaskNames.ZIP_AND_SEND_LETTER_PDFS,
kwargs={'filenames': filenames},
kwargs={'filenames_to_zip': filenames},
queue=QueueNames.PROCESS_FTP
)
@@ -111,13 +111,14 @@ def group_letters(letter_pdfs):
running_filesize = 0
list_of_files = []
for letter in letter_pdfs:
if running_filesize + letter['Size'] > current_app.config['MAX_LETTER_PDF_ZIP_FILESIZE']:
yield list_of_files
running_filesize = 0
list_of_files = []
if letter['Key'].lower().endswith('.pdf'):
if running_filesize + letter['Size'] > current_app.config['MAX_LETTER_PDF_ZIP_FILESIZE']:
yield list_of_files
running_filesize = 0
list_of_files = []
running_filesize += letter['Size']
list_of_files.append(letter)
running_filesize += letter['Size']
list_of_files.append(letter)
if list_of_files:
yield list_of_files