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
|
import web
from social.actions import do_auth, do_complete, do_disconnect
from social.apps.webpy_app.utils import psa, load_strategy
urls = (
'/login/(?P<backend>[^/]+)/?', 'auth',
'/complete/(?P<backend>[^/]+)/?', 'complete',
'/disconnect/(?P<backend>[^/]+)/?', 'disconnect',
'/disconnect/(?P<backend>[^/]+)/(?P<association_id>\d+)/?', 'disconnect',
)
class BaseViewClass(object):
def __init__(self, *args, **kwargs):
self.session = web.web_session
method = web.ctx.method == 'POST' and 'post' or 'get'
self.strategy = load_strategy()
self.data = web.input(_method=method)
super(BaseViewClass, self).__init__(*args, **kwargs)
def get_current_user(self):
if not hasattr(self, '_user'):
if self.session.get('logged_in'):
self._user = self.strategy.get_user(
self.session.get('user_id')
)
else:
self._user = None
return self._user
def login_user(self, user):
self.session['logged_in'] = True
self.session['user_id'] = user.id
class auth(BaseViewClass):
def GET(self, backend):
return self._auth(backend)
def POST(self, backend):
return self._auth(backend)
@psa('/complete/%(backend)s/')
def _auth(self, backend):
return do_auth(self.backend)
class complete(BaseViewClass):
def GET(self, backend, *args, **kwargs):
return self._complete(backend, *args, **kwargs)
def POST(self, backend, *args, **kwargs):
return self._complete(backend, *args, **kwargs)
@psa('/complete/%(backend)s/')
def _complete(self, backend, *args, **kwargs):
return do_complete(
self.backend,
login=lambda backend, user, social_user: self.login_user(user),
user=self.get_current_user(), *args, **kwargs
)
class disconnect(BaseViewClass):
@psa()
def POST(self, backend, association_id=None):
return do_disconnect(self.backend, self.get_current_user(),
association_id)
app_social = web.application(urls, locals())
|
