/
groupInvite.py
77 lines (63 loc) · 2.95 KB
/
groupInvite.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
import webapp2
from google.appengine.ext import ndb
import os
import urllib
import jinja2
import hashlib
import logging
from services import tlguser_service
from services import group_service
JINJA_ENVIRONMENT = jinja2.Environment(
loader=jinja2.FileSystemLoader(os.path.dirname(__file__)),
extensions=['jinja2.ext.autoescape'])
class InvitePage(webapp2.RequestHandler):
def get(self):
invite = group_service.getInvite(self.request.get('inviteKey'))
if invite:
tlguser = tlguser_service.getUserByEmail(invite.email)
if tlguser:
#make member
group = invite.group.get()
member = group_service.addMemberFromObjects(tlguser, group)
#delete invite
invite.key.delete()
#confirmation page
template_values = {'invite':invite, 'user':tlguser, 'host_url':self.request.host_url}
template = JINJA_ENVIRONMENT.get_template('inviteConfirm.html')
self.response.write(template.render(template_values))
else:
#signup page
template_values = {'invite':invite}
template = JINJA_ENVIRONMENT.get_template('inviteSignup.html')
self.response.write(template.render(template_values))
else:
logging.debug("no invite")
class InviteSignup(webapp2.RequestHandler):
def post(self):
invite = group_service.getInvite(self.request.get('inviteKey'))
if invite:
if self.request.get('password') == self.request.get('confirm'):
#make new user
p = hashlib.md5()
p.update(self.request.get('password'))
tlguser = tlguser_service.addUser(invite.email, p.hexdigest(), self.request.get('name'), self.request.host_url)
#make member
group = invite.group.get()
member = group_service.addMemberFromObjects(tlguser, group)
#delete invite
invite.key.delete()
#confirmation page
template_values = {'invite':invite, 'user':tlguser, 'host_url':self.request.host_url}
template = JINJA_ENVIRONMENT.get_template('inviteConfirm.html')
self.response.write(template.render(template_values))
return
else:
template_values = {'invite':invite,
'message': 'Please check your passwords match'
}
template = JINJA_ENVIRONMENT.get_template('inviteSignup.html')
self.response.write(template.render(template_values))
app = webapp2.WSGIApplication([
('/groupInvite/invitePage', InvitePage),
('/groupInvite/inviteSignup', InviteSignup)
], debug=True)