Add virus scan stage when uploading a letter

This commit is contained in:
Katie Smith
2019-09-06 11:04:35 +01:00
parent c577416866
commit be6b5b922a
2 changed files with 27 additions and 1 deletions

View File

@@ -1,8 +1,10 @@
import uuid
from io import BytesIO
from flask import redirect, render_template, request, url_for
from flask import flash, redirect, render_template, request, url_for
from app import current_service
from app.extensions import antivirus_client
from app.main import main
from app.main.forms import PDFUploadForm
from app.utils import user_has_permissions
@@ -20,6 +22,13 @@ def upload_letter(service_id):
form = PDFUploadForm()
if form.validate_on_submit():
pdf_file_bytes = form.file.data.read()
virus_free = antivirus_client.scan(BytesIO(pdf_file_bytes))
if not virus_free:
flash('Your file has failed the virus check', 'dangerous')
return render_template('views/uploads/choose-file.html', form=form), 400
upload_id = uuid.uuid4()
return redirect(