mirror of
				https://github.com/graphql-python/graphene-django.git
				synced 2025-11-04 09:57:53 +03:00 
			
		
		
		
	* 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
 |