from db import db


class AppleAlbum(db.Model):
    """Apple album model"""

    __tablename__ = "AppleAlbum"

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

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