voussoir/etiquette
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add parameter albums to purge_empty_albums as starting set.

Browse source 
Like purge_deleted_files, you can provide a list of albums to
act as the starting point instead of letting it use the default
of all albums.
This commit is contained in:
voussoir 2018-07-22 20:09:16 -07:00
parent c840845c7d
commit 2a45f4e17e
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
etiquette/photodb.py
View file

@ -298,15 +298,20 @@ class PDBPhotoMixin:
self.commit() self.commit()
@decorators.transaction @decorators.transaction
def purge_empty_albums(self, *, commit=True): def purge_empty_albums(self, albums=None, *, commit=True):
to_check = list(self.get_albums()) if albums is None:
to_check = set(self.get_albums())
else:
to_check = set()
for album in albums:
to_check.update(album.walk_children())
while to_check: while to_check:
album = to_check.pop() album = to_check.pop()
if album.get_children() or album.get_photos(): if album.get_children() or album.get_photos():
continue continue
# This may have been the last child of an otherwise empty parent. # This may have been the last child of an otherwise empty parent.
to_check.extend(album.get_parents()) to_check.update(album.get_parents())
album.delete(commit=False) album.delete(commit=False)
if commit: if commit: