plugin.py

import os
import json


class JSONSchemaPlugin(Plugin):
    """Provides support for the JSON Schema as TypedDict."""

    JSONSchema = 'jsonschema_typed.types.JSONSchema'
    JSONSchemaBase = 'jsonschema_typed.types.JSONSchemaBase'

    def get_type_analyze_hook(self, fullname: str) -> Optional[Callable]:
        """Produce an analyzer callback if a JSONSchema annotation is found."""
        if fullname == self.JSONSchema:
            def callback(ctx: AnalyzeTypeContext) -> TypedDictType:
                """Generate annotations from a JSON Schema."""
                schema_path = ctx.type.args[0].literal_value
                full_schema_path = os.path.abspath(schema_path)
                with open(full_schema_path) as f:
                    schema = json.load(f)
                return make_type(ctx, schema, full_schema_path)
            return callback