"""Model for company brand (mysql db)."""

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

from permissions.connectors import mysql


class CompanyBrand(mysql.Base):
    """Class for company brand table."""

    __tablename__ = 'company_brand'

    id = Column(Integer, primary_key=True)
    uuid = Column(String)
    name = Column(String)
    display_name = Column(String)
    parent_company_id = Column(Integer, ForeignKey('parent_company.id'))

    @classmethod
    def get_by_uuid(cls, session: orm.session.Session, uuid: str) -> 'CompanyBrand | None':
        """Return a CompanyBrand given a uuid."""
        return session.query(cls).filter_by(uuid=uuid).one_or_none()