-
Notifications
You must be signed in to change notification settings - Fork 0
/
views.py
38 lines (29 loc) · 993 Bytes
/
views.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
import webapp2
from google.appengine.api import users
from models import Poll
from utils import render
class MainPage(webapp2.RequestHandler):
def get(self):
user = users.get_current_user()
if user:
polls = Poll.all()
self.response.write(render('index.html', {'polls': polls}))
else:
self.redirect(users.create_login_url(self.request.uri))
class AddPoll(webapp2.RequestHandler):
def post(self):
question = self.request.get('question')
poll = Poll()
poll.question = question
if users.get_current_user():
poll.author = users.get_current_user()
poll.put()
self.redirect('/')
class DetailPoll(webapp2.RequestHandler):
def get(self, key):
user = users.get_current_user()
if user:
poll = Poll.get(key)
self.response.write(poll.question)
else:
self.redirect(users.create_login_url(self.request.uri))