From 265c8d5f81b7c82b13385cf46a73bc86b322fe4d Mon Sep 17 00:00:00 2001 From: Parantapa Bhattacharya Date: Fri, 17 Jun 2016 19:06:09 +0530 Subject: [PATCH] Add function to get user_id of authenticating user --- tweepy/auth.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/tweepy/auth.py b/tweepy/auth.py index c157ad8fd..80045f27a 100644 --- a/tweepy/auth.py +++ b/tweepy/auth.py @@ -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) @@ -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) @@ -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):