model.py

from flask import Flask
from marshmallow import Schema, fields, pre_load, validate
from flask_marshmallow import Marshmallow
from flask_sqlalchemy import SQLAlchemy
from flask_redis import FlaskRedis

ma = Marshmallow()
db = SQLAlchemy()
redis_cache = FlaskRedis()

class FoodModel(db.Model):
    __tablename__ = 'foods'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(250), nullable=False)
    description = db.Column(db.String(250))
    creation_date = db.Column(db.TIMESTAMP, server_default=db.func.current_timestamp(), nullable=False)
    restaurant_id = db.Column(db.Integer, db.ForeignKey('restaurants.id', ondelete='CASCADE'), nullable=False)
    restaurant = db.relationship('RestaurantModel', backref=db.backref('foods', lazy='dynamic' ))
    menu_id = db.Column(db.Integer, db.ForeignKey('menus.id', ondelete='CASCADE'), nullable=False)
    menu = db.relationship('MenuModel')

    def __init__(self, name, description, restaurant_id, menu_id):
        self.name = name
        self.description = description
        self.restaurant_id = restaurant_id
        self.menu_id = menu_id


class MenuModel(db.Model):
    __tablename__ = 'menus'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(250), nullable=False)
    restaurant_id = db.Column(db.Integer, db.ForeignKey('restaurants.id', ondelete='CASCADE'), nullable=False)
    restaurant = db.relationship('RestaurantModel')

    def __init__(self, name, restaurant_id):
        self.name = name
        self.restaurant_id = restaurant_id


class RestaurantModel(db.Model):
    __tablename__ = 'restaurants'
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(150), unique=True, nullable=False)

    def __init__(self, name):
        self.name = name


class RestaurantSchema(ma.Schema):
    id = fields.Integer()
    name = fields.String(required=True)

class MenuSchema(ma.Schema):
    id = fields.Integer()
    restaurant_id = fields.Integer(required=True)
    name = fields.String(required=True)

class FoodSchema(ma.Schema):
    id = fields.Integer(dump_only=True)
    restaurant_id = fields.Integer(required=True)
    name = fields.String(required=True, validate=validate.Length(1))
    description = fields.String()
    creation_date = fields.DateTime()