"""Model for subaccount."""

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

from permissions.connectors import mysql


class Subaccount(mysql.Base):
    """(Very minimal) class for subaccount table."""

    __tablename__ = 'subaccount'

    subaccount_id = Column(Integer, primary_key=True)
    subaccount_uuid = Column(String)
    vendor_id = Column(Integer, ForeignKey('vendor.vendor_id'))


def get_subaccount_by_uuid(session: orm.session.Session, uuid: str) -> Subaccount:
    """Return a dict of subaccount information given a subaccount uuid."""
    subaccount = session.query(Subaccount).filter_by(subaccount_uuid=uuid).one()
    return subaccount