try again

Author: ianardee

mindee/parsing/v2/dynamic_field.py

@@ -1,9 +1,5 @@
 from enum import Enum
 
-from mindee.error import MindeeApiV2Error
-from mindee.parsing import v2
-from mindee.parsing.common.string_dict import StringDict
-
 
 class FieldType(str, Enum):
     """Field types."""
@@ -28,22 +24,3 @@ def __init__(self, field_type: FieldType, indent_level=0) -> None:
     def multi_str(self) -> str:
         """String representation of the field in a list."""
         return str(self)
-
-
-def get_field_type(raw_response: StringDict, indent_level: int = 0) -> DynamicField:
-    """Get appropriate field types."""
-    if isinstance(raw_response, dict):
-        if "value" in raw_response:
-            field_file = getattr(v2, "simple_field")
-            field_class = getattr(field_file, FieldType.SIMPLE.value)
-        elif "items" in raw_response:
-            field_file = getattr(v2, "list_field")
-            field_class = getattr(field_file, FieldType.LIST.value)
-        elif "fields" in raw_response:
-            field_file = getattr(v2, "object_field")
-            field_class = getattr(field_file, FieldType.OBJECT.value)
-        else:
-            raise MindeeApiV2Error(f"Unrecognized field format in {raw_response}.")
-        return field_class(raw_response, indent_level)
-
-    raise MindeeApiV2Error(f"Unrecognized field format {raw_response}.")

mindee/parsing/v2/get_field_type.py

@@ -0,0 +1,27 @@
+from importlib import import_module
+
+from mindee.error import MindeeApiV2Error
+from mindee.parsing.common.string_dict import StringDict
+from mindee.parsing.v2.dynamic_field import DynamicField, FieldType
+
+
+def get_field_type(raw_response: StringDict, indent_level: int = 0) -> DynamicField:
+    """Get appropriate field types."""
+    if isinstance(raw_response, dict):
+        if "value" in raw_response:
+            field_file = import_module("mindee.parsing.v2.simple_field")
+            # field_file = getattr(v2, "simple_field")
+            field_class = getattr(field_file, FieldType.SIMPLE.value)
+        elif "items" in raw_response:
+            field_file = import_module("mindee.parsing.v2.list_field")
+            # field_file = getattr(v2, "list_field")
+            field_class = getattr(field_file, FieldType.LIST.value)
+        elif "fields" in raw_response:
+            field_file = import_module("mindee.parsing.v2.object_field")
+            # field_file = getattr(v2, "object_field")
+            field_class = getattr(field_file, FieldType.OBJECT.value)
+        else:
+            raise MindeeApiV2Error(f"Unrecognized field format in {raw_response}.")
+        return field_class(raw_response, indent_level)
+
+    raise MindeeApiV2Error(f"Unrecognized field format {raw_response}.")

mindee/parsing/v2/inference_result_fields.py

@@ -1,7 +1,8 @@
 from typing import Dict
 
 from mindee.parsing.common.string_dict import StringDict
-from mindee.parsing.v2.dynamic_field import DynamicField, get_field_type
+from mindee.parsing.v2.dynamic_field import DynamicField, FieldType
+from mindee.parsing.v2.get_field_type import get_field_type
 
 
 class InferenceResultFields(Dict[str, DynamicField]):
@@ -22,7 +23,7 @@ def __getattr__(self, item):
     def __str__(self) -> str:
         str_fields = ""
         for field_key, field_value in self.items():
-            if field_value.field_type == "SimpleField":
+            if field_value.field_type == FieldType.SIMPLE:
                 str_fields += f"\n:{field_key}: {field_value}"
             else:
                 str_fields += f"\n:{field_key}:{field_value}"

mindee/parsing/v2/list_field.py

@@ -1,7 +1,8 @@
 from typing import List
 
 from mindee.parsing.common.string_dict import StringDict
-from mindee.parsing.v2.dynamic_field import DynamicField, FieldType, get_field_type
+from mindee.parsing.v2.dynamic_field import DynamicField, FieldType
+from mindee.parsing.v2.get_field_type import get_field_type
 
 
 class ListField(DynamicField):