Skip to content
Open
33 changes: 33 additions & 0 deletions Lesson-2/Objective-1-Solution/database_setup.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
import os
import sys
from sqlalchemy import Column, ForeignKey, Integer, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship
from sqlalchemy import create_engine

Base = declarative_base()


class Restaurant(Base):
__tablename__ = 'restaurant'

id = Column(Integer, primary_key=True)
name = Column(String(250), nullable=False)


class MenuItem(Base):
__tablename__ = 'menu_item'

name = Column(String(80), nullable=False)
id = Column(Integer, primary_key=True)
description = Column(String(250))
price = Column(String(8))
course = Column(String(250))
restaurant_id = Column(Integer, ForeignKey('restaurant.id'))
restaurant = relationship(Restaurant)


engine = create_engine('sqlite:///restaurantmenu.db')


Base.metadata.create_all(engine)
Loading