diff --git a/app/model.py b/app/model.py index 25ad312..ac9c039 100644 --- a/app/model.py +++ b/app/model.py @@ -10,4 +10,38 @@ class FoodSearch: self.cursor.execute(query) results = self.cursor.fetchall() - return results \ No newline at end of file + 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