Ask Solem 12 vuotta sitten
vanhempi
commit
3fb0927b69
2 muutettua tiedostoa jossa 6 lisäystä ja 8 poistoa
  1. 1 0
      celery/security/__init__.py
  2. 5 8
      celery/security/serialization.py

+ 1 - 0
celery/security/__init__.py

@@ -84,3 +84,4 @@ def setup_security(allowed_serializers=None, key=None, cert=None, store=None,
     with open(key) as kf:
         with open(cert) as cf:
             register_auth(kf.read(), cf.read(), store, digest, serializer)
+    registry._set_default_serializer('auth')

+ 5 - 8
celery/security/serialization.py

@@ -80,14 +80,15 @@ class SecureSerializer(object):
         signer_cert = self._cert_store[signer]
 
         sig_len = signer_cert._cert.get_pubkey().bits() >> 3
-        signature = raw_payload[first_sep+len(sep):first_sep+len(sep)+sig_len]
-        end_of_sig = first_sep+len(sep)+sig_len+len(sep)
-        
+        signature = raw_payload[
+            first_sep + len(sep):first_sep + len(sep) + sig_len
+        ]
+        end_of_sig = first_sep + len(sep) + sig_len+len(sep)
+
         v = raw_payload[end_of_sig:].split(sep)
 
         values = [bytes_to_str(signer), bytes_to_str(signature),
                   bytes_to_str(v[0]), bytes_to_str(v[1]), bytes_to_str(v[2])]
-                  
 
         return {
             'signer': values[0],
@@ -98,8 +99,6 @@ class SecureSerializer(object):
         }
 
 
-
-
 def register_auth(key=None, cert=None, store=None, digest='sha1',
                   serializer='json'):
     """register security serializer"""
@@ -110,5 +109,3 @@ def register_auth(key=None, cert=None, store=None, digest='sha1',
     registry.register('auth', s.serialize, s.deserialize,
                       content_type='application/data',
                       content_encoding='utf-8')
-
-    registry._set_default_serializer('auth')