facebook.py 995 B

123456789101112131415161718192021222324252627
  1. from json import dumps
  2. from urllib.parse import parse_qsl
  3. def facebook_compliance_fix(session):
  4. def _compliance_fix(r):
  5. # if Facebook claims to be sending us json, let's trust them.
  6. if "application/json" in r.headers.get("content-type", {}):
  7. return r
  8. # Facebook returns a content-type of text/plain when sending their
  9. # x-www-form-urlencoded responses, along with a 200. If not, let's
  10. # assume we're getting JSON and bail on the fix.
  11. if "text/plain" in r.headers.get("content-type", {}) and r.status_code == 200:
  12. token = dict(parse_qsl(r.text, keep_blank_values=True))
  13. else:
  14. return r
  15. expires = token.get("expires")
  16. if expires is not None:
  17. token["expires_in"] = expires
  18. token["token_type"] = "Bearer"
  19. r._content = dumps(token).encode()
  20. return r
  21. session.register_compliance_hook("access_token_response", _compliance_fix)
  22. return session