graphene/graphene/utils/get_fields.py

from collections import OrderedDict

from .get_graphql_type import get_graphql_type
from .is_graphene_type import is_graphene_type
from ..types.field import Field, InputField
from ..types.unmountedtype import UnmountedType


def get_fields_from_attrs(in_type, attrs):
    for attname, value in list(attrs.items()):
        is_field = isinstance(value, (Field, InputField))
        is_field_proxy = isinstance(value, UnmountedType)
        if not (is_field or is_field_proxy):
            continue
        field = value.as_mounted(in_type) if is_field_proxy else value
        yield attname, field


def get_fields_from_types(bases):
    fields = set()
    for _class in bases:
        for attname, field in get_graphql_type(_class).get_fields().items():
            if attname in fields:
                continue
            fields.add(attname)
            yield attname, field


def get_fields(in_type, attrs, bases):
    fields = []

    graphene_bases = tuple(
        base._meta.graphql_type for base in bases if is_graphene_type(base) and not base._meta.abstract
    )

    extended_fields = list(get_fields_from_types(graphene_bases))
    local_fields = list(get_fields_from_attrs(in_type, attrs))

    field_names = set(f[0] for f in local_fields)
    for name, extended_field in extended_fields:
        if name in field_names:
            continue
        fields.append((name, extended_field))
        field_names.add(name)

    fields.extend(local_fields)

    return OrderedDict(sorted(fields, key=lambda kv: kv[1]))