From ce98e89c65c9920a97c429e9c504df825d5ea544 Mon Sep 17 00:00:00 2001 From: Jaisen Mathai Date: Sun, 22 Jan 2017 23:54:28 -0800 Subject: [PATCH] Change source to path and check if media is None #107 (#192) --- elodie/tools/add_original_name.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/elodie/tools/add_original_name.py b/elodie/tools/add_original_name.py index 3e6ec07..6b7d50c 100644 --- a/elodie/tools/add_original_name.py +++ b/elodie/tools/add_original_name.py @@ -33,13 +33,17 @@ def main(argv): result.append((source, status)) else: - status = add_original_name(source, subclasses) + status = add_original_name(path, subclasses) result.append((path, status)) result.write() def add_original_name(source, subclasses): media = Media.get_class_by_file(source, subclasses) + if media is None: + print('{} is not a valid media object'.format(source)) + return + metadata = media.get_metadata() if metadata['original_name'] is not None: print('{} already has OriginalFileName...Skipping'.format(source))