mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-06-02 20:40:51 -04:00
58 lines
1.9 KiB
Python
58 lines
1.9 KiB
Python
import uuid
|
|
from io import BytesIO
|
|
|
|
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
|
|
|
|
MAX_FILE_UPLOAD_SIZE = 2 * 1024 * 1024 # 2MB
|
|
|
|
|
|
@main.route("/services/<service_id>/uploads")
|
|
@user_has_permissions('send_messages')
|
|
def uploads(service_id):
|
|
return render_template('views/uploads/index.html')
|
|
|
|
|
|
@main.route("/services/<service_id>/upload-letter", methods=['GET', 'POST'])
|
|
@user_has_permissions('send_messages')
|
|
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
|
|
|
|
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(
|
|
url_for(
|
|
'main.uploaded_letter_preview',
|
|
service_id=current_service.id,
|
|
file_id=upload_id,
|
|
original_filename=form.file.data.filename,
|
|
)
|
|
)
|
|
|
|
return render_template('views/uploads/choose-file.html', form=form)
|
|
|
|
|
|
@main.route("/services/<service_id>/preview-letter/<file_id>")
|
|
@user_has_permissions('send_messages')
|
|
def uploaded_letter_preview(service_id, file_id):
|
|
original_filename = request.args.get('original_filename')
|
|
|
|
return render_template('views/uploads/preview.html', original_filename=original_filename)
|