mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2024-11-14 17:40:24 -08:00
0acf78f196
Added migration code to migrate SickBeard section with fork sickrage-api to new SiCKRAGE section
30 lines
796 B
Python
30 lines
796 B
Python
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
|