1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
#!/usr/bin/python
#
# Copyright (C) 2007 Google Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
__author__ = 'kunalmshah.userid (Kunal Shah)'
import sys
import os.path
import getopt
import gdata.auth
import gdata.docs.service
class OAuthSample(object):
"""Sample class demonstrating the three-legged OAuth process."""
def __init__(self, consumer_key, consumer_secret):
"""Constructor for the OAuthSample object.
Takes a consumer key and consumer secret, store them in class variables,
creates a DocsService client to be used to make calls to
the Documents List Data API.
Args:
consumer_key: string Domain identifying third_party web application.
consumer_secret: string Secret generated during registration.
"""
self.consumer_key = consumer_key
self.consumer_secret = consumer_secret
self.gd_client = gdata.docs.service.DocsService()
def _PrintFeed(self, feed):
"""Prints out the contents of a feed to the console.
Args:
feed: A gdata.docs.DocumentListFeed instance.
"""
if not feed.entry:
print 'No entries in feed.\n'
docs_list = list(enumerate(feed.entry, start = 1))
for i, entry in docs_list:
print '%d. %s\n' % (i, entry.title.text.encode('UTF-8'))
def _ListAllDocuments(self):
"""Retrieves a list of all of a user's documents and displays them."""
feed = self.gd_client.GetDocumentListFeed()
self._PrintFeed(feed)
def Run(self):
"""Demonstrates usage of OAuth authentication mode and retrieves a list of
documents using the Document List Data API."""
print '\nSTEP 1: Set OAuth input parameters.'
self.gd_client.SetOAuthInputParameters(
gdata.auth.OAuthSignatureMethod.HMAC_SHA1,
self.consumer_key, consumer_secret=self.consumer_secret)
print '\nSTEP 2: Fetch OAuth Request token.'
request_token = self.gd_client.FetchOAuthRequestToken()
print 'Request Token fetched: %s' % request_token
print '\nSTEP 3: Set the fetched OAuth token.'
self.gd_client.SetOAuthToken(request_token)
print 'OAuth request token set.'
print '\nSTEP 4: Generate OAuth authorization URL.'
auth_url = self.gd_client.GenerateOAuthAuthorizationURL()
print 'Authorization URL: %s' % auth_url
raw_input('Manually go to the above URL and authenticate.'
'Press a key after authorization.')
print '\nSTEP 5: Upgrade to an OAuth access token.'
self.gd_client.UpgradeToOAuthAccessToken()
print 'Access Token: %s' % (
self.gd_client.token_store.find_token(request_token.scopes[0]))
print '\nYour Documents:\n'
self._ListAllDocuments()
print 'STEP 6: Revoke the OAuth access token after use.'
self.gd_client.RevokeOAuthToken()
print 'OAuth access token revoked.'
def main():
"""Demonstrates usage of OAuth authentication mode.
Prints a list of documents. This demo uses HMAC-SHA1 signature method.
"""
# Parse command line options
try:
opts, args = getopt.getopt(sys.argv[1:], '', ['consumer_key=',
'consumer_secret='])
except getopt.error, msg:
print ('python oauth_example.py --consumer_key [consumer_key] '
'--consumer_secret [consumer_secret] ')
sys.exit(2)
consumer_key = ''
consumer_secret = ''
# Process options
for option, arg in opts:
if option == '--consumer_key':
consumer_key = arg
elif option == '--consumer_secret':
consumer_secret = arg
while not consumer_key:
consumer_key = raw_input('Please enter consumer key: ')
while not consumer_secret:
consumer_secret = raw_input('Please enter consumer secret: ')
sample = OAuthSample(consumer_key, consumer_secret)
sample.Run()
if __name__ == '__main__':
main()
|
