From 4dd69dabe086c99bef494b1797210e0074853e33 Mon Sep 17 00:00:00 2001 From: sosokker Date: Tue, 2 May 2023 15:27:01 +0700 Subject: [PATCH] Delete model.py --- model.py | 47 ----------------------------------------------- 1 file changed, 47 deletions(-) delete mode 100644 model.py diff --git a/model.py b/model.py deleted file mode 100644 index ac9c039..0000000 --- a/model.py +++ /dev/null @@ -1,47 +0,0 @@ -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