def do_POST(self):
try:
self.send_response(301)
self.send_header('Content-type', 'text/html')
self.end_headers()
ctype, pdict = cgi.parse_header(
self.headers.getheader('content-type'))
if ctype == 'multipart/form-data':
fields = cgi.parse_multipart(self.rfile, pdict)
messagecontent = fields.get('restaurant_name')
restaurant_edit = fields.get('name_to_be_edited')
restaurant_delete = fields.get('name_to_be_deleted')
name = messagecontent[0]
if restaurant_delete is not None:
name_delete = restaurant_delete[0]
print "++++++++++++"
print name_delete
print "++++++++++++"
if name_delete:
print "deleting section"
restaurant_delete_query = session.query(
Restaurant).filter_by(name='%s' %
name_delete).all()[0]
session.delete(restaurant_delete_query)
session.commit()
if restaurant_edit is not None:
name2 = restaurant_edit[0]
restaurant = session.query(Restaurant).filter_by(
name='%s' % name2).all()
newrestaurant = restaurant[0]
else:
newrestaurant = Restaurant()
newrestaurant.id = random.randint(1, 10000)
print newrestaurant.name
print newrestaurant.id
print messagecontent
newrestaurant.name = messagecontent[0]
session.add(newrestaurant)
session.commit()
except:
print(traceback.print_exc())
