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

Merge pull request #4090 from alphagov/limit-csv-file-size-177535141

Browse Source 
Reject CSV / Spreadsheet files larger than 10Mb
This commit is contained in:
Ben Thorner
2021-12-07 17:00:05 +00:00
committed by GitHub
parent 1a4204fed1 0ce7f72b07
commit b04bb51971
2 changed files with 29 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
app/main/forms.py
View File

@@ -9,6 +9,7 @@ from flask_login import current_user
from flask_wtf import FlaskForm as Form
from flask_wtf.file import FileAllowed
from flask_wtf.file import FileField as FileField_wtf
from flask_wtf.file import FileSize
from notifications_utils.columns import Columns
from notifications_utils.countries.data import Postage
from notifications_utils.formatters import strip_all_whitespace
@@ -1512,8 +1513,14 @@ class ChangePasswordForm(StripWhitespaceForm):
class CsvUploadForm(StripWhitespaceForm):
file = FileField('Add recipients', validators=[DataRequired(
message='Please pick a file'), CsvFileValidator()])
file = FileField('Add recipients', validators=[
DataRequired(message='Please pick a file'),
CsvFileValidator(),
FileSize(
max_size=10e6, # 10Mb
message='File must be smaller than 10Mb'
)
])
class ChangeNameForm(StripWhitespaceForm):