"""VendContact Model.

This VendContact model uses sqlalchemy.
"""

import enum

from sqlalchemy import Column, Enum, Integer, String, orm

from account.connectors import mysql


class YNEnum(enum.Enum):
    """Y/N enum."""

    Y = 'Y'
    N = 'N'


class VendContact(mysql.BaseModel):
    """VendContact DB Model."""

    __tablename__ = 'vend_contact'

    vend_contact_id = Column(Integer, name='id', primary_key=True)
    vendor_id = Column(Integer)
    subaccount_id = Column(Integer)
    active = Column(Enum(YNEnum))
    master = Column(Enum(YNEnum))
    auth0_user_id = Column(String)

    @classmethod
    def get_master_contact(cls, tx: orm.session.Session, vendor_id: int):
        """Get master contact for vendor."""
        return tx.query(cls).filter_by(vendor_id=vendor_id, master='Y', active='Y').all()