From 11291a582ebef7fe95ac46d5ed625e9b5efcda28 Mon Sep 17 00:00:00 2001 From: Cedric Leporcq Date: Fri, 12 Nov 2021 09:17:38 +0100 Subject: [PATCH] Remove confusing unused argument from get_date_regex --- ordigi/collection.py | 2 +- ordigi/utils.py | 15 ++++++--------- tests/test_collection.py | 2 +- 3 files changed, 8 insertions(+), 11 deletions(-) diff --git a/ordigi/collection.py b/ordigi/collection.py index c2fb28d..acbe503 100644 --- a/ordigi/collection.py +++ b/ordigi/collection.py @@ -135,7 +135,7 @@ class FPath: elif item == 'name': # Remove date prefix added to the name. part = stem - for regex in utils.get_date_regex(stem).values(): + for regex in utils.get_date_regex().values(): part = re.sub(regex, '', part) elif item == 'date': date = metadata['date_media'] diff --git a/ordigi/utils.py b/ordigi/utils.py index b4edbd7..6ca3bf7 100644 --- a/ordigi/utils.py +++ b/ordigi/utils.py @@ -37,7 +37,8 @@ def distance_between_two_points(lat1, lon1, lat2, lon2): return r * sqrt(x * x + y * y) -def get_date_regex(string, user_regex=None): +def get_date_regex(user_regex=None): + """Return date regex generator""" if user_regex: regex = {'a': re.compile(user_regex)} else: @@ -63,13 +64,14 @@ def get_date_regex(string, user_regex=None): def get_date_from_string(string, user_regex=None): + """Retrieve date stamp from string""" # If missing datetime from EXIF data check if filename is in datetime format. # For this use a user provided regex if possible. # Otherwise assume a filename such as IMG_20160915_123456.jpg as default. matches = [] - for i, rx in get_date_regex(string, user_regex).items(): - match = re.findall(rx, string) + for i, regex in get_date_regex(user_regex).items(): + match = re.findall(regex, string) if match != []: if i == 'c': match = [('20' + match[0][0], match[0][1], match[0][2])] @@ -80,7 +82,7 @@ def get_date_from_string(string, user_regex=None): if len(match) != 1: # The time string is not uniq continue - matches.append((match[0], rx)) + matches.append((match[0], regex)) # We want only the first match for the moment break @@ -89,11 +91,6 @@ def get_date_from_string(string, user_regex=None): try: # Convert str to int date_object = tuple(map(int, matches[0][0])) - - time = False - if len(date_object) > 3: - time = True - date = datetime(*date_object) except (KeyError, ValueError): return None diff --git a/tests/test_collection.py b/tests/test_collection.py index c9b6493..5421b3d 100644 --- a/tests/test_collection.py +++ b/tests/test_collection.py @@ -86,7 +86,7 @@ class TestFPath: assert part == file_path.suffix[1:], file_path elif item == 'name': expected_part = file_path.stem - for rx in utils.get_date_regex(expected_part).values(): + for rx in utils.get_date_regex().values(): part = re.sub(rx, '', expected_part) assert part == expected_part, file_path elif item == 'custom':