Source code for ralph.mcp.protocol._drain_class

"""DrainClass — drain class used for capability defaults."""

from __future__ import annotations

from enum import StrEnum



[docs]
class DrainClass(StrEnum):
    """Drain class used for capability defaults."""

    PLANNING = "planning"
    DEVELOPMENT = "development"
    ANALYSIS = "analysis"
    REVIEW = "review"
    FIX = "fix"
    COMMIT = "commit"


[docs]
    def allows_write(self) -> bool:
        """Return whether this drain class allows write operations."""
        return self in {DrainClass.DEVELOPMENT, DrainClass.FIX}




__all__ = ["DrainClass"]