"""Country Model."""
from sqlalchemy import Column
from sqlalchemy import Integer
from sqlalchemy import String

from product.connectors import mysql


class Country(mysql.BaseModel):
    """Country Model class."""

    __tablename__ = 'country'

    id = Column(  # noqa
        Integer, primary_key=True, autoincrement=True, nullable=False)
    name = Column(String(45), nullable=False)
    country_code = Column(String(2), nullable=False, unique=True)

    def to_dict(self):
        """Convert object to dict.

        Returns:
            dict
        """
        return {
            'id': self.id,
            'name': self.name,
            'country_code': self.country_code,
        }


def get_country_ids(country_codes):
    """Returns country ID's for given country codes.

    Args:
        country_codes (list):

    Returns:
        list
    """
    with mysql.db_session() as session:
        result = session.query(Country).filter(
            Country.country_code.in_(country_codes)
        ).all()

        return [row.to_dict() for row in result]