feat: implement select field resolver

Co-authored-by: Abadulrehman <[email protected]>

Merge-request: ROMMAN-MR-196
Merged-by: Fahim Ali Zain <[email protected]>

Author: Abadulrehman

frappe_graphql/utils/resolver/__init__.py

@@ -5,6 +5,7 @@
 
 from .root_query import setup_root_query_resolvers
 from .link_field import setup_link_field_resolvers
+from .select_fields import setup_select_field_resolvers
 from .child_tables import setup_child_table_resolvers
 from .translate import setup_translatable_resolvers
 from .utils import get_singular_doctype
@@ -62,7 +63,3 @@ def setup_doctype_resolver(meta: Meta, gql_type: GraphQLType):
 def _doctype_resolver(obj, info: GraphQLResolveInfo, **kwargs):
     dt = get_singular_doctype(info.parent_type.name)
     return dt
-
-
-def setup_select_field_resolvers(meta: Meta, gql_type: GraphQLType):
-    pass

frappe_graphql/utils/resolver/select_fields.py

@@ -0,0 +1,35 @@
+from graphql import GraphQLType, GraphQLResolveInfo, GraphQLNonNull, GraphQLEnumType
+
+import frappe
+from frappe.model.meta import Meta
+
+from .translate import _translatable_resolver
+from .utils import get_frappe_df_from_resolve_info
+
+
+def setup_select_field_resolvers(meta: Meta, gql_type: GraphQLType):
+
+    for df in meta.get_select_fields():
+
+        if df.fieldname not in gql_type.fields:
+            continue
+
+        gql_field = gql_type.fields[df.fieldname]
+        gql_field.resolve = _select_field_resolver
+
+
+def _select_field_resolver(obj, info: GraphQLResolveInfo, **kwargs):
+
+    df = get_frappe_df_from_resolve_info(info)
+    return_type = info.return_type
+
+    if isinstance(return_type, GraphQLNonNull):
+        return_type = return_type.of_type
+
+    if isinstance(return_type, GraphQLEnumType):
+        return frappe.scrub(obj.get(info.field_name)).upper()
+
+    if df and df.translatable:
+        return _translatable_resolver(obj, info, **kwargs)
+
+    return obj.get(info.field_name)

frappe_graphql/utils/resolver/translate.py

@@ -10,6 +10,10 @@ def setup_translatable_resolvers(meta: Meta, gql_type: GraphQLType):
             continue
 
         gql_field = gql_type.fields[df_fieldname]
+
+        if gql_field.resolve:
+            continue
+
         gql_field.resolve = _translatable_resolver