"""track_artist module."""

from sqlalchemy import Column
from sqlalchemy import Integer
from sqlalchemy import String

from artist.connectors import mysql


class TrackArtist(mysql.BaseModel):
    """Track artist Model.

    Track artist metadata.
    """

    __tablename__ = 'track_artist'

    id = Column(    # noqa
        Integer,
        primary_key=True,
        autoincrement=True,
        nullable=False
    )
    artist_info_id = Column(Integer)
    artist_name = Column('name', String)
    role = Column('type', String)
    track_id = Column(Integer, nullable=False)

    def to_dict(self):
        """Return dictionary of track artist data.

        Returns:
            dict: track artist metadata.
        """
        return {
            'id': self.id,
            'name': self.artist_name,
            'role': self.role,
            'track_id': self.track_id
        }