summaryrefslogtreecommitdiff
path: root/fbin/monkey.py
diff options
context:
space:
mode:
authorJon Bergli Heier <snakebite@jvnv.net>2021-06-09 19:19:56 +0200
committerJon Bergli Heier <snakebite@jvnv.net>2021-06-09 19:21:34 +0200
commite96bedf7477d392b8821f76ca85038c198c84375 (patch)
treec56deba384f45da8958d848113b4da9fa73fe3f9 /fbin/monkey.py
parent7c13b1038482d68c2ab581dad16fa81c0a09034e (diff)
Fix linting errors
Style, unused imports, unused variables, etc. as reported by flake8. Configuration for flake8 has been added to setup.cfg.
Diffstat (limited to 'fbin/monkey.py')
-rw-r--r--fbin/monkey.py6
1 files changed, 5 insertions, 1 deletions
diff --git a/fbin/monkey.py b/fbin/monkey.py
index c6458ad..89df4f3 100644
--- a/fbin/monkey.py
+++ b/fbin/monkey.py
@@ -4,6 +4,7 @@ from flask import current_app
import werkzeug.formparser
import werkzeug.wrappers
+
def werkzeug_patch():
global werkzeug_orig_stream_factory
@@ -11,15 +12,18 @@ def werkzeug_patch():
def custom_stream_factory(total_content_length, filename, content_type, content_length=None):
if total_content_length > 1024 * 500:
- return tempfile.NamedTemporaryFile('wb+', prefix = 'upload_', dir = current_app.config['FILE_DIRECTORY'], delete = True)
+ return tempfile.NamedTemporaryFile('wb+', prefix='upload_',
+ dir=current_app.config['FILE_DIRECTORY'], delete=True)
return werkzeug_orig_stream_factory(total_content_length, filename, content_type, content_length)
werkzeug.formparser.default_stream_factory = custom_stream_factory
werkzeug.wrappers.default_stream_factory = custom_stream_factory
+
def werkzeug_reset():
werkzeug.formparser.default_stream_factory = werkzeug_orig_stream_factory
werkzeug.wrappers.default_stream_factory = werkzeug_orig_stream_factory
+
def patch():
werkzeug_patch()