Inverse-Rezeptsuche/application/db.py

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)