/
frontend.py
149 lines (124 loc) · 4.27 KB
/
frontend.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
import os
import urllib
import jinja2
import webapp2
import logging
from model import Type,Product
from google.appengine.api import users
JINJA_ENVIRONMENT = jinja2.Environment(
loader=jinja2.FileSystemLoader(os.path.join(os.path.dirname(__file__) + "/html/frontend")),
extensions=['jinja2.ext.autoescape'])
class MainPage(webapp2.RequestHandler):
def get(self):
if users.get_current_user():
url = users.create_logout_url(self.request.uri)
url_linktext = 'Click here to log out'
if users.is_current_user_admin():
self.redirect('/booktype')
else:
url = users.create_login_url(self.request.uri)
url_linktext = 'Click here to log in'
all_book_types = Type.query().order(Type.code).fetch()
template_values = {
'all_book_types': all_book_types,
'url': url,
'url_linktext': url_linktext,
}
template = JINJA_ENVIRONMENT.get_template('/homepage.html')
self.response.write(template.render(template_values))
class RegisterPage(webapp2.RequestHandler):
def get(self):
if users.get_current_user():
url = users.create_logout_url(self.request.uri)
url_linktext = 'Click here to log out'
if users.is_current_user_admin():
self.redirect('/booktype')
else:
url = users.create_login_url(self.request.uri)
url_linktext = 'Click here to log in'
all_book_types = Type.query().order(Type.code).fetch()
template_values = {
'all_book_types': all_book_types,
'url': url,
'url_linktext': url_linktext,
}
template = JINJA_ENVIRONMENT.get_template('/register.html')
self.response.write(template.render(template_values))
class PolicyPage(webapp2.RequestHandler):
def get(self):
if users.get_current_user():
url = users.create_logout_url(self.request.uri)
url_linktext = 'Click here to log out'
if users.is_current_user_admin():
self.redirect('/booktype')
else:
url = users.create_login_url(self.request.uri)
url_linktext = 'Click here to log in'
all_book_types = Type.query().order(Type.code).fetch()
template_values = {
'all_book_types': all_book_types,
'url': url,
'url_linktext': url_linktext,
}
template = JINJA_ENVIRONMENT.get_template('/policy.html')
self.response.write(template.render(template_values))
class BookPage(webapp2.RequestHandler):
def get(self):
if users.get_current_user():
url = users.create_logout_url(self.request.uri)
url_linktext = 'Click here to log out'
if users.is_current_user_admin():
self.redirect('/booktype')
else:
url = users.create_login_url(self.request.uri)
url_linktext = 'Click here to log in'
all_book_types = Type.query().order(Type.code).fetch()
all_book_details = Product.query().order(Product.bookId).fetch()
type_code = self.request.get('code')
if type_code is None:
logging.info('No code supplied')
self.redirect('/')
else:
logging.info('Book type code %s', type_code)
book_type = Type.query(Type.code == type_code).get()
if book_type is None:
logging.info('Cannot find type with code %s', type_code)
self.redirect('/')
template_values = {
'all_book_types': all_book_types,
'book_type': book_type,
'all_book_details': all_book_details,
'url': url,
'url_linktext': url_linktext,
}
template = JINJA_ENVIRONMENT.get_template('/product.html')
self.response.write(template.render(template_values))
class BookInfoPage(webapp2.RequestHandler):
def get(self):
if users.get_current_user():
url = users.create_logout_url(self.request.uri)
url_linktext = 'Click here to log out'
if users.is_current_user_admin():
self.redirect('/booktype')
else:
url = users.create_login_url(self.request.uri)
url_linktext = 'Click here to log in'
all_book_types = Type.query().order(Type.code).fetch()
book_code = self.request.get('code')
if book_code is None:
logging.info('No code supplied')
self.redirect('/')
else:
logging.info('Book type code %s', book_code)
book_details = Product.query(Product.bookId == book_code).get()
if book_details is None:
logging.info('Cannot find type with code %s', book_code)
self.redirect('/')
template_values = {
'all_book_types': all_book_types,
'book_details': book_details,
'url': url,
'url_linktext': url_linktext,
}
template = JINJA_ENVIRONMENT.get_template('/productdetails.html')
self.response.write(template.render(template_values))