Rooba
/
kombu
mirror of https://github.com/celery/kombu.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Adds ability to disable content_types in the serializer registry

This commit is contained in:
Ask Solem 2011-11-28 13:09:23 +00:00
parent 9fa04affa8
commit 2c459e4c8c
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
kombu/serialization.py
View File

@ -65,6 +65,7 @@ class SerializerRegistry(object):
self._default_encode = None self._default_encode = None
self._default_content_type = None self._default_content_type = None
self._default_content_encoding = None self._default_content_encoding = None
self._disabled_content_types = set()
self.type_to_name = {} self.type_to_name = {}
def register(self, name, encoder, decoder, content_type, def register(self, name, encoder, decoder, content_type,
@ -75,6 +76,11 @@ class SerializerRegistry(object):
self._decoders[content_type] = decoder self._decoders[content_type] = decoder
self.type_to_name[content_type] = name self.type_to_name[content_type] = name
def disable(self, name):
if '/' not in name:
name = self.type_to_name[name]
self._disabled_content_types.add(name)
def unregister(self, name): def unregister(self, name):
try: try:
content_type = self._encoders[name][0] content_type = self._encoders[name][0]
@ -134,7 +140,10 @@ class SerializerRegistry(object):
payload = encoder(data) payload = encoder(data)
return content_type, content_encoding, payload return content_type, content_encoding, payload
def decode(self, data, content_type, content_encoding): def decode(self, data, content_type, content_encoding, force=False):
if content_type in self._disabled_content_types:
raise SerializerNotInstalled(
"Content-type %r has been disabled." % (content_type, ))
content_type = content_type or 'application/data' content_type = content_type or 'application/data'
content_encoding = (content_encoding or 'utf-8').lower() content_encoding = (content_encoding or 'utf-8').lower()