mirror of
https://github.com/graphql-python/graphene-django.git
synced 2024-11-22 01:27:01 +03:00
9a773b9d7b
* Use ruff in pre-commit * Add pyupgrade * Add isort * Add bugbear * Fix B015 Pointless comparison * Fix B026 * B018 false positive * Remove flake8 and isort config from setup.cfg * Remove black and flake8 from dev dependencies * Update black * Show list of fixes applied with autofix on * Fix typo * Add C4 flake8-comprehensions * Add ruff to dev dependencies * Fix up
42 lines
1.1 KiB
Python
42 lines
1.1 KiB
Python
class Registry:
|
|
def __init__(self):
|
|
self._registry = {}
|
|
self._field_registry = {}
|
|
|
|
def register(self, cls):
|
|
from .types import DjangoObjectType
|
|
|
|
assert issubclass(
|
|
cls, DjangoObjectType
|
|
), f'Only DjangoObjectTypes can be registered, received "{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):
|
|
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
|