"""Model for vendor."""

from sqlalchemy import Column, Integer, String, orm

from permissions.connectors import mysql


class Vendor(mysql.Base):
    """(Very minimal) class for vendor table."""

    __tablename__ = 'vendor'

    vendor_id = Column(Integer, primary_key=True)
    vendor_uuid = Column(String)

    @classmethod
    def get_by_uuid(cls, session: orm.session.Session, uuid: str) -> 'Vendor':
        """Return a vendor given a vendor uuid."""
        vendor = session.query(cls).filter_by(vendor_uuid=uuid).one()
        return vendor


def get_vendor_id_by_uuid(session: orm.session.Session, uuid: str) -> int:
    """Return the id of a vendor identified by uuid."""
    vendor = Vendor.get_by_uuid(session, uuid)
    return vendor.vendor_id