from common import utils from models.base import BaseModel class Album(BaseModel): """Represents an album.""" def __init__(self, db, id=None, **kwargs): self._db = db if id is not None: for row in self._db.execute("SELECT * FROM album WHERE id = ?", (id,)): setattr(self, "id", id) setattr(self, "name", row[1]) setattr(self, "date", row[2]) else: for (key, value) in kwargs.items(): setattr(self, key, value) def delete(self): for track in self.tracks: track.delete() with self._db.conn: delete_album = "DELETE FROM album WHERE id = ?" self._db.execute(delete_album, (self.id,)) delete_track_rel = "DELETE FROM album_track WHERE album_id = ?" self._db.execute(delete_track_rel, (self.id,)) delete_artist_rel = "DELETE FROM album_artist WHERE album_id = ?" self._db.execute(delete_artist_rel, (self.id,)) return True @property def db(self): return self._db @db.setter def db(self, db): self._db = db @property def artists(self): from models.artist import Artist if not hasattr(self, "_artists"): setattr(self, "_artists", []) for row in self._db.execute("SELECT artist.* FROM artist INNER " "JOIN album_artist ON artist.id = " "album_artist.artist_id WHERE " "album_id = ? ORDER BY name ASC", (self.id,)): artist = Artist(id=row[0], db=self._db, name=row[1], sortname=row[2], musicbrainz_artistid=row[3]) self._artists.append(artist) return self._artists @property def tracks(self): from models.track import Track if not hasattr(self, "_tracks"): setattr(self, "_tracks", []) for row in self._db.execute("SELECT track.* FROM track INNER " "JOIN album_track ON track.id = " "album_track.track_id WHERE " "album_id = ? ORDER BY tracknumber " "ASC", (self.id,)): track = Track(id=row["id"], db=self._db, tracknumber=row["tracknumber"], name=row["name"], grouping=row["grouping"], filename=row["filename"]) self._tracks.append(track) return self._tracks def save(self): dirty_attributes = {} # check if the internal dict has been modified for (attr, value) in self.__dict__.items(): if self.__data[attr] != getattr(self, attr): dirty_attributes[attr] = value if len(dirty_attributes) > 0: set_clause = utils.update_clause_from_dict(dirty_attributes) dirty_attributes["id"] = self.id sql = " ".join(("UPDATE album", set_clause, "WHERE id = :id")) with self._db.conn: self._db.execute(sql, dirty_attributes) @classmethod def search(cls, database, **search_params): """Find an album with the given params Args: name: dict, with 'data' and 'operator' keys date: dict, with 'data' and 'operator' keys musicbrainz_albumid: dict, with 'data' and 'operator' keys """ albums = [] # unpack search params where_params = {} value_params = {} for (attr, value) in search_params.items(): where_params[attr] = value["operator"] if value["operator"] == "BETWEEN": items = value["data"].split(" ") value_params["".join((attr, "1"))] = items[0] value_params["".join((attr, "2"))] = items[2] else: value_params[attr] = value["data"] where_clause = utils.make_where_clause(where_params) result = None if where_clause: statement = " ".join(("SELECT * FROM album", where_clause)) result = database.execute(statement, value_params) else: result = database.execute("SELECT * FROM album") for row in result: albums.append( Album(id=row["id"], db=database, name=row["name"], date=row["date"]) ) return albums @classmethod def all(cls, database, order="album.id", direction="ASC", limit=None, offset=None): albums = [] select_string = """SELECT * FROM album LEFT JOIN album_artist ON album_artist.album_id = album.id LEFT JOIN artist ON album_artist.artist_id = artist.id ORDER BY %s %s""" % (order, direction) if limit is not None and offset is not None: select_string = " ".join((select_string, "LIMIT %s OFFSET %s" % (limit, offset))) result = database.execute(select_string) for row in result: albums.append( Album(id=row["id"], db=database, name=row["name"], date=row["date"]) ) return albums