-
Notifications
You must be signed in to change notification settings - Fork 0
/
edit_factory.py
147 lines (124 loc) · 5.22 KB
/
edit_factory.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
from google.appengine.api import users
from google.appengine.ext import db, webapp
from google.appengine.ext.webapp import template
import bene_query
import bene_util
import os
import urllib
"""
Creates a form for Producers to enter information
about a Factory
"""
class EditFactoryPage(webapp.RequestHandler):
def get(self):
user = users.get_current_user()
if not user: # need to sign in
self.redirect('/?signin=True')
return
if bene_query.getCurrentUser().isConsumer: # consumers can't access this
self.redirect('/')
return
_producer = bene_query.getCurrentProducer()
if _producer == None: # no producer signed up, so ask to sign up
self.redirect('/')
return
if not _producer.verified: # if producer is not verified
self.redirect('/producerhome?%s' % urllib.urlencode({'verify': True}))
return
ID = self.request.get('id')
if not ID:
'''
TODO: If no ID sent, default to page with all factories?
'''
self.redirect('/')
return
_factory = db.get(ID)
''' an error in getting the factory will be redirected to exception handler'''
if _factory.owner != user: # if not owner of factory
self.redirect('/producerhome?%s' % urllib.urlencode({'not_owner': True}))
return
template_values = bene_util.initTemplate(self.request.uri)
template_values['id'] = ID
template_values['name_old'] = _factory.name
template_values['address_old'] = _factory.address
template_values['unique_old'] = _factory.unique
path = os.path.join(os.path.dirname(__file__), 'editfactory.html')
self.response.out.write(template.render(path, template_values))
return
'''
Exception handler
'''
def handle_exception(self, exception, debug_mode):
if debug_mode:
super(EditFactoryPage, self).handle_exception(exception, debug_mode)
else:
template_values = bene_util.initTemplate(self.request.uri)
path = os.path.join(os.path.dirname(__file__), 'not_found.html')
self.response.out.write(template.render(path, template_values))
return
"""
Page that stores Factory in datastore
"""
class StoreEditedFactoryPage(webapp.RequestHandler):
def post(self):
user = users.get_current_user()
if not user: # need to sign in
self.redirect('/?signin=True')
return
if bene_query.getCurrentUser().isConsumer: # consumers can't access this
self.redirect('/')
return
_producer = bene_query.getCurrentProducer()
if _producer == None: # no producer signed up, so ask to sign up
self.redirect('/')
return
if not _producer.verified: # if producer is not verified
self.redirect('/producerhome?%s' % urllib.urlencode({'verify': True}))
return
ID = self.request.get('id')
if not ID:
'''
TODO: If no ID sent, default to page with all factories?
'''
self.redirect('/')
return
_factory = db.get(ID)
''' an error in getting the factory will be redirected to exception handler'''
if _factory.owner != user: # if not owner of factory
self.redirect('/producerhome?%s' % urllib.urlencode({'not_owner': True}))
return
_factory.name = self.request.get('name')
_factory.address = self.request.get('address')
_unique_save = _factory.unique
_factory.unique = self.request.get('unique')
if bene_util.doesFactoryExist(_factory) and _unique_save != _factory.unique : # if factory with same unique exists
self.redirect('/editfactory?%s' % (urllib.urlencode({'id' : ID, 'repeatedit' : True})))
return
_picture = self.request.get('picture')
_factory.addPicture(_picture)
_location = self.request.get('location')
fields = _location.split(',')
if len(fields) == 2:
try:
lat = float(fields[0])
lon = float(fields[1])
gp = db.GeoPt(lat, lon)
except ValueError:
gp = None
else:
gp = None
_factory.location = gp
_factory.put()
self.redirect('/viewfactory?%s' % urllib.urlencode({'id': ID}))
return
'''
Exception handler
'''
def handle_exception(self, exception, debug_mode):
if debug_mode:
super(StoreEditedFactoryPage, self).handle_exception(exception, debug_mode)
else:
template_values = bene_util.initTemplate(self.request.uri)
path = os.path.join(os.path.dirname(__file__), 'not_found.html')
self.response.out.write(template.render(path, template_values))
return