diff --git a/model.py b/model.py new file mode 100644 index 0000000..ac9c039 --- /dev/null +++ b/model.py @@ -0,0 +1,47 @@ +import sqlite3 + +class FoodSearch: + def __init__(self, db_path): + self.conn = sqlite3.connect(db_path) + self.cursor = self.conn.cursor() + + def search(self, user_input) -> list: + query = f"SELECT * FROM food_data WHERE product_name LIKE '%{user_input}%'" + self.cursor.execute(query) + results = self.cursor.fetchall() + + return results + +import sqlite3 + +class FoodNutrient: + def __init__(self, db_path): + self.conn = sqlite3.connect(db_path) + self.cursor = self.conn.cursor() + self.nutrient_columns = [ + 'energy-kcal_100g', + 'fat_100g', + 'saturated-fat_100g', + 'cholesterol_100g', + 'carbohydrates_100g', + 'fiber_100g', + 'proteins_100g', + 'salt_100g', + 'sodium_100g', + 'potassium_100g', + 'calcium_100g', + 'iron_100g', + 'vitamin-a_100g', + 'vitamin-c_100g' + ] + + def throw_nutrient(self, index_val): + nutrient_cols = ", ".join([f"`{col}`" for col in self.nutrient_columns]) + query = f"SELECT {nutrient_cols} FROM food_data WHERE `index` = {index_val}" + self.cursor.execute(query) + nutrient_data = self.cursor.fetchone() + + return {col: nutrient_data[i] for i, col in enumerate(self.nutrient_columns)} + +fn = FoodNutrient('food_data.db') +print(fn.throw_nutrient(3)) \ No newline at end of file