83 lines
2.8 KiB
Python
83 lines
2.8 KiB
Python
import sqlalchemy as db
|
|
from sqlalchemy import Column, String, Integer, Numeric, Table, DateTime, ARRAY, ForeignKey, create_engine, LargeBinary, Enum, Text
|
|
from sqlalchemy.orm import sessionmaker, relationship, column_property
|
|
from datetime import datetime
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
import enum
|
|
from flask_sqlalchemy import SQLAlchemy
|
|
from flask import Flask
|
|
|
|
engine = db.create_engine('sqlite:///./test.sqlite', echo=False)
|
|
connection = engine.connect()
|
|
Base = declarative_base()
|
|
Session = sessionmaker(bind=engine)
|
|
|
|
# https://docs.sqlalchemy.org/en/13/orm/basic_relationships.html#association-object
|
|
class Link(Base):
|
|
__tablename__ = 'link'
|
|
recipe_id = Column(Integer, ForeignKey('recipe.recipe_id'), primary_key=True)
|
|
ingredient_id = Column(Integer, ForeignKey('ingredient.ingredient_id'), primary_key=True)
|
|
ingredient_amount = Column('ingredient_amount', Text)
|
|
ingredient_amount_mu = Column('ingredient_amount_mu', Text) # measurement unit
|
|
|
|
recipe = relationship("Recipe", back_populates="ingredient")
|
|
ingredient = relationship("Ingredient", back_populates="recipe")
|
|
|
|
def ingredients(self):
|
|
return self.ingredient.name
|
|
|
|
class Recipe(Base):
|
|
__tablename__ = "recipe"
|
|
recipe_id = Column('recipe_id', Integer, primary_key=True, autoincrement=True)
|
|
name = Column('name', Text)
|
|
instructions = Column('instructions', Text)
|
|
url = Column('url', Text)
|
|
img = Column('img', LargeBinary)
|
|
ingredient = relationship("Link", back_populates="recipe")
|
|
|
|
def ingredients(self):
|
|
l = []
|
|
for i in self.ingredient:
|
|
l.append(i.ingredients())
|
|
return l
|
|
|
|
def ingredientDict(self):
|
|
l = {}
|
|
for i in self.ingredient:
|
|
l[i.ingredients()] = [i.ingredient_amount, i.ingredient_amount_mu]
|
|
return l
|
|
|
|
def serialize(self):
|
|
ingredients = []
|
|
|
|
if self.img is not None:
|
|
img = self.img.decode('utf-8')
|
|
else:
|
|
img = None
|
|
|
|
data = {
|
|
"recipe_id": self.recipe_id,
|
|
"name":self.name,
|
|
"instructions":self.instructions,
|
|
"url": self.url,
|
|
"img": img,
|
|
"ingredients": self.ingredients()
|
|
}
|
|
return data
|
|
|
|
class Ingredient(Base):
|
|
__tablename__ = "ingredient"
|
|
ingredient_id = Column('ingredient_id', Integer, primary_key=True, autoincrement=True)
|
|
name = Column('name', Text)
|
|
recipe = relationship("Link", back_populates="ingredient")
|
|
trunks = relationship("Trunk")
|
|
|
|
class Trunk(Base):
|
|
__tablename__ = "trunk"
|
|
trunk_id = Column('trunk_id', Integer, primary_key=True, autoincrement=True)
|
|
name = Column('name', Text)
|
|
ingredient_id = Column(Integer, ForeignKey('ingredient.ingredient_id'))
|
|
|
|
|
|
Base.metadata.create_all(engine)
|