"""track_writer module."""

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

from artist.connectors import mysql


class TrackWriter(mysql.BaseModel):
    """Track writer Model.

    Track writer metadata.
    """

    __tablename__ = 'track_writer'

    track_writer_id = Column(
        Integer,
        primary_key=True,
        autoincrement=True,
        nullable=False
    )
    writer_name = Column(String)
    unique_track_id = Column(Integer, nullable=False)
    artist_info_id = Column(Integer)

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

        Returns:
            dict: track writer metadata.
        """
        return {
            'id': self.track_writer_id,
            'track_id': self.unique_track_id,
            'name': self.writer_name
        }