Merge a5e80a1 into c1ceb10

noirbizarre · Apr 11, 2018 · 32753d9 · 32753d9
2 parents c1ceb10 + a5e80a1
commit 32753d9
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/flask_fs/backends/gridfs.py b/flask_fs/backends/gridfs.py
@@ -54,7 +54,17 @@ def read(self, filename):
         return f.read()
 
     def write(self, filename, content):
-        return self.fs.put(self.as_binary(content), filename=filename)
+        kwargs = {
+            'filename': filename
+        }
+
+        if hasattr(content, 'content_type'):
+            kwargs['content_type'] = content.content_type
+
+        return self.fs.put(
+            self.as_binary(content),
+            **kwargs
+        )
 
     def delete(self, filename):
         regex = '^{0}'.format(re.escape(filename))