def post(self):
upload_files = self.get_uploads('file') # 'file' is file upload field in the form
blob_info = upload_files[0]
# Get the current user's Couple entity to set it as a ancestor for each entry in the csv.Dialect
key = "Couple_Key|" + self.user.user_id()
couple_key = ra_memcache.cache_entity(key,self.user.user_id(),None,models.Couple.by_user_id,keys_only=True)
self.process_csv(blob_info,couple_key)
# Delete file after import
blobstore.delete(blob_info.key())
# Update Couple hitlist
hitlist_key = "Hitlist|" + str(couple_key.id())
ra_memcache.hitlist_cache(hitlist_key,couple_key,update=True)
self.redirect("/")
def check_hitlist(self,couple_key,eatery_id):
""" Checks that a given eatery id is in the couple's hitlist"""
hitlist_key = "Hitlist|" + str(couple_key.id())
# Get a list of Entity keys that are associated with this user.
hitlist_keys = ra_memcache.hitlist_cache(hitlist_key,couple_key)
# Loop through hitlist to make sure that the requested Eatery is in their hitlist.
found = False
for e_key in hitlist_keys:
if e_key.id() == eatery_id:
found = True
break
return found
def get(self):
key = "Couple_Key|" + self.user.user_id()
couple_key = ra_memcache.cache_entity(key,self.user.user_id(),None,models.Couple.by_user_id,keys_only=True)
if couple_key:
# get their hitlist.
hitlist_key = "Hitlist|" + str(couple_key.id())
hitlist_keys = ra_memcache.hitlist_cache(hitlist_key,couple_key)
# Build a list of Eatery entities to render.
hitlist = []
for e_key in hitlist_keys:
key = 'Eatery|' + str(e_key.id())
eatery = ra_memcache.cache_entity(key,e_key.id(),couple_key,models.Eatery.by_id)
# filter for Eateries with a last visit date.
if eatery.LastVisitDate:
# Add Trip for each Eatery, if FirstTripDate != Last Trip add another trip on First Trip date
self.add_trip(eatery.key().id(),eatery.LastVisitDate,couple_key)
if eatery.FirstTripDate and eatery.LastVisitDate > eatery.FirstTripDate:
self.add_trip(eatery.key().id(),eatery.FirstTripDate,couple_key)
self.redirect('/')
def get_hitlist(self,couple_key,filters):
hitlist_key = "Hitlist|" + str(couple_key.id())
# Get a list of Entity keys that are associated with this user.
hitlist_keys = ra_memcache.hitlist_cache(hitlist_key,couple_key)
# Build a list of Eatery entities to render.
hitlist = []
for e_key in hitlist_keys:
key = 'Eatery|' + str(e_key.id())
eatery = ra_memcache.cache_entity(key,e_key.id(),couple_key,models.Eatery.by_id)
if filters:
match = True
# Go through dictionary of filters, if a property of the entity doesn't match a filter do not include in result.
for f in filters:
if hasattr(eatery,f):
if filters[f] and not re.search(filters[f].upper(),getattr(eatery,f).upper()):
match = False
break
if match:
hitlist.append(eatery)
else:
hitlist.append(eatery)
return hitlist
def post(self):
submit = self.request.get('submit')
if submit:
couple_memcache_key = "Couple_Key|" + self.user.user_id()
couple_key = ra_memcache.cache_entity(couple_memcache_key,self.user.user_id(),None,models.Couple.by_user_id,keys_only=True)
if couple_key:
eatery_id = self.request.get("id")
if eatery_id:
if eatery_id.isnumeric():
eatery_id = int(eatery_id)
# check if valid eatery id for the couple_key.
edit_found = self.check_hitlist(couple_key, eatery_id)
new = False
elif eatery_id.upper() == "NEW":
# User is posting data for new Eatery
edit_found = False
new = True
else:
# Fail the post attempt and 403 it.
edit_found = False
new = False
if edit_found or new:
# verify user inputs
failed = False
error_dict = {}
# Verify that the user entered a restaurant name.
restaurant_name = self.request.get("restaurantName")
if not restaurant_name:
failed = True
error_name = "You must enter a name."
error_dict['error_name'] = error_name
cuisine_type = self.request.get("cuisineType")
city = self.request.get("city")
state = self.request.get("state")
notes_comments = self.request.get("notesComments")
completed = self.request.get("completed").upper()
if completed == "TRUE":
completed = True
else:
completed = False
first_trip_date = self.request.get("firstTripDate")
# Check if a value was entered first.
if first_trip_date:
first_trip_date = helpers.convert_string_to_date(first_trip_date)
if not first_trip_date:
failed = True
error_first_trip = "Invalid date format."
error_dict['error_first_trip'] = error_first_trip
else:
first_trip_date = None
last_visit_date = self.request.get("lastVisitDate")
# Check if a value was entered first.
if last_visit_date:
last_visit_date = helpers.convert_string_to_date(last_visit_date)
if not last_visit_date:
failed = True
error_last_visit = "Invalid date format."
error_dict['error_last_visit'] = error_last_visit
else:
last_visit_date = None
number_of_trips = self.request.get("numberOfTrips")
if number_of_trips and not number_of_trips.isnumeric():
failed = True
error_number_of_trips = "Invalid number entered."
error_dict['error_number_of_trips'] = error_number_of_trips
else:
number_of_trips = int(number_of_trips)
p1_Rating = self.request.get("p1Rating")
if p1_Rating:
check = self.check_rating(1,p1_Rating)
if check[0]:
p1_Rating = int(p1_Rating)
else:
error_dict[check[1]] = check[2]
else:
p1_Rating = None
p2_Rating = self.request.get("p2Rating")
if p2_Rating:
check = self.check_rating(2,p2_Rating)
if check[0]:
p2_Rating = int(p2_Rating)
else:
error_dict[check[1]] = check[2]
else:
p2_Rating = None
# Get additional fields.
street_address = self.request.get("streetAddress")
# Verify that the zip code is a number.
zip_code = self.request.get("zipCode")
if zip_code:
if zip_code.isnumeric():
zip_code = int(zip_code)
else:
failed = True
error_zip_code = "Please enter a valid 5 digit zip code."
else:
zip_code = None
# Get yelp ID
yelp_business_id = self.request.get("yelpBusinessID")
# Get eatery entity.
if edit_found:
# handler for edits.
key = 'Eatery|' + str(eatery_id)
eatery = ra_memcache.cache_entity(key,eatery_id,couple_key,models.Eatery.by_id)
if failed:
if edit_found:
# add eatery to render dictionary
error_dict['eatery'] = eatery
# get couple object
couple_memcache_key = "Couple|" + self.user.user_id()
couple = ra_memcache.cache_entity(couple_memcache_key,self.user.user_id(),None,models.Couple.by_user_id,keys_only=False)
# add couple to render dictionary
error_dict['couple'] = couple
if edit_found:
# render edit template for edits.
self.render('hitlist-edit.html',**error_dict)
else:
# Render create template for creates.
self.render('hitlist-create.html',**error_dict)
else:
# User inputs validated prepare to commit to
days_last_trip = 0
# Calculate average rating.
if p1_Rating and p2_Rating:
average_rating = (p1_Rating + p2_Rating) / 2.0
elif p1_Rating:
average_rating = float(p1_Rating)
elif p2_Rating:
average_rating = float(p2_Rating)
else:
average_rating = None
# change eatery entry to reflect changes.
if edit_found:
eatery.RestaurantName = restaurant_name
else:
# Or create new entity.
eatery = models.Eatery(RestaurantName=restaurant_name,parent=couple_key)
eatery.CuisineType = cuisine_type
eatery.City = city
eatery.State = state
eatery.NotesComments = notes_comments
eatery.Completed = completed
eatery.FirstTripDate = first_trip_date
eatery.LastVisitDate = last_visit_date
eatery.NumberOfTrips = number_of_trips
eatery.P1Rating = p1_Rating
eatery.P2Rating = p2_Rating
eatery.AverageRating = average_rating
eatery.StreetAddress = street_address
eatery.ZipCode = zip_code
eatery.YelpBusinessID = yelp_business_id
# Save new Eatery to DB
eatery.put()
# if new trip add to trip db.
self.increment_trip_db(eatery=eatery, couple_key=couple_key,
first_trip_date=first_trip_date, last_visit_date=last_visit_date,
number_of_trips=number_of_trips, edit=edit_found)
if new:
eatery_id = eatery.key().id()
key = 'Eatery|' + str(eatery_id)
hitlist_key = "Hitlist|" + str(couple_key.id())
ra_memcache.hitlist_cache(hitlist_key,couple_key,update=True)
# refresh memcache
eatery = ra_memcache.cache_entity(key,eatery_id,couple_key,models.Eatery.by_id,update=True)
# Redirect to hitlist.
self.redirect("/")
else:
self.error(403)
else:
self.redirect("/")
else:
self.redirect("/register")