from db import db


class SpotifyAlbum(db.Model):
    """Spotify album model"""

    __tablename__ = "SpotifyAlbum"

    album_id = db.Column(db.BigInteger, db.ForeignKey("Album.id", ondelete="CASCADE"), primary_key=True)
    spotify_id = db.Column(db.String, nullable=False, unique=True)
    url = db.Column(db.String, nullable=False, unique=True)
    cover_art = db.Column(db.String, nullable=False)

    def __str__(self):
        return f"Spotify album id {self.album_id}"