projects / z_archive / twitter.git / blame
| Commit | Line | Data |
|---|---|---|
| 1b31d642 | 1 | |
| ddeba164 MV |
2 | import webbrowser |
| 3 | import time | |
| 1b31d642 MV |
4 | |
| 5 | from api import Twitter | |
| 6 | from oauth import OAuth, write_token_file | |
| 7 | ||
| 8 | def oauth_dance(app_name, consumer_key, consumer_secret, token_filename=None): | |
| 9 | """ | |
| 10 | Perform the OAuth dance with some command-line prompts. Return the | |
| 11 | oauth_token and oauth_token_secret. | |
| 12 | ||
| 13 | Provide the name of your app in `app_name`, your consumer_key, and | |
| 14 | consumer_secret. This function will open a web browser to let the | |
| 15 | user Allow your app to access their Twitter account. PIN | |
| 16 | authentication is used. | |
| 17 | ||
| 18 | If a token_filename is given, the oauth tokens will be written to | |
| 19 | the file. | |
| 20 | """ | |
| 21 | print ("Hi there! We're gonna get you all set up to use %s." % app_name) | |
| 22 | twitter = Twitter( | |
| 23 | auth=OAuth('', '', consumer_key, consumer_secret), | |
| 24 | format='') | |
| 25 | oauth_token, oauth_token_secret = parse_oauth_tokens( | |
| 26 | twitter.oauth.request_token()) | |
| 27 | print """ | |
| 28 | In the web browser window that opens please choose to Allow | |
| 29 | access. Copy the PIN number that appears on the next page and paste or | |
| 30 | type it here: | |
| 31 | """ | |
| 32 | webbrowser.open( | |
| 33 | 'http://api.twitter.com/oauth/authorize?oauth_token=' + | |
| 34 | oauth_token) | |
| 35 | time.sleep(2) # Sometimes the last command can print some | |
| 36 | # crap. Wait a bit so it doesn't mess up the next | |
| 37 | # prompt. | |
| 38 | oauth_verifier = raw_input("Please type the PIN: ").strip() | |
| 39 | twitter = Twitter( | |
| 40 | auth=OAuth( | |
| ddeba164 | 41 | oauth_token, oauth_token_secret, consumer_key, consumer_secret), |
| 1b31d642 MV |
42 | format='') |
| 43 | oauth_token, oauth_token_secret = parse_oauth_tokens( | |
| 44 | twitter.oauth.access_token(oauth_verifier=oauth_verifier)) | |
| 45 | if token_filename: | |
| 46 | write_token_file( | |
| 47 | token_filename, oauth_token, oauth_token_secret) | |
| ddeba164 MV |
48 | |
| 49 | print "That's it! Your authorization keys have been written to %s." % ( | |
| 50 | token_filename) | |
| 51 | return oauth_token, oauth_token_secret | |
| 52 | ||
| 53 | def parse_oauth_tokens(result): | |
| 54 | for r in result.split('&'): | |
| 55 | k, v = r.split('=') | |
| 56 | if k == 'oauth_token': | |
| 57 | oauth_token = v | |
| 58 | elif k == 'oauth_token_secret': | |
| 59 | oauth_token_secret = v | |
| 1b31d642 | 60 | return oauth_token, oauth_token_secret |