Better Type Parsing Interface

jsonschema_pydantic/types/__init__.py

@@ -0,0 +1,10 @@
+# Exports generic type parser
+from ._type_parser import GenericTypeParser
+
+# Exports Implementations
+from .int_type_parser import IntTypeParser  # isort:skip
+from .object_type_parser import ObjectTypeParser  # isort:skip
+from .string_type_parser import StringTypeParser  # isort:skip
+from .array_type_parser import ArrayTypeParser  # isort:skip
+from .boolean_type_parser import BooleanTypeParser  # isort:skip
+from .float_type_parser import FloatTypeParser  # isort:skip

jsonschema_pydantic/types/_type_parser.py

@@ -0,0 +1,28 @@
+from abc import ABC, abstractmethod
+from typing import Generic, Self, TypeVar
+
+T = TypeVar("T")
+
+
+class GenericTypeParser(ABC, Generic[T]):
+    @property
+    @abstractmethod
+    def mapped_type(self) -> type[T]: ...
+
+    @property
+    @abstractmethod
+    def json_schema_type(self) -> str: ...
+
+    @staticmethod
+    @abstractmethod
+    def from_properties(
+        name: str, properties: dict[str, any]
+    ) -> tuple[type[T], dict[str, any]]: ...
+
+    @classmethod
+    def get_impl(cls, type_name: str) -> Self:
+        for subcls in cls.__subclasses__():
+            if subcls.json_schema_type == type_name:
+                return subcls
+
+        raise ValueError(f"Unknown type: {type_name}")

jsonschema_pydantic/types/array_type_parser.py

@@ -0,0 +1,21 @@
+from jsonschema_pydantic.types._type_parser import GenericTypeParser
+
+from typing import TypeVar
+
+V = TypeVar("V")
+
+
+class ArrayTypeParser(GenericTypeParser):
+    mapped_type = list
+
+    json_schema_type = "array"
+
+    @classmethod
+    def from_properties(cls, name, properties):
+        _item_type = properties["items"]["type"]
+        if _item_type == "object":
+            _item_type = type
+        else:
+            _item_type = GenericTypeParser.get_impl(_item_type).mapped_type
+
+        return list[_item_type], {}

jsonschema_pydantic/types/boolean_type_parser.py

@@ -0,0 +1,11 @@
+from jsonschema_pydantic.types._type_parser import GenericTypeParser
+
+
+class BooleanTypeParser(GenericTypeParser):
+    mapped_type = bool
+
+    json_schema_type = "boolean"
+
+    @staticmethod
+    def from_properties(name, properties):
+        return bool, {}

jsonschema_pydantic/types/float_type_parser.py

@@ -0,0 +1,11 @@
+from jsonschema_pydantic.types._type_parser import GenericTypeParser
+
+
+class FloatTypeParser(GenericTypeParser):
+    mapped_type = float
+
+    json_schema_type = "number"
+
+    @staticmethod
+    def from_properties(name, properties):
+        return float, {}

jsonschema_pydantic/types/int_type_parser.py

@@ -0,0 +1,11 @@
+from jsonschema_pydantic.types._type_parser import GenericTypeParser
+
+
+class IntTypeParser(GenericTypeParser):
+    mapped_type = int
+
+    json_schema_type = "integer"
+
+    @staticmethod
+    def from_properties(name, properties):
+        return int, {}

jsonschema_pydantic/types/object_type_parser.py

@@ -0,0 +1,14 @@
+from jsonschema_pydantic.types._type_parser import GenericTypeParser
+
+
+class ObjectTypeParser(GenericTypeParser):
+    mapped_type = object
+
+    json_schema_type = "object"
+
+    @staticmethod
+    def from_properties(name, properties):
+        from jsonschema_pydantic.schema_converter import SchemaConverter
+
+        _type = SchemaConverter.build_object(name, properties)
+        return _type, {}

jsonschema_pydantic/types/string_type_parser.py

@@ -0,0 +1,11 @@
+from jsonschema_pydantic.types._type_parser import GenericTypeParser
+
+
+class StringTypeParser(GenericTypeParser):
+    mapped_type = str
+
+    json_schema_type = "string"
+
+    @staticmethod
+    def from_properties(name, properties):
+        return str, {}