forked from hackerdojo/hd-vote
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
183 lines (154 loc) · 6.03 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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
import cgi
from google.appengine.ext import webapp, db
from google.appengine.ext.webapp import util, template
from google.appengine.api import urlfetch, memcache, users, mail
import logging, urllib, os, random
from datetime import datetime, timedelta
from models import Issue, Choice, Vote
class MainPage(webapp.RequestHandler):
def get(self):
user = users.get_current_user()
if user:
logout_url = users.create_logout_url('/')
else:
login_url = users.create_login_url('/')
issues = Issue.all().order('-creation_date').filter('visibility',"public").fetch(30)
success_type = self.request.get('success')
success_msg = None
if success_type == 'vote':
success_msg = 'Your vote was successfully cast!'
if success_type == 'updated':
success_msg = 'Your vote was successfully updated!'
#created_by = Issue.issues_created_by(member=user,limit=20)
#voted_on = Issue.issues_voted_on(member=user,limit=20)
#recent_results = [issue for issue in voted_on if issue.has_results]
#recent_voted = [issue for issue in voted_on if issue.is_active()]
#recent_results = Issue.recent_results(limit=20)
self.response.out.write(template.render('templates/overview.html', locals()))
class NewHandler(webapp.RequestHandler):
def get(self):
user = users.get_current_user()
if user:
logout_url = users.create_logout_url('/')
else:
self.redirect(users.create_login_url(self.request.uri))
return
option_one = "Yes"
option_two = "No"
self.response.out.write(template.render('templates/new.html', locals()))
def post(self):
user = users.get_current_user()
if not user:
self.redirect(users.create_login_url(self.request.uri))
return
duration_amount = int(self.request.get('duration_amount'))
multiplier = int(self.request.get('duration_multiplier'))
visibility = self.request.get('visibility')
hashcode = random_string()
issue = Issue(
visibility = visibility,
title = cgi.escape(self.request.get('title')),
description = cgi.escape(self.request.get('description')),
duration = duration_amount * multiplier,
urlcode = hashcode)
issue.put()
if self.request.get('option1'):
issue.add_choice(cgi.escape(self.request.get('option1')))
if self.request.get('option2'):
issue.add_choice(cgi.escape(self.request.get('option2')))
if self.request.get('option3'):
issue.add_choice(cgi.escape(self.request.get('option3')))
if self.request.get('option4'):
issue.add_choice(cgi.escape(self.request.get('option4')))
if self.request.get('option5'):
issue.add_choice(cgi.escape(self.request.get('option5')))
self.redirect('/issue/%s' % issue.urlcode)
class EditHandler(webapp.RequestHandler):
def get(self,urlcode):
user = users.get_current_user()
if user:
logout_url = users.create_logout_url('/')
else:
self.redirect(users.create_login_url(self.request.uri))
return
issue = Issue.get_issue_by_urlcode(urlcode)
choices = issue.choices
self.response.out.write(template.render('templates/edit.html', locals()))
def post(self,urlcode):
user = users.get_current_user()
if user:
logout_url = users.create_logout_url('/')
else:
self.redirect(users.create_login_url(self.request.uri))
return
issue = Issue.get_issue_by_urlcode(urlcode)
if self.request.get('extend'):#if extending vote
choices = issue.choices
extend_amount = int(self.request.get('extend_amount')) * int(self.request.get('extend_multiplier'))
issue.extend_duration(extend_amount)
self.response.out.write(template.render('templates/edit.html', locals()))
else:#otherwise we are saving changes
if issue.vote_count():
raise Exception('Unable to change issue text once votes have been cast')
duration_amount = int(self.request.get('duration_amount'))
multiplier = int(self.request.get('duration_multiplier'))
issue.duration = duration_amount * multiplier
if self.request.get('title'):
issue.title = cgi.escape(self.request.get('title'))
if self.request.get('description'):
issue.description = cgi.escape(self.request.get('description'))
if self.request.get('option1') and self.request.get('option2'):
choices = issue.choices
db.delete(choices)
issue.add_choice(cgi.escape(self.request.get('option1')))
issue.add_choice(cgi.escape(self.request.get('option2')))
if self.request.get('option3'):
issue.add_choice(cgi.escape(self.request.get('option3')))
if self.request.get('option4'):
issue.add_choice(cgi.escape(self.request.get('option4')))
if self.request.get('option5'):
issue.add_choice(cgi.escape(self.request.get('option5')))
issue.put()
#choices = issue.choices
self.redirect('/issue/%s' % issue.urlcode)
#self.response.out.write(template.render('templates/edit.html', locals()))
class IssueHandler(webapp.RequestHandler):
def get(self,urlcode):
user = users.get_current_user()
if user:
logout_url = users.create_logout_url('/')
else:
self.redirect(users.create_login_url(self.request.uri))
return
issue = Issue.get_issue_by_urlcode(urlcode)
issue.update_status()
#vote = issue.vote_for_member(user)
issueUrl = self.request.uri
self.response.out.write(template.render('templates/issue.html', locals()))
def post(self,urlcode):
user = users.get_current_user()
if user:
logout_url = users.create_logout_url('/')
else:
self.redirect(users.create_login_url(self.request.uri))
issue = Issue.get_issue_by_urlcode(urlcode)
#vote = issue.vote_for_member()
new_choice = Choice.get_by_id(int(self.request.get('choice')))
was_updated = issue.register_vote(new_choice)
if was_updated:
self.redirect('/?success=updated')
else:
self.redirect('/?success=vote')
def random_string():
hashbase = '1234567890abcdefghijklmnopqrstuvwxyz'
return ''.join(random.sample(hashbase,len(hashbase)))
def main():
application = webapp.WSGIApplication([
('/',MainPage),
('/new',NewHandler),
('/issue/(\w+).*',IssueHandler),
('/edit/(\w+).*',EditHandler)],
debug=True)
util.run_wsgi_app(application)
if __name__ == '__main__':
main()