class Registry(object): def __init__(self): self._registry = {} self._field_registry = {} def register(self, cls): from .schema import SIGsObjectType assert issubclass( cls, SIGsObjectType ), 'Apenas SIGsObjectTypes podem ser registrados, received "{}"'.format( cls.__name__ ) self._registry[cls._meta.model] = cls def get_type_for_model(self, model): return self._registry.get(model) def register_converted_field(self, field, converted): self._field_registry[field] = converted def get_converted_field(self, field): return self._field_registry.get(field) registry = None def get_global_registry(): global registry if not registry: registry = Registry() return registry def reset_global_registry(): global registry registry = None