Source code for ace.centralize.graphql.selections.alert

"""
Predefined SelectionSets for Alert entity.

Based on the Alert GraphQL schema output types:
    AlertDetail, AlertAffectedEntity, CreateAlertsOutput,
    AddEntitiesToAlertOutput, DeleteAlertsOutput, RemoveAffectedEntitiesOutput
"""

from ace.core.graphql import SelectionSet




class AlertAffectedEntitySelections:
    """Reusable selection sets for AlertAffectedEntity fields."""

    STANDARD = SelectionSet(
        "entityTypeId",
        "entityTypeName",
        "ids",
    )





class AlertSelections:
    """Reusable selection sets for Alert fields."""

    MINIMAL = SelectionSet(
        "alertId",
    )

    STANDARD = SelectionSet(
        "alertId",
        "alertTypeId",
        "entityId",
        "entityTypeId",
        "entityTypeName",
    )

    FULL = SelectionSet(
        "alertId",
        "alertTypeId",
        "entityId",
        "entityTypeId",
        "entityTypeName",
        nested={
            "alertAffectedEntities": AlertAffectedEntitySelections.STANDARD,
        }
    )

    # Kept for backward compatibility
    WITH_AFFECTED_ENTITIES = FULL