darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-19 14:09:20 -04:00
Code Issues Packages Projects Releases Wiki Activity

Add max file size check when uploading a letter

Browse Source
This commit is contained in:
Katie Smith
2019-09-06 11:13:05 +01:00
parent be6b5b922a
commit a103dbf801
2 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
app/main/views/uploads.py
View File

@@ -9,6 +9,8 @@ from app.main import main
from app.main.forms import PDFUploadForm
from app.utils import user_has_permissions
MAX_FILE_UPLOAD_SIZE = 2 * 1024 * 1024 # 2MB
@main.route("/services/<service_id>/uploads")
@user_has_permissions('send_messages')
@@ -29,6 +31,10 @@ def upload_letter(service_id):
flash('Your file has failed the virus check', 'dangerous')
return render_template('views/uploads/choose-file.html', form=form), 400
if len(pdf_file_bytes) > MAX_FILE_UPLOAD_SIZE:
flash('Your file must be smaller than 2MB', 'dangerous')
return render_template('views/uploads/choose-file.html', form=form), 400
upload_id = uuid.uuid4()
return redirect(