summaryrefslogtreecommitdiff
path: root/fbin/file_storage
diff options
context:
space:
mode:
authorJon Bergli Heier <snakebite@jvnv.net>2019-12-07 11:14:01 +0100
committerJon Bergli Heier <snakebite@jvnv.net>2019-12-07 11:14:01 +0100
commitf1cfaef5b0f358c94fe7f5e7c2eb5a6d261a85ae (patch)
tree5828d85f02868311d4e9d5f1842a1f6c9b80d5e9 /fbin/file_storage
parente52a152b16e51c329a281f144864d846bcacdde6 (diff)
Add max file size configuration
This allows configuring max file sizes for both registered and anonymous users. For registered users the USER_FILE_SIZE_LIMIT is used, and ANONYMOUS_FILE_SIZE_LIMIT for anonymous users. If the size is not specified or None, the limit is not enforced. Setting the limit to 0 effectively disables uploads.
Diffstat (limited to 'fbin/file_storage')
-rw-r--r--fbin/file_storage/base.py11
-rw-r--r--fbin/file_storage/exceptions.py2
-rw-r--r--fbin/file_storage/filesystem.py3
3 files changed, 14 insertions, 2 deletions
diff --git a/fbin/file_storage/base.py b/fbin/file_storage/base.py
index 9f09199..e2ca1a6 100644
--- a/fbin/file_storage/base.py
+++ b/fbin/file_storage/base.py
@@ -1,17 +1,26 @@
import datetime
from .. import db
+from .exceptions import *
class BaseStorage:
def __init__(self, app):
self.app = app
- def add_file(self, file_hash, filename, size, user=None, ip=None):
+ def verify_file(self, file):
+ user = file.user_id is not None
+ size_limit = self.app.config.get('USER_FILE_SIZE_LIMIT' if user else 'ANONYMOUS_FILE_SIZE_LIMIT')
+ if size_limit is not None and file.size > size_limit:
+ raise FileSizeError('The file size is too large (max {})'.format(db.File.pretty_size(size_limit)))
+
+ def add_file(self, file_hash, filename, size, user=None, ip=None, verify=True):
'''Adds the file to the database.
Call from store_file after the file is successfully stored.'''
with db.session_scope() as sess:
f = db.File(file_hash, filename, size, datetime.datetime.utcnow(), user.id if user else None, ip)
+ # Raises on invalid files
+ self.verify_file(f)
sess.add(f)
sess.commit()
sess.refresh(f)
diff --git a/fbin/file_storage/exceptions.py b/fbin/file_storage/exceptions.py
new file mode 100644
index 0000000..140221b
--- /dev/null
+++ b/fbin/file_storage/exceptions.py
@@ -0,0 +1,2 @@
+class StorageError(Exception): pass
+class FileSizeError(StorageError): pass
diff --git a/fbin/file_storage/filesystem.py b/fbin/file_storage/filesystem.py
index 1259002..07d29db 100644
--- a/fbin/file_storage/filesystem.py
+++ b/fbin/file_storage/filesystem.py
@@ -22,7 +22,8 @@ class Storage(BaseStorage):
size = os.path.getsize(temp_path)
try:
new_file = self.add_file(file_hash, uploaded_file.filename, size, user, ip)
- os.rename(temp_path, new_file.get_path())
+ if new_file:
+ os.rename(temp_path, new_file.get_path())
return new_file
except:
os.unlink(temp.name)