graphql.py

"""GraphQL schema."""

import strawberry
from django.conf import settings
from strawberry.extensions import DisableIntrospection, QueryDepthLimiter
from strawberry_django.optimizer import DjangoOptimizerExtension

from apps.api.internal.mutations import ApiMutations
from apps.api.internal.queries import ApiKeyQueries
from apps.github.api.internal.queries import GithubQuery
from apps.mentorship.api.internal.mutations import (
    ModuleMutation,
    ProgramMutation,
)
from apps.mentorship.api.internal.queries import (
    MentorshipQuery,
    ModuleQuery,
    ProgramQuery,
)
from apps.nest.api.internal.mutations import NestMutations
from apps.owasp.api.internal.mutations import SponsorMutation
from apps.owasp.api.internal.queries import OwaspQuery


@strawberry.type
class Mutation(ApiMutations, ModuleMutation, NestMutations, ProgramMutation, SponsorMutation):
    """Schema mutations."""


@strawberry.type
class Query(
    ApiKeyQueries,
    GithubQuery,
    MentorshipQuery,
    ModuleQuery,
    OwaspQuery,
    ProgramQuery,
):
    """Schema queries."""


extensions = [
    QueryDepthLimiter(max_depth=5),
    DjangoOptimizerExtension(),
]

if not settings.DEBUG and not settings.IS_FUZZ_ENVIRONMENT:
    extensions.append(DisableIntrospection())

schema = strawberry.Schema(extensions=extensions, mutation=Mutation, query=Query)