Function generate

sdk/python/codegen/src/codegen/generator.py:195–262

Code generation main function.

(schema: GraphQLSchema, schema_version: str = "")

@joiner
def generate(schema: GraphQLSchema, schema_version: str = "") -> Iterator[str]:
    """Code generation main function."""
    yield textwrap.dedent(
        """\
        # Code generated by dagger. DO NOT EDIT.

        import warnings  # noqa: F401
        from collections.abc import Callable
        from dataclasses import dataclass
        from typing import Protocol, runtime_checkable

        from typing_extensions import Self

        from dagger.client._core import Arg
        from dagger.client._guards import typecheck
        from dagger.client.base import Enum, Input, Root, Scalar, Type
        """,
    )

    # Pre-create handy maps to make handler code simpler.
    ids = frozenset(n for n, t in schema.type_map.items() if is_id_type(t))

    # shared state between all handler instances
    ctx = Context(ids=ids, schema=schema, schema_version=schema_version)

    handlers: tuple[Handler, ...] = (
        Scalar(ctx),
        Enum(ctx),
        Input(ctx),
        InterfaceProtocol(ctx),
        Object(ctx),
    )

    if ctx.legacy_sdk_compat:
        for type_name in legacy_id_names(schema):
            yield legacy_id_class(type_name)
            ctx.defined.add(type_name)

    # Split into two iterators to update ctx.remaining.
    types_n, types_g = itertools.tee(get_grouped_types(handlers, schema.type_map))

    # Track types that haven't been defined yet, to format as a forward reference.
    ctx.remaining.update(name for _, name, _ in types_n)

    for handler, type_name, named_type in types_g:
        yield handler.render(named_type)
        ctx.process_type(type_name)

    yield ""
    yield ""
    yield "class Client(Query):"
    yield indent(
        '"""The Dagger client.\n'
        "\n"
        "Inherits all Query API methods and adds connection management.\n"
        '"""'
    )
    ctx.defined.add("Client")