class Registry(object): def __init__(self): self._registry = {} self._field_registry = {} def register(self, cls): from .types import DjangoObjectType assert issubclass( cls, DjangoObjectType ), 'Only DjangoObjectTypes can be registered, received "{}"'.format( cls.__name__ ) assert cls._meta.registry == self, "Registry for a Model have to match." # assert self.get_type_for_model(cls._meta.model) == cls, ( # 'Multiple DjangoObjectTypes registered for "{}"'.format(cls._meta.model) # ) if not getattr(cls._meta, "skip_registry", False): self._registry[cls._meta.model] = cls def get_type_for_model(self, model): print(model) if isinstance(model, str): model = self.get_model_by_string(model) return self._registry.get(model) def get_model_by_string(self, model): print(self.__dict__) print(self._registry.__dict__) print(self._field_registry) print(self._registry.keys()) return list(filter(lambda x: model == x.__name__, self._registry.keys()))[0] 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