Skip to content

Commit

Permalink
Make load_payload() signature consistent between types (#75)
Browse files Browse the repository at this point in the history
Fixes #74.

The `JSONWebSignatureSerializer.load_payload` signature change is potentially breaking if anybody passed `return_header` as a second non-keyword argument to the method in the past. `itsdangerous.py` itself is using kwargs correctly everywhere so this shouldn't be an issue.

For the mixin, the change ensures it will pass additional arguments correctly.
  • Loading branch information
ambv authored and untitaker committed Jan 6, 2017
1 parent f1480f3 commit b00d8fd
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions itsdangerous.py
Original file line number Diff line number Diff line change
Expand Up @@ -700,7 +700,7 @@ def __init__(
self.algorithm_name = algorithm_name
self.algorithm = self.make_algorithm(algorithm_name)

def load_payload(self, payload, return_header=False):
def load_payload(self, payload, serializer=None, return_header=False):
payload = want_bytes(payload)
if b'.' not in payload:
raise BadPayload('No "." found in value')
Expand All @@ -724,7 +724,7 @@ def load_payload(self, payload, return_header=False):
if not isinstance(header, dict):
raise BadHeader('Header payload is not a JSON object',
header=header)
payload = Serializer.load_payload(self, json_payload)
payload = Serializer.load_payload(self, json_payload, serializer=serializer)
if return_header:
return payload, header
return payload
Expand Down Expand Up @@ -847,7 +847,7 @@ class URLSafeSerializerMixin(object):
the string so that it can safely be placed in a URL.
"""

def load_payload(self, payload):
def load_payload(self, payload, *args, **kwargs):
decompress = False
if payload.startswith(b'.'):
payload = payload[1:]
Expand All @@ -863,7 +863,7 @@ def load_payload(self, payload):
except Exception as e:
raise BadPayload('Could not zlib decompress the payload before '
'decoding the payload', original_error=e)
return super(URLSafeSerializerMixin, self).load_payload(json)
return super(URLSafeSerializerMixin, self).load_payload(json, *args, **kwargs)

def dump_payload(self, obj):
json = super(URLSafeSerializerMixin, self).dump_payload(obj)
Expand Down

0 comments on commit b00d8fd

Please sign in to comment.