import requests_oauthlib (a dependency)

[yelp-api-cdsw] / requests_oauthlib / compliance_fixes / facebook.py

diff --git a/requests_oauthlib/compliance_fixes/facebook.py b/requests_oauthlib/compliance_fixes/facebook.py
new file mode 100644 (file)
index 0000000..c3c3012
--- /dev/null
+++ b/requests_oauthlib/compliance_fixes/facebook.py
@@ -0,0 +1,17 @@
+from json import dumps
+from oauthlib.common import urldecode
+
+
+def facebook_compliance_fix(session):
+
+    def _compliance_fix(r):
+        token = dict(urldecode(r.text))
+        expires = token.get('expires')
+        if expires is not None:
+            token['expires_in'] = expires
+        token['token_type'] = 'Bearer'
+        r._content = dumps(token)
+        return r
+
+    session.register_compliance_hook('access_token_response', _compliance_fix)
+    return session