Fix crash when saving an image (#10)

springload · Apr 23, 2018 · 6e7af5b · 6e7af5b
1 parent 13c1bdc
commit 6e7af5b
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/wagtailenforcer/utilities.py b/wagtailenforcer/utilities.py
@@ -9,7 +9,13 @@
 
 
 def check_for_virus(instance):
-    has_virus, name = is_infected(instance.file.read())
+    if instance.file.closed:
+        with open(instance.file.path, 'rb') as file:
+            file_content = file.read()
+    else:
+        file_content = file.read()
+
+    has_virus, name = is_infected(file_content)
 
     if has_virus:
         raise EnforcerVirusException(_('Virus "{}" was detected').format(name))