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

Add get_things_by_sql for quick, bespoke queries.

Browse source
This commit is contained in:
voussoir 2020-09-28 14:07:28 -07:00
parent 98f706f486
commit 585882028d
1 changed files with 23 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
etiquette/photodb.py
View file

@ -53,6 +53,9 @@ class PDBAlbumMixin:
album_ids = (album_id for (album_id,) in album_rows) album_ids = (album_id for (album_id,) in album_rows)
return self.get_albums_by_id(album_ids) return self.get_albums_by_id(album_ids)
def get_albums_by_sql(self, query, bindings=None):
return self.get_things_by_sql('album', query, bindings)
def get_root_albums(self): def get_root_albums(self):
''' '''
Yield Albums that have no parent. Yield Albums that have no parent.
@ -154,6 +157,9 @@ class PDBBookmarkMixin:
def get_bookmarks_by_id(self, ids): def get_bookmarks_by_id(self, ids):
return self.get_things_by_id('bookmark', ids) return self.get_things_by_id('bookmark', ids)
def get_bookmarks_by_sql(self, query, bindings=None):
return self.get_things_by_sql('bookmark', query, bindings)
@decorators.required_feature('bookmark.new') @decorators.required_feature('bookmark.new')
@decorators.transaction @decorators.transaction
def new_bookmark(self, url, title=None, *, author=None): def new_bookmark(self, url, title=None, *, author=None):
@ -374,6 +380,11 @@ class PDBCacheManagerMixin:
thing_cache[thing.id] = thing thing_cache[thing.id] = thing
yield thing yield thing
def get_things_by_sql(self, thing_type, query, bindings=None):
thing_rows = self.sql_select(query, bindings)
thing_ids = (thing_id for (thing_id,) in thing_rows)
return self.get_things_by_id(thing_type, thing_ids)
#################################################################################################### ####################################################################################################
class PDBPhotoMixin: class PDBPhotoMixin:
@ -436,6 +447,9 @@ class PDBPhotoMixin:
if count <= 0: if count <= 0:
break break
def get_photos_by_sql(self, query, bindings=None):
return self.get_things_by_sql('photo', query, bindings)
@decorators.required_feature('photo.new') @decorators.required_feature('photo.new')
@decorators.transaction @decorators.transaction
def new_photo( def new_photo(
@ -1193,6 +1207,9 @@ class PDBTagMixin:
def get_tags_by_id(self, ids): def get_tags_by_id(self, ids):
return self.get_things_by_id('tag', ids) return self.get_things_by_id('tag', ids)
def get_tags_by_sql(self, query, bindings=None):
return self.get_things_by_sql('tag', query, bindings)
@decorators.required_feature('tag.new') @decorators.required_feature('tag.new')
@decorators.transaction @decorators.transaction
def new_tag(self, tagname, description=None, *, author=None): def new_tag(self, tagname, description=None, *, author=None):
@ -1350,6 +1367,12 @@ class PDBUserMixin:
def get_users(self): def get_users(self):
return self.get_things('user') return self.get_things('user')
def get_users_by_id(self, ids):
return self.get_things_by_id('user', ids)
def get_users_by_sql(self):
return self.get_things_by_sql('user', query, bindings)
@decorators.required_feature('user.login') @decorators.required_feature('user.login')
def login(self, username=None, id=None, *, password): def login(self, username=None, id=None, *, password):
''' '''