YHandler/YHandler.py at master · timmsc/YHandler · GitHub

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
import requests
from rauth import OAuth1Service
from requests import request
import webbrowser
import csv
import os.path
import pickle as pk

GET_TOKEN_URL = 'https://api.login.yahoo.com/oauth/v2/get_token'
AUTHORIZATION_URL = 'https://api.login.yahoo.com/oauth/v2/request_auth'
REQUEST_TOKEN_URL = 'https://api.login.yahoo.com/oauth/v2/get_request_token'
CALLBACK_URL = 'oob'

sessionf = 'yahoo_session.txt'

class YHandler(object):

	def __init__(self, authf):
		self.authf = authf
		self.authd = self.get_authvals_csv(self.authf)
		self.session = self.deserialize_session()
		#self.session = None
		# Get a real consumer key & secret
		self.yahoo = OAuth1Service(
			name='yahoo',
			consumer_key=self.authd['consumer_key'],
			consumer_secret=self.authd['consumer_secret'],
			request_token_url=REQUEST_TOKEN_URL,
			access_token_url=GET_TOKEN_URL,
			authorize_url=AUTHORIZATION_URL,
			#session_obj=self.session,
			base_url='https://api.login.yahoo.com/oauth/v2/')

	def get_authvals_csv(self, authf):
		vals = {}	#dict of vals to be returned
		with open(authf, 'rb') as f:
			f_iter = csv.DictReader(f)
			vals = f_iter.next()
		return vals


	def write_authvals_csv(self, authd, authf):
		f = open(authf, 'wb')
		fieldnames = tuple(authd.iterkeys())
		headers = dict((n,n) for n in fieldnames)
		f_iter = csv.DictWriter(f, fieldnames=fieldnames)
		f_iter.writerow(headers)
		f_iter.writerow(authd)
		f.close

	def serialize_session(self, session):
		return pk.dump(session, open('session.dat', 'w+'))

	def deserialize_session(self):
        	try:
	           with open('session.dat'): pass
        	except IOError:
	           print 'Oh dear. Couldn\'t open session file'
		   return None
		return pk.load(open('session.dat'))

	def new_session(self):
		request_token, request_token_secret = self.yahoo.get_request_token(data = { 'oauth_callback': CALLBACK_URL })
		auth_url = self.yahoo.get_authorize_url(request_token)
		print 'Visit this URL in your browser: ' + auth_url
		pin = raw_input('Enter PIN from browser: ')
		# need to get &oauth_session_handle=AKVdNElJthnrHDwnYDuj6fJ2ayRbJvkePz9AKwi9dQAfb4bd here; not sure how.  This is need in refresh call to reuse session after it has expired.
		session = self.yahoo.get_auth_session(request_token, request_token_secret, method='POST', data={'oauth_verifier': pin})
		f = open(sessionf, 'wb')
		f.write(session.access_token + '\n')
		f.write(session.access_token_secret + '\n')
		self.serialize_session(session)
		return session

	def reuse_session(self):
		f = open(sessionf, 'rb')
                access_token = f.readline().rstrip('\n')
                access_token_secret = f.readline().rstrip('\n')
		print 'access_token: ' + access_token
		print 'access_token_secret: ' + access_token_secret
		#print 'self.yahoo.signature_obj: ' + self.yahoo.signature_obj
		session = self.yahoo.get_session((access_token, access_token_secret), self.yahoo.signature_obj)
		return session

	def api_req(self, querystring, req_meth='GET', data=None, headers=None):
		base_url = 'http://fantasysports.yahooapis.com/fantasy/v2/'
		url = base_url + querystring
		if (self.session is None and not os.path.exists(sessionf)):
			self.session = self.new_session()
		else:
			self.session = self.reuse_session()
		print "calling url: ", url
		r=self.session.get(url)

		return r