"""Records model."""
from enum import Enum

from db import db


class RecordType(Enum):
    Playlist = 3
    Artist = 4


class Record(db.Model):
    """Records model."""

    __tablename__ = "Record"

    id = db.Column(db.BigInteger, primary_key=True)
    name = db.Column(db.String, nullable=True)

    artist_id = db.Column(db.BigInteger, db.ForeignKey("Artist.id", ondelete="CASCADE"), nullable=False)

    record_type_id = db.Column(db.Integer, db.ForeignKey("RecordType.id"), nullable=False)

    artist = db.relationship("Artist")

    record_type = db.relationship("RecordType")

    __table_args__ = (db.UniqueConstraint("name", "artist_id", "record_type_id"),)

    def __init__(self, name: str, artist_id: int, record_type_id: int):
        """Constructor.

        Args:
            name (str): Record name.
            artist_id (int): Artist ID.
            record_type_id (int): Record type ID.
        """
        self.name = name
        self.artist_id = artist_id
        self.record_type_id = record_type_id