"""Model representing a product_provided_store_artists."""

from oto import response
from sqlalchemy import Column
from sqlalchemy import Enum
from sqlalchemy import Integer

from product_digital.connectors import mysql


class ProductProvidedStoreArtists(mysql.BaseModel):
    """product_provided_store_artists Model."""

    __tablename__ = 'product_provided_store_artists'

    product_provided_store_artists_id = Column(
        Integer, primary_key=True, autoincrement=True, nullable=False)
    spotify = Column(Enum('Y', 'N'), default='N')
    release_id = Column(Integer)

    def to_dict(self):
        """Return a dictionary of these properties."""
        return {
            'product_provided_store_artists_id': self.product_provided_store_artists_id,
            'spotify': self.spotify,
            'release_id': self.release_id
        }


def create(release_id):
    """Create a new product_provided_store_artists entry.

    Returns:
        Response: Response containing the product_provided_store_artists entry saved.
    """
    product_provided_store_artists = ProductProvidedStoreArtists(
        release_id=release_id, spotify='Y')
    with mysql.db_session() as session:
        session.add(product_provided_store_artists)
        session.flush()
        return response.Response(
            message=product_provided_store_artists.to_dict(), status=201)