def homepage():
if "restaurant_id" not in request.cookies:
return redirect("/")
rid = request.cookies.get("restaurant_id")
print rid
foods = apriori.getRules(int(rid))
length = request.args.get("length")
return render_template("homepage.html", foods=foods, length=10)
def getGroups():
from apriori import getRules
rules = getRules()
groups = []
for itemset in rules:
tempset = []
for item in itemset[:-1]:
#print(str for str in item)
tempset.extend([str for str in item])
groups.append(tempset)
return groups
def get_single_relation():
if "restaurant_id" not in request.cookies:
return redirect("/")
rid = request.cookies.get("restaurant_id")
fid = request.args.get("fid")
foods = apriori.getRules(int(rid))
single_food = None
print fid
for f in foods:
if int(f["prefix_foods"][0]["id"]) == int(fid) and len(f["prefix_foods"]) == 1:
single_food = f
break
# print single_food
def rule_cmp(x, y):
return x["conf"] < y["conf"]
if single_food != None and len(single_food["rules"]) > 7:
single_food["rules"] = sorted(single_food["rules"], cmp=rule_cmp)[0:6]
return render_template("single_relation.html", single_food=single_food)
