Source code for oauth_dropins.twitter_auth
"""Utility functions for generating Twitter OAuth headers and making API calls.
This is a separate module from twitter.py so that projects like granary can use
it without pulling in App Engine dependencies.
Supports Python 3. Should not depend on App Engine API or SDK packages.
"""
from __future__ import absolute_import
from future import standard_library
standard_library.install_aliases()
import urllib.error, urllib.parse, urllib.request
from . import appengine_config
import requests
import requests_oauthlib
import tweepy
from .webutil import util
[docs]def signed_urlopen(url, token_key, token_secret, headers=None, **kwargs):
"""Wraps urllib2.urlopen() and adds an OAuth signature.
"""
if headers is None:
headers = {}
# if this is a post, move the body params into the URL. Tweepy's OAuth
# signing doesn't work if they're in the body; Twitter returns a 401.
data = kwargs.get('data')
if data:
method = 'POST'
url += ('&' if '?' in url else '?') + data
kwargs['data'] = ''
else:
method = 'GET'
headers.update(auth_header(url, token_key, token_secret, method=method))
try:
return util.urlopen(urllib.request.Request(url, headers=headers, **kwargs))
except BaseException as e:
util.interpret_http_exception(e)
raise
[docs]def tweepy_auth(token_key, token_secret):
"""Returns a tweepy.OAuthHandler.
"""
assert (appengine_config.TWITTER_APP_KEY and
appengine_config.TWITTER_APP_SECRET), (
"Please fill in the twitter_app_key and twitter_app_secret files in "
"your app's root directory.")
handler = tweepy.OAuthHandler(appengine_config.TWITTER_APP_KEY,
appengine_config.TWITTER_APP_SECRET)
handler.set_access_token(token_key, token_secret)
return handler