mirror of
https://github.com/graphql-python/graphene.git
synced 2024-11-14 21:56:54 +03:00
30 lines
872 B
Python
30 lines
872 B
Python
|
class Registry(object):
|
||
|
def __init__(self):
|
||
|
self._registry = {}
|
||
|
self._registry_models = {}
|
||
|
|
||
|
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)
|
||
|
self._registry[cls._meta.model] = cls
|
||
|
|
||
|
def get_type_for_model(self, model):
|
||
|
return self._registry.get(model)
|
||
|
|
||
|
|
||
|
registry = None
|
||
|
|
||
|
|
||
|
def get_global_registry():
|
||
|
global registry
|
||
|
if not registry:
|
||
|
registry = Registry()
|
||
|
return registry
|
||
|
|
||
|
|
||
|
def reset_global_registry():
|
||
|
global registry
|
||
|
registry = None
|