def get(self): user = users.get_current_user() global sort_dist if (user): u_query = UserModel.query(UserModel.user==user).fetch() if (len(u_query)>0): u = u_query[0] current_location = u.last_location lat = u.last_latitude lg = u.last_longitude url = users.create_logout_url(self.request.url) url_linktext = 'Logout' restaurant_query = RestaurantModel.query().order(-RestaurantModel.TotalScore).fetch() restaurant_info = [] if (len(restaurant_query) > 0): for restaurant in restaurant_query: a = restaurant.latitude l = restaurant.longitude tmp = "/view_picture/%s" % restaurant.Blob_key part = urllib.urlencode({'name':restaurant.name}) tmp1 = "/order?%s" % part score = restaurant.TotalScore restaurant_info.append((restaurant.name, tmp, a, l, tmp1, score)) if (sort_dist): restaurant_info.sort(key=lambda tup: (tup[2]-lat)**2+(tup[3]-lg)**2) # # Initialize a cart for this user # cart_query = CartModel.query(CartModel.user==user).fetch() # if (len(cart_query)<1): # cart = CartModel() # cart.user = user # cart.orders = [] # cart.total = 0.0 # cart.put() # current_location = 'UT-Austin' #get the current location in database template_values = { 'user': user, 'url': url, 'url_linktext': url_linktext, 'current_location': current_location, 'restaurant_query_len': len(restaurant_query), 'restaurant_info': restaurant_info } template = JINJA_ENVIRONMENT.get_template('templates/main.html') self.response.write(template.render(template_values)) else: self.redirect('/location') else: self.redirect('/')
def post(self): u = users.get_current_user() name = self.request.get('location') dis = self.request.get('distance') lat = float(self.request.get('latitude')) lg = float(self.request.get('longitude')) user_query = UserModel.query(UserModel.user==u).fetch() if (len(user_query)>0): user = user_query[0] else: user = UserModel() user.user = u user.last_latitude = lat user.last_longitude = lg user.last_location = name user.put() self.redirect('/main')