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

from sqlalchemy import Column, Integer, String, orm

from permissions.connectors import mysql


class ParentCompany(mysql.Base):
    """Class for parent company table."""

    __tablename__ = 'parent_company'

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

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