Skip to content

Commit

Permalink
Add function to get user_id of authenticating user
Browse files Browse the repository at this point in the history
  • Loading branch information
parantapa committed Jun 17, 2016
1 parent cd46550 commit b10bef7
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions tweepy/auth.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,6 +44,7 @@ def __init__(self, consumer_key, consumer_secret, callback=None):
self.access_token_secret = None
self.callback = callback
self.username = None
self.user_id = None
self.oauth = OAuth1Session(consumer_key,
client_secret=consumer_secret,
callback_uri=self.callback)
Expand Down Expand Up @@ -102,6 +103,8 @@ def get_access_token(self, verifier=None):
resp = self.oauth.fetch_access_token(url)
self.access_token = resp['oauth_token']
self.access_token_secret = resp['oauth_token_secret']
self.username = resp["screen_name"]
self.user_id = resp["user_id"]
return self.access_token, self.access_token_secret
except Exception as e:
raise TweepError(e)
Expand Down Expand Up @@ -134,11 +137,23 @@ def get_username(self):
user = api.verify_credentials()
if user:
self.username = user.screen_name
self.user_id = user.id
else:
raise TweepError('Unable to get username,'
' invalid oauth token!')
return self.username

def get_user_id(self):
if self.user_id is None:
api = API(self)
user = api.verify_credentials()
if user:
self.username = user.screen_name
self.user_id = user.id
else:
raise TweepError('Unable to get user_id,'
' invalid oauth token!')
return self.user_id

class OAuth2Bearer(AuthBase):
def __init__(self, bearer_token):
Expand Down

0 comments on commit b10bef7

Please sign in to comment.