Fixed storage for correct generated migrations

GitRon · Mar 29, 2018 · bdc9624 · bdc9624
1 parent 141c47b
commit bdc9624
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/chunked_upload/settings.py b/chunked_upload/settings.py
@@ -41,9 +41,8 @@ def default_upload_to(instance, filename):
     STORAGE = getattr(settings, 'CHUNKED_UPLOAD_STORAGE_CLASS', lambda: None)()
 # Use temporary storage for chunks
 else:
-    import tempfile
-    from django.core.files.storage import FileSystemStorage
-    STORAGE = FileSystemStorage(location=tempfile.gettempdir)
+    from chunked_upload.storages import TemporaryFileStorage
+    STORAGE = TemporaryFileStorage()
 
 
 # Boolean that defines if the ChunkedUpload model is abstract or not

diff --git a/chunked_upload/storages.py b/chunked_upload/storages.py
@@ -0,0 +1,7 @@
+import tempfile
+
+from django.core.files.storage import FileSystemStorage
+
+
+class TemporaryFileStorage(FileSystemStorage):
+    location = tempfile.gettempdir()