<commit_before>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'

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

    @property
    def serialize(self):
        return {
            'name'  : self.name,
            'id'    : self.id
        }


class MenuItem(Base):
    __tablename__ = 'menu_item'

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

    @property
    def serialize(self):
        return {
            'name'          : self.name,
            'id'            : self.id,
            'description'   : self.description,
            'price'         : self.price,
            'course'        : self.course
        }

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

Base.metadata.create_all(engine)<commit_msg>Added UserID table to restaurantmenu.db setup<commit_after>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 UserID(Base):
    __tablename__ = 'user_id'
    
    id = Column(
        Integer, primary_key = True
    ) 
    name = Column(
        String(80), nullable = False
    )
    email = Column(
        String(80), nullable = False
    )
    picture = Column(
        String(150), nullable = False
    )

    @property
    def serialize(self):
        return {
            'name'  : self.name,
            'id'    : self.id,
            'email' : self.email,
            'picture': self.picture
        }
          


class Restaurant(Base):
    __tablename__ = 'restaurant'

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

    user_id = Column(
        Integer, ForeignKey('user_id.id')
    )
    user_id = relationship(UserID)

    @property
    def serialize(self):
        return {
            'name'  : self.name,
            'id'    : self.id
        }


class MenuItem(Base):
    __tablename__ = 'menu_item'

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

    @property
    def serialize(self):
        return {
            'name'          : self.name,
            'id'            : self.id,
            'description'   : self.description,
            'price'         : self.price,
            'course'        : self.course
        }

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

Base.metadata.create_all(engine)
{'fraction_non_alphanumeric': 0.09917119176907688, 'fraction_numerical': 0.00943126607602172, 'mean_word_length': 1.8455284552845528, 'pattern_counts': {'":': 0, '<': 3, '<?xml version=': 0, '>': 3, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '1409076', 'n_tokens_mistral': 1090, 'n_tokens_neox': 1042, 'n_words': 234}