Esempio n. 1
0
    def get(self):
        keyword = request.args.get('keyword')
        if not keyword:
            return redirect(url_for('index'))

        kw = keyword.lower().split(',')

        temp = {}
        rests = Rest.get_rests_by_kw(kw[0].lstrip(), kw[len(kw)-1].lstrip(), None)
        all_rests = Rest.get_all_by_kw(kw[0].lstrip())

        for r in rests:
            temp[r.restid] = r
        for a in all_rests:
            if a.restid not in temp:
                rests.append(a)

        prices = sorted(rests, key=lambda rest: rest.price)
        lowest = prices[0].price
        highest = prices[len(prices)-1].price

        price_range = "[" + str(lowest) + "," + str(highest) + "]"

        for rest in rests:
            rest.country_rating = CountryRating.get_country_rating(rest.restid)

        return render_template('search_result.html', rests=rests, keyword=keyword, sort_by="rating",
                               price_range=price_range, lowest=lowest, highest=highest)
Esempio n. 2
0
    def post(self):
        op = request.form.get('op')
        if op == 'search':
            keyword = request.form.get('keyword')
            return redirect(url_for('search_result') + '?keyword=' + keyword)
        elif op == 'update':
            price_range = request.form.get('p_range').split(',')
            sort_by = request.form.get('sort_by')
            keyword = request.form.get('keyword')
            country = request.form.get('country')
            current_lat = request.form.get('current_lat')
            current_long = request.form.get('current_long')
            if not keyword:
                return redirect(url_for('index'))

            kw = keyword.lower().split(',')
            if sort_by == 'rating':
                temp = {}
                rests = Rest.get_rests_by_kw(kw[0].lstrip(), kw[len(kw)-1].lstrip(), country)
                all_rests = Rest.get_all_by_kw(kw[0].lstrip())
                for r in rests:
                    temp[r.restid] = r
                for a in all_rests:
                    if a.restid not in temp:
                        rests.append(a)
                print "range:", price_range[0]
                print "range:", price_range[1]
                for w in rests:
                    if w.price < int(price_range[0]) or w.price > int(price_range[1]):
                        rests.remove(w)
            else:
                if current_lat and current_long:
                    rests = Rest.get_rests_by_distance(kw[0].lstrip(), current_lat, current_long)
                else:
                    temp = {}
                    rests = Rest.get_rests_by_kw(kw[0].lstrip(), kw[len(kw)-1].lstrip(), None)
                    all_rests = Rest.get_all_by_kw(kw[0].lstrip())

                    for r in rests:
                        temp[r.restid] = r
                    for a in all_rests:
                        if a.restid not in temp:
                            rests.append(a)
                    for w in rests:
                        if w.price < int(price_range[0]) or w.price > int(price_range[1]):
                            rests.remove(w)

            for rest in rests:
                rest.country_rating = CountryRating.get_country_rating(rest.restid)

            return render_template('search_result.html', rests=rests, keyword=keyword,
                                   sort_by=sort_by, country=country,
                                   price_range='['+price_range[0]+','+ price_range[1]+']',
                                   lowest=int(price_range[0]),highest=int(price_range[1]))
        else:
            abort(404)