Make sure the file to write with GridFS has a "content_type" attribute

noirbizarre · Apr 11, 2018 · a5e80a1 · a5e80a1
1 parent a88e603
commit a5e80a1
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/flask_fs/backends/gridfs.py b/flask_fs/backends/gridfs.py
@@ -54,10 +54,16 @@ def read(self, filename):
         return f.read()
 
     def write(self, filename, content):
+        kwargs = {
+            'filename': filename
+        }
+
+        if hasattr(content, 'content_type'):
+            kwargs['content_type'] = content.content_type
+
         return self.fs.put(
             self.as_binary(content),
-            filename=filename,
-            content_type=content.content_type
+            **kwargs
         )
 
     def delete(self, filename):