You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
761 B
28 lines
761 B
6 years ago
|
from json import dumps, loads
|
||
|
|
import re
|
||
|
|
|
||
|
|
from oauthlib.common import to_unicode
|
||
|
|
|
||
|
|
|
||
|
|
def plentymarkets_compliance_fix(session):
|
||
|
|
|
||
|
|
def _to_snake_case(n):
|
||
|
|
return re.sub('(.)([A-Z][a-z]+)', r'\1_\2', n).lower()
|
||
|
|
|
||
|
|
def _compliance_fix(r):
|
||
|
|
# Plenty returns the Token in CamelCase instead of _
|
||
|
|
if 'application/json' in r.headers.get('content-type', {}) and r.status_code == 200:
|
||
|
|
token = loads(r.text)
|
||
|
|
else:
|
||
|
|
return r
|
||
|
|
|
||
|
|
fixed_token = {}
|
||
|
|
for k, v in token.items():
|
||
|
|
fixed_token[_to_snake_case(k)] = v
|
||
|
|
|
||
|
|
r._content = to_unicode(dumps(fixed_token)).encode('UTF-8')
|
||
|
|
return r
|
||
|
|
|
||
|
|
session.register_compliance_hook('access_token_response', _compliance_fix)
|
||
|
|
return session
|