Create menu.py
Browse files- routes/menu.py +10 -0
routes/menu.py
ADDED
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import pandas as pd
|
2 |
+
from utils.file_initializer import MENU_FILE
|
3 |
+
|
4 |
+
def load_menu(preferences=None, allergies=None):
|
5 |
+
menu = pd.read_excel(MENU_FILE)
|
6 |
+
if preferences:
|
7 |
+
menu = menu[menu["Preferences"].str.contains(preferences, na=False)]
|
8 |
+
if allergies:
|
9 |
+
menu = menu[~menu["Allergens"].str.contains(allergies, na=False)]
|
10 |
+
return menu
|