from sqlalchemy import Column, ForeignKey, Integer, String, Table, orm

from account.connectors import mysql


class ServiceTier(mysql.BaseModel):
    """ServiceTier DB model."""

    __tablename__ = 'service_tier'

    id = Column(Integer, primary_key=True)
    uuid = Column(String, nullable=False, unique=True)
    name = Column(String, nullable=False)
    display_name = Column(String, nullable=False)

    @classmethod
    def get_by_uuid(cls, uuid: str, session: orm.Session):
        return session.query(cls).filter_by(uuid=uuid).one_or_none()


vendor_service_tier = Table(
    'vendor_service_tier',
    mysql.BaseModel.metadata,
    Column('service_tier_uuid', ForeignKey('service_tier.uuid')),
    Column('vendor_id', ForeignKey('vendor.vendor_id')),
)