add a pre_request hook for weibo

Author: lepture

example/weibo.py

@@ -26,10 +26,7 @@
 def index():
     if 'oauth_token' in session:
         access_token = session['oauth_token'][0]
-        # weibo is a shit !!!! It cannot be authorized by Bearer Token.
-        resp = weibo.get('statuses/home_timeline.json', data={
-            'access_token': access_token
-        })
+        resp = weibo.get('statuses/home_timeline.json')
         return jsonify(resp.data)
     return redirect(url_for('login'))
 
@@ -64,5 +61,17 @@ def get_weibo_oauth_token():
     return session.get('oauth_token')
 
 
+def change_weibo_header(uri, headers, body):
+    """Since weibo is a rubbish server, it does not follow the standard,
+    we need to change the authorization header for it."""
+    auth = headers.get('Authorization')
+    if auth:
+        auth = auth.replace('Bearer', 'OAuth2')
+        headers['Authorization'] = auth
+    return uri, headers, body
+
+weibo.pre_request = change_weibo_header
+
+
 if __name__ == '__main__':
     app.run()

flask_oauthlib/client.py

@@ -321,6 +321,13 @@ def request(self, url, data=None, headers=None, format='urlencoded',
             uri, headers, body = client.add_token(
                 url, http_method=method, body=data, headers=headers
             )
+
+        if hasattr(self, 'pre_request'):
+            # this is desgined for some rubbish serice like weibo
+            # since they don't follow the standards, we need to
+            # change the uri, headers, or body
+            uri, headers, body = self.pre_request(uri, headers, body)
+
         resp, content = make_request(
             uri, headers, data=body, method=method
         )