def render_search():
create_whoosh_dir()
rest_ix = get_restaurant_index()
loc_ix = get_location_index()
cat_ix = get_category_index()
restSearchableFields = ["name","phonenum"]
locSearchableFields = ["address","neighborhood", "zipcode"]
catSearchableFields = ["name"]
restDataList = []
locDataList = []
catDataList = []
if request.method == 'POST':
search_query = request.form['search']
restDataList = search_results(rest_ix, search_query, restSearchableFields)
locDataList = search_results(loc_ix, search_query, locSearchableFields)
catDataList = search_results(cat_ix, search_query, catSearchableFields)
constructRelatedModels(restDataList, locDataList, catDataList)
return render_template('search_results.html',
restDataNames=Restaurant.getDataNames() + ["context"],
restDataList=restDataList,
locDataNames=Location.getDataNames() + ["context"],
locDataList=locDataList,
catDataNames=Category.getDataNames() + ["context"],
catDataList=catDataList)
def render_restaurant():
tableDataDict = getTableDataDict("Restaurants","restaurant", Restaurant)
restDataDictList = getDataDictList(getModels(Restaurant, tableDataDict['offset'],
tableDataDict['sortby'], tableDataDict['direction']))
return render_template('template_db.html',
dataNames=Restaurant.getDataNames(),
dataList=restDataDictList,
**tableDataDict
)
def render_category_id(category_id=None):
catModel = Category.query.get_or_404(category_id)
imgList = [restModel.imageurl for restModel in catModel.restlist][:5]
while len(imgList) < 5:
imgList = imgList + [url for url in imgList]
imgList = imgList[:5]
relatedRestModels = getDataDictList(catModel.restlist)
return render_template('category.html',
catModel = catModel,
restAttrs = Restaurant.getDataNames(),
imgList = imgList,
restListModels = dumps(relatedRestModels))
def test_get_data_names_2(self):
self.assertEqual(Restaurant.getDataNames()[0], "id")
def test_get_data_names_1(self):
self.assertEqual(Restaurant.getDataNames(), ["id", "name", "phonenum", "rating", "reviewcount"])
def test_get_data_names_0(self):
self.assertEqual(len(Restaurant.getDataNames()), 5)
