From 1c58a7347e92b9a5378bed44ecf357eafb83b94f Mon Sep 17 00:00:00 2001 From: Dan Puttick Date: Mon, 6 Mar 2017 14:55:33 -0500 Subject: [PATCH] If no extentions FileBase.ext is now None --- kittengroomer/helpers.py | 5 ++++- tests/test_kittengroomer.py | 1 - 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/kittengroomer/helpers.py b/kittengroomer/helpers.py index 9ba2b5e..3e270dc 100644 --- a/kittengroomer/helpers.py +++ b/kittengroomer/helpers.py @@ -53,7 +53,10 @@ class FileBase(object): def _determine_extension(self): _, ext = os.path.splitext(self.src_path) - return ext.lower() + ext = ext.lower() + if ext == '': + ext = None + return ext def _determine_mimetype(self): if os.path.islink(self.src_path): diff --git a/tests/test_kittengroomer.py b/tests/test_kittengroomer.py index 8339e84..bb5ab5d 100644 --- a/tests/test_kittengroomer.py +++ b/tests/test_kittengroomer.py @@ -130,7 +130,6 @@ class TestFileBase: def test_has_extension(self, temp_file, temp_file_no_ext): assert temp_file.has_extension() is True assert temp_file_no_ext.has_extension() is False - assert temp_file_no_ext.log_details.get('no_extension') is True def test_add_log_details(self, generic_conf_file): generic_conf_file.add_log_details('test', True)