darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-08-24 16:23:44 -04:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1797 from alphagov/ken-test-key-antivirus

Browse Source 
Add antivirus check on precompiled letters sent with test key
This commit is contained in:
kentsanggds
2018-03-23 15:04:08 +00:00
committed by GitHub
parent d14ac2aa81 1a9bc2a5cf
commit 5d38989cd9
6 changed files with 82 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
app/v2/notifications/post_notifications.py
View File

@@ -227,7 +227,7 @@ def process_letter_notification(*, letter_data, api_key, template, reply_to_text
if precompiled:
try:
if should_send:
if should_send or (precompiled and api_key.key_type == KEY_TYPE_TEST):
status = NOTIFICATION_PENDING_VIRUS_CHECK
letter_content = base64.b64decode(letter_data['content'])
pages = pdf_page_count(io.BytesIO(letter_content))
@@ -273,8 +273,16 @@ def process_letter_notification(*, letter_data, api_key, template, reply_to_text
)
else:
if precompiled and api_key.key_type == KEY_TYPE_TEST:
upload_letter_pdf(notification, letter_content, is_test_letter=True)
update_notification_status_by_reference(notification.reference, NOTIFICATION_DELIVERED)
filename = upload_letter_pdf(notification, letter_content)
# call task to add the filename to anti virus queue
notify_celery.send_task(
name=TaskNames.SCAN_FILE,
kwargs={'filename': filename},
queue=QueueNames.ANTIVIRUS,
)
else:
update_notification_status_by_reference(notification.reference, NOTIFICATION_DELIVERED)
return notification