/
main.py
executable file
·116 lines (96 loc) · 4 KB
/
main.py
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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import cgi
import wsgiref.handlers
import logging
import os
from google.appengine.api import users
from google.appengine.ext import webapp
from google.appengine.api import memcache
import gdata.photos.service
import gdata.media
import gdata.service
import gdata.alt.appengine
import atom
import atom.http_interface
import atom.token_store
import atom.url
import settings
# メインページ
#
# 指定されたユーザーの画像を Picasa から取ってきて表示します。
# 有効なセッショントークンを持っている場合は非公開に設定している画像も表示されます。
#
# テンプレートの使い方がよくわからなかったのでソース中に HTML をごりごり書いてます…
#
class MainPage(webapp.RequestHandler):
def get(self):
html = '<div style="text-align: right;">%s</div>' % render_auth_link()
if self.request.get('user_id'):
gd_client = get_gdata_client()
session_token = memcache.get('token')
if session_token:
# セッショントークンがあれば client にセット
logging.info("session_token=%s" % session_token)
gd_client.current_token = session_token
user_id = self.request.get('user_id').encode('utf-8')
html += render_user_id_form(user_id)
html += render_photos(gd_client, user_id)
else:
html += render_user_id_form('')
self.response.out.write("<html><body>%s</body></html>" % html)
# セッショントークンを取得するためのコントローラー
#
# Picasa 側で「アクセス許可」の操作を行った後、ここに帰ってきます。
# ワンタイムトークン(AuthSubToken)をセッショントークンにアップグレードします。
#
# このデモではセッショントークンを memcache に入れてますが、
# 実際にはユーザーアカウントと結びつけて DataStore に入れることになると思います。
#
class UpgradeTokenPage(webapp.RequestHandler):
def get(self):
gd_client = get_gdata_client()
auth_sub_token = gdata.auth.extract_auth_sub_token_from_url(self.request.uri)
logging.info("auth_sub_token=%s" % auth_sub_token)
if auth_sub_token:
session_token = gd_client.upgrade_to_session_token(auth_sub_token)
if session_token:
memcache.add('token', session_token, 3600)
self.response.out.write("<html><body>トークンを取得しました。</body></html>")
def render_auth_link():
gd_client = get_gdata_client()
next_url = atom.url.Url('http', settings.HOST_NAME, path='/upgrade_to_session_token')
return '<a href="%s" target="_brank">トークンを発行する</a>' % gd_client.GenerateAuthSubURL(
next_url, ('http://picasaweb.google.com/data/',), secure=False, session=True)
def get_gdata_client():
gd_client = gdata.photos.service.PhotosService()
gdata.alt.appengine.run_on_appengine(gd_client)
return gd_client
def render_user_id_form(user_id):
form_html = """
<form method="get">
<div>User ID:</div>
<input type="text" name="user_id" value="%s" />
<input type="submit" value="Show" />
</form>
"""
return form_html % user_id
def render_photos(gd_client, user_id):
html = ''
for album in gd_client.GetUserFeed(user=user_id).entry:
html += '<div style="border: dotted 1px gray; margin: 10px; padding: 0 10px;">'
html += '<h2>%s</h2>\n' % album.title.text
html += '<p>\n'
for photo in gd_client.GetFeed('/data/feed/api/user/%s/albumid/%s?kind=photo' % (user_id, album.gphoto_id.text)).entry:
html += '<img src="%s" />\n' % photo.media.thumbnail[2].url
html += '</p>\n</div>\n'
return html
def main():
application = webapp.WSGIApplication([
('/', MainPage),
('/upgrade_to_session_token', UpgradeTokenPage),
],
debug=True)
wsgiref.handlers.CGIHandler().run(application)
if __name__ == "__main__":
main()