65 lines
No EOL
2.1 KiB
Python
65 lines
No EOL
2.1 KiB
Python
import json
|
|
|
|
from ingredients import Ingredient, find_ingredients_by_recipe_id
|
|
|
|
from pydantic import BaseModel, Field
|
|
from typing import List, ClassVar, Tuple
|
|
|
|
class Recipe(BaseModel):
|
|
KEYS: ClassVar[List[str]] = ['id', 'name', 'link', 'image_urls', 'raw_data']
|
|
id: int
|
|
name: str
|
|
link: str
|
|
raw_data: str
|
|
image_urls: List[str] = []
|
|
ingredients: List[Ingredient] = []
|
|
|
|
async def create(conn):
|
|
await conn.execute('''
|
|
CREATE TABLE IF NOT EXISTS Recipe (
|
|
id INTEGER PRIMARY KEY,
|
|
name TEXT,
|
|
link TEXT,
|
|
image_urls TEXT,
|
|
raw_data TEXT
|
|
);''')
|
|
|
|
async def insert_recipe(conn, recipe: Recipe):
|
|
async with conn.execute('''
|
|
INSERT INTO Recipe (name, link, raw_data, image_urls)
|
|
VALUES (?, ?, ?, ?)
|
|
''', (recipe.name, recipe.link, recipe.raw_data, json.dumps(recipe.image_urls))) as cursor:
|
|
recipe.id = cursor.lastrowid
|
|
|
|
def row_to_recipe(col_tuples: List[Tuple[str, ...]]) -> Recipe:
|
|
d = {k:v for k,v in col_tuples}
|
|
d['image_urls'] = json.loads(d['image_urls'])
|
|
return Recipe(**d)
|
|
|
|
async def find_recipe_by_id(conn, recipe_id: int) -> Recipe:
|
|
async with conn.execute(f'''
|
|
SELECT {','.join(Recipe.KEYS)} FROM Recipe
|
|
WHERE id = ?
|
|
LIMIT 1
|
|
''', (recipe_id,)) as cursor:
|
|
async for row in cursor:
|
|
return row_to_recipe(zip(Recipe.KEYS, row))
|
|
|
|
async def find_recipes_by_name(conn, name: str) -> List[Recipe]:
|
|
async with conn.execute(f'''
|
|
SELECT {','.join(Recipe.KEYS)} FROM Recipe
|
|
WHERE name LIKE ?
|
|
''', (f'%{name}%',)) as cursor:
|
|
async for row in cursor:
|
|
yield row_to_recipe(zip(Recipe.KEYS, row))
|
|
|
|
async def get_all(conn) -> List[Recipe]:
|
|
async with conn.execute(f'''
|
|
SELECT {','.join(Recipe.KEYS)} FROM Recipe
|
|
''') as cursor:
|
|
async for row in cursor:
|
|
yield row_to_recipe(zip(Recipe.KEYS, row))
|
|
|
|
async def load_ingredients(conn, recipe: Recipe):
|
|
async for ingredient in find_ingredients_by_recipe_id(conn, recipe.id):
|
|
recipe.ingredients.append(ingredient) |