"""Vendor Advance model."""

from sqlalchemy import Column, Integer, String

from contracts import response
from contracts.connectors import mysql


class ServiceType(mysql.BaseModel):
    """ServiceType Model for contract service types."""

    __tablename__ = 'service_type'

    _id = Column(Integer, name='id', primary_key=True, nullable=False)
    service_type = Column(String, name='service_type', nullable=True)

    def to_dict(self):
        """Get a dict representation."""
        return {'id': self._id, 'service_type': self.service_type}


def get_service_types():
    """Get all service types.

    Returns:
        response.Response: service types.

    """
    with mysql.db_session() as session:
        query = session.query(ServiceType).order_by(ServiceType.service_type)
        service_types = [types.to_dict() for types in query.all()]
        payload = {'items': service_types}

        return response.Response(message=payload)