Code source de pywws.TwitterAuth

#!/usr/bin/env python

# pywws - Python software for USB Wireless Weather Stations
# http://github.com/jim-easterbrook/pywws
# Copyright (C) 2008-13  Jim Easterbrook  jim@jim-easterbrook.me.uk

# This program is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

"""Authorise pywws to post to your Twitter account
::

%s

This program authorises :py:mod:`pywws.ToTwitter` to post to a Twitter
account. You need to create an account before running
:py:mod:`TwitterAuth`. It opens a web browser window (or gives you a
URL to copy to your web browser) where you log in to your Twitter
account. If the login is successful the browser will display a 7 digit
number which you then copy to :py:mod:`TwitterAuth`.

See :doc:`../guides/twitter` for more detail on using Twitter with
pywws.

"""

__docformat__ = "restructuredtext en"
__usage__ = """
 usage: python -m pywws.TwitterAuth [options] data_dir
 options are:
  -h or --help       display this help
 data_dir is the root directory of the weather data
"""
__doc__ %= __usage__
__usage__ = __doc__.split('\n')[0] + __usage__

import getopt
import oauth2 as oauth
import sys
import urlparse
import webbrowser

from pywws import DataStore
from pywws.ToTwitter import consumer_key, consumer_secret

[docs]def TwitterAuth(params): consumer = oauth.Consumer(consumer_key, consumer_secret) client = oauth.Client(consumer) # step 1 - obtain a request token resp, content = client.request( 'https://api.twitter.com/oauth/request_token', 'POST') if resp['status'] != '200': print 'Failed to get request token. [%s]' % resp['status'] return 1 request_token = dict(urlparse.parse_qsl(content)) # step 2 - redirect the user redirect_url = 'https://api.twitter.com/oauth/authorize?oauth_token=%s' % ( request_token['oauth_token']) if not webbrowser.open(redirect_url, new=2, autoraise=0): print 'Please use a web browser to open the following URL' print redirect_url pin = raw_input('Please enter the PIN shown in your web browser: ').strip() # step 3 - convert the request token to an access token token = oauth.Token( request_token['oauth_token'], request_token['oauth_token_secret']) token.set_verifier(pin) client = oauth.Client(consumer, token) resp, content = client.request( 'https://api.twitter.com/oauth/access_token', 'POST') if resp['status'] != '200': print 'Failed to get access token. [%s]' % resp['status'] return 1 access_token = dict(urlparse.parse_qsl(content)) params.set('twitter', 'key', access_token['oauth_token']) params.set('twitter', 'secret', access_token['oauth_token_secret']) print 'Success! Authorisation data has been stored in %s' % params._path return 0
[docs]def main(argv=None): if argv is None: argv = sys.argv try: opts, args = getopt.getopt(argv[1:], "h", ['help']) except getopt.error, msg: print >>sys.stderr, 'Error: %s\n' % msg print >>sys.stderr, __usage__.strip() return 1 # process options for o, a in opts: if o in ('-h', '--help'): print __usage__.strip() return 0 # check arguments if len(args) != 1: print >>sys.stderr, "Error: 1 argument required" print >>sys.stderr, __usage__.strip() return 2 return TwitterAuth(DataStore.params(args[0]))
if __name__ == "__main__": sys.exit(main())