Merge pull request #1779 from takeflight/refactor/document-get-extension

Use os.path.splitext() instead of custom logic in Document.file_extension

wagtail/wagtaildocs/models.py

@@ -40,11 +40,7 @@ class Document(models.Model, TagSearchable):
 
     @property
     def file_extension(self):
-        parts = self.filename.split('.')
-        if len(parts) > 1:
-            return parts[-1]
-        else:
-            return ''
+        return os.path.splitext(self.filename)[1][1:]
 
     @property
     def url(self):