enums.py

import graphene

from user.enums import enum_map as user_enum_map
from user_group.enums import enum_map as user_group_enum_map
from project.enums import enum_map as project_enum_map
from analysis_framework.enums import enum_map as analysis_framework_enum_map
from lead.enums import enum_map as lead_enum_map
from entry.enums import enum_map as entry_enum_map
from export.enums import enum_map as export_enum_map
from quality_assurance.enums import enum_map as quality_assurance_enum_map
from analysis.enums import enum_map as analysis_enum_map
from notification.enums import enum_map as notification_enum_map
from unified_connector.enums import enum_map as unified_connector_enum_map
from assisted_tagging.enums import enum_map as assisted_tagging_enum_map
from ary.enums import enum_map as ary_enum_map
from assessment_registry.enums import enum_map as assessment_reg_enum_map

ENUM_TO_GRAPHENE_ENUM_MAP = {
    **user_enum_map,
    **user_group_enum_map,
    **project_enum_map,
    **analysis_framework_enum_map,
    **lead_enum_map,
    **entry_enum_map,
    **export_enum_map,
    **quality_assurance_enum_map,
    **analysis_enum_map,
    **notification_enum_map,
    **unified_connector_enum_map,
    **assisted_tagging_enum_map,
    **ary_enum_map,
    **assessment_reg_enum_map,
}

ENUM_TO_GRAPHENE_ENUM_DESCRIPTION_MAP = {
    enum: getattr(enum._meta.enum, '__description__', {})
    for enum in ENUM_TO_GRAPHENE_ENUM_MAP.values()
}


def generate_type_for_enum(name, Enum):
    EnumMetaType = type(
        f'AppEnumCollection{name}',
        (graphene.ObjectType,),
        {
            'enum': graphene.NonNull(Enum),
            'label': graphene.NonNull(graphene.String),
            'description': graphene.String(),
        }
    )
    return graphene.Field(
        graphene.List(
            graphene.NonNull(EnumMetaType),
        ),
        resolver=lambda *_: [
            EnumMetaType(
                enum=e,
                label=e.label,
                description=ENUM_TO_GRAPHENE_ENUM_DESCRIPTION_MAP[Enum].get((e.value, e.label)),
            )
            for e in Enum._meta.enum
        ]
    )


def generate_type_for_enums(name):
    return type(
        name,
        (graphene.ObjectType,),
        {
            enum_field_name: generate_type_for_enum(
                enum_field_name,
                enum_,
            )
            for enum_field_name, enum_ in ENUM_TO_GRAPHENE_ENUM_MAP.items()
        },
    )


AppEnumCollection = generate_type_for_enums('AppEnumCollection')