def add_album_with_contributor(title, username):
"""
Used when creating the album, as it needs to be related to the creator
"""
album = Album(title=title)
album.save()
ContributorAlbum(slug=album.slug, username=username).save()
return album
def add_contributor_album(slug, username):
"""
Used when adding an existent contributor to an existent album
"""
contrib = Contributor.get(username)
album = Album.get(slug)
ContributorAlbum(slug=album.slug, username=contrib.username).save()
def get_albums_by_username(username):
return [Album.get(calbum.slug) for calbum in ContributorAlbum.scan({"username": condition.EQ(username)})]
def get_album_by_slug(slug):
return Album.get(slug)
def get_album_by_title(title):
gen = Album.scan({"title": condition.EQ(title)})
for data in gen: return data
