From 7705848714e2eab3b15abec9ae6bd9f8f724bd05 Mon Sep 17 00:00:00 2001 From: Ethan Dalool Date: Fri, 11 Sep 2020 15:02:33 -0700 Subject: [PATCH] Let new_album take multiple associated directories. --- etiquette/photodb.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/etiquette/photodb.py b/etiquette/photodb.py index 31a8904..b77ae97 100644 --- a/etiquette/photodb.py +++ b/etiquette/photodb.py @@ -79,7 +79,7 @@ class PDBAlbumMixin: title=None, description=None, *, - associated_directory=None, + associated_directories=None, author=None, photos=None, ): @@ -106,8 +106,10 @@ class PDBAlbumMixin: album = self.get_cached_instance('album', data) - if associated_directory is not None: - album.add_associated_directory(associated_directory) + associated_directories = associated_directories or () + if isinstance(associated_directories, str): + associated_directories = [associated_directories] + album.add_associated_directories(associated_directories) if photos is not None: photos = [self.get_photo(photo) for photo in photos]