Class OrgModel

app/api/agentops/opsboard/models.py:150–539

Organization model that maps to the orgs table

class OrgModel(BaseModel):
    """Organization model that maps to the orgs table"""

    __tablename__ = "orgs"
    __table_args__ = {"schema": "public"}

    id = model.Column(model.UUID, primary_key=True, default=uuid.uuid4)
    name = model.Column(model.String, nullable=False)
    prem_status = model.Column(model.Enum(PremStatus), default=PremStatus.free, nullable=False)
    subscription_id = model.Column(model.String, nullable=True)

    users = orm.relationship("UserOrgModel", back_populates="org", cascade="all, delete-orphan")
    projects = orm.relationship("ProjectModel", back_populates="org")
    invites = orm.relationship("OrgInviteModel", back_populates="org", cascade="all, delete-orphan")

    def set_current_user(self, user_id: str | UUID) -> None:
        """
        Set the current user for this organization. THis is the user performing
        the request.

        This allows us to access `current_user_role` as a property later.
        """
        user_org = self.get_user_membership(user_id)
        self._current_user_role = user_org.role

    @property
    def current_user_role(self) -> OrgRoles:
        """Get the current user's role in this organization."""
        # this property helps us render this field in the response schema
        if not hasattr(self, "_current_user_role"):
            raise AttributeError("Current user role is not set. Call set_current_user() first.")

        return self._current_user_role

    @property
    def is_freeplan(self) -> bool:
        """Check if the organization is on a free plan."""
        return self.prem_status == PremStatus.free

    @property
    def current_member_count(self) -> int:
        """The number of users that are a member of this organization."""
        # Use optimized count if available
        if hasattr(self, '_member_count'):
            return self._member_count
        # this includes the count of invites, too, so we can prevent the admin
        # from over-inviting users
        if not hasattr(self, 'users') or not hasattr(self, 'invites'):
            raise AttributeError(
                "Member count not available. Use get_by_id or get_all_for_user to load relationships."
            )
        return len(self.users) + len(self.invites)

    @property
    def max_member_count(self) -> int | None:
        """The maximum number of users that can be a member of this organization."""
        return FREEPLAN_MAX_USERS if self.is_freeplan else None