__init__.py

import inspect
from typing import Dict

from . import schema_definitions
from .itembase import Edge, ItemBase
from .schema_definitions import *
from .utils import resolve_forward_refs

__version__ = "0.0.1"


def get_schema() -> Dict[str, type]:
    return {
        k: v
        for k, v in inspect.getmembers(schema_definitions)
        if inspect.isclass(v) and issubclass(v, schema_definitions.Item)
    }


_schema_definitions = get_schema()
resolve_forward_refs(_schema_definitions)