This repository has been archived by the owner on Apr 24, 2021. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 4
/
main.py
121 lines (99 loc) · 4.14 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
117
118
119
120
121
# Copyright 2015 Google Inc. All rights reserved.
#
# 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.
# [START all]
import cgi
import urllib
import webapp2
from google.appengine.ext import ndb
from google.appengine.api import images
from google.appengine.api import users
# [START model]
class Greeting(ndb.Model):
"""Models a Guestbook entry with an author, content, avatar, and date."""
author = ndb.StringProperty()
content = ndb.TextProperty()
avatar = ndb.BlobProperty()
date = ndb.DateTimeProperty(auto_now_add=True)
# [END model]
def guestbook_key(guestbook_name=None):
"""Constructs a Datastore key for a Guestbook entity with guestbook_name."""
return ndb.Key('Guestbook', guestbook_name or 'default_guestbook')
class MainPage(webapp2.RequestHandler):
def get(self):
self.response.out.write('<html><body>')
guestbook_name = self.request.get('guestbook_name')
greetings = Greeting.query(
ancestor=guestbook_key(guestbook_name)) \
.order(-Greeting.date) \
.fetch(10)
for greeting in greetings:
if greeting.author:
self.response.out.write(
'<b>%s</b> wrote:' % greeting.author)
else:
self.response.out.write('An anonymous person wrote:')
# [START display_image]
self.response.out.write('<div><img src="/img?img_id=%s"></img>' %
greeting.key.urlsafe())
self.response.out.write('<blockquote>%s</blockquote></div>' %
cgi.escape(greeting.content))
# [END display_image]
# [START form]
self.response.out.write("""
<form action="/sign?%s" enctype="multipart/form-data" method="post">
<div><textarea name="content" rows="3" cols="60"></textarea></div>
<div><label>Avatar:</label></div>
<div><input type="file" name="img"/></div>
<div><input type="submit" value="Sign Guestbook"></div>
</form>
<hr>
<form>Guestbook name: <input value="%s" name="guestbook_name">
<input type="submit" value="switch"></form>
</body>
</html>""" % (urllib.urlencode({'guestbook_name': guestbook_name}),
cgi.escape(guestbook_name)))
# [END form]
# [START image_handler]
class Image(webapp2.RequestHandler):
def get(self):
greeting_key = ndb.Key(urlsafe=self.request.get('img_id'))
greeting = greeting_key.get()
if greeting.avatar:
self.response.headers['Content-Type'] = 'image/png'
self.response.out.write(greeting.avatar)
else:
self.response.out.write('No image')
# [END image_handler]
# [START sign_handler]
class Guestbook(webapp2.RequestHandler):
def post(self):
guestbook_name = self.request.get('guestbook_name')
greeting = Greeting(parent=guestbook_key(guestbook_name))
if users.get_current_user():
greeting.author = users.get_current_user().nickname()
greeting.content = self.request.get('content')
# Get image data
avatar = self.request.get('img')
# Transform the image
avatar = images.resize(avatar, 32, 32)
greeting.avatar = avatar
greeting.put()
self.redirect('/?' + urllib.urlencode(
{'guestbook_name': guestbook_name}))
# [END sign_handler]
app = webapp2.WSGIApplication([('/', MainPage),
('/img', Image),
('/sign', Guestbook)],
debug=True)
# [END all]