codekingpro/portable-devtools

1import json2 3 4def mailchimp_compliance_fix(session):5    def _null_scope(r):6        token = json.loads(r.text)7        if "scope" in token and token["scope"] is None:8            token.pop("scope")9        r._content = json.dumps(token).encode()10        return r11 12    def _non_zero_expiration(r):13        token = json.loads(r.text)14        if "expires_in" in token and token["expires_in"] == 0:15            token["expires_in"] = 360016        r._content = json.dumps(token).encode()17        return r18 19    session.register_compliance_hook("access_token_response", _null_scope)20    session.register_compliance_hook("access_token_response", _non_zero_expiration)21    return session22