Refactor FileBase.extension

2017-07-17 10:10:21 -04:00 · 2017-07-17 10:10:21 -04:00 · 5aaeac6f16
parent fcbf2e6574
commit 5aaeac6f16
1 changed files with 3 additions and 3 deletions
--- a/kittengroomer/helpers.py
+++ b/kittengroomer/helpers.py
@ -49,10 +49,10 @@ class FileBase(object):
    @property
    def extension(self):
        _, ext = os.path.splitext(self.filename)
-        ext = ext.lower()
        if ext == '':
-            ext = None
-        return ext
+            return None
+        else:
+            return ext.lower()

    @property
    def maintype(self):