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

Replace get_album_by_path with get_albums_by_path generator.

Browse source
This commit is contained in:
voussoir 2020-09-11 15:15:12 -07:00
parent 7705848714
commit 1cc70616a5
1 changed files with 14 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
etiquette/photodb.py
View file

@ -46,26 +46,26 @@ class PDBAlbumMixin:
def get_album_by_id(self, id): def get_album_by_id(self, id):
return self.get_thing_by_id('album', id) return self.get_thing_by_id('album', id)
def get_album_by_path(self, filepath):
'''
Return the album with the `associated_directory` of this value,
NOT case-sensitive.
'''
filepath = pathclass.Path(filepath).absolute_path
query = 'SELECT albumid FROM album_associated_directories WHERE directory == ?'
bindings = [filepath]
album_row = self.sql_select_one(query, bindings)
if album_row is None:
raise exceptions.NoSuchAlbum(filepath)
album_id = album_row[0]
return self.get_album(album_id)
def get_albums(self): def get_albums(self):
yield from self.get_things(thing_type='album') yield from self.get_things(thing_type='album')
def get_albums_by_id(self, ids): def get_albums_by_id(self, ids):
return self.get_things_by_id('album', ids) return self.get_things_by_id('album', ids)
def get_albums_by_path(self, directory):
'''
Yield Albums with the `associated_directory` of this value,
NOT case-sensitive.
'''
directory = pathclass.Path(directory)
query = 'SELECT albumid FROM album_associated_directories WHERE directory == ?'
bindings = [directory.absolute_path]
album_rows = self.sql_select(query, bindings)
for album_row in album_rows:
album_id = album_row[0]
yield self.get_album(album_id)
def get_root_albums(self): def get_root_albums(self):
''' '''
Yield Albums that have no parent. Yield Albums that have no parent.