def get_autocomplete_data(group, attribute):
groups = dict(
users=User.get_all(),
products=Product.get_all(),
customers = Customer.get_all(),
expenses=Expense.get_all(),
warehouses=Warehouse.get_all()
)
data = []
for obj in groups[group]:
value = getattr(obj, attribute)
if value not in set(data):
data.append(value)
return data
def main_page():
users = User.get_all()
views = View.get_all()
if request.method == "POST":
search_term = request.form['search_term']
user = User.search(search_term)
if user:
return redirect(
url_for('user.update', id=user.id)
)
view = View.search(search_term)
if view:
return redirect(
url_for('view.update', id=view.id)
)
return render_template(
"admin/main-page.html",
user_heads=username_head,
users=users,
views=views
)
def test_should_return_list_of_all_users(self):
users = User.get_all()
self.assertEqual(users, [self.user, self.user2])
