"""Genre Modeling."""

from sqlalchemy import Column
from sqlalchemy import String
from sqlalchemy import SmallInteger

from backend.connectors import mysql


class Genre(mysql.BaseModel):
    """Model for Genre."""

    __tablename__ = 'genre'

    genre_id = Column(SmallInteger, primary_key=True, autoincrement=True)
    genre = Column(String(15), nullable=False)

    def to_dict(self):
        """Genre data dictionary.

        Returns:
            dict
        """
        return {
            'genre_id': self.genre_id,
            'genre': self.genre
        }