diff --git a/lib/seed.py b/lib/seed.py index b16becbbb..636660569 100644 --- a/lib/seed.py +++ b/lib/seed.py @@ -1,3 +1,20 @@ #!/usr/bin/env python3 # Script goes here! +from sqlalchemy import Column, ForeignKey, Integer, String +from sqlalchemy.orm import relationship +from sqlalchemy.ext.declarative import declarative_base + +Base = declarative_base() + +class Freebie(Base): + __tablename__ = 'freebies' + + id = Column(Integer, primary_key=True) + dev_id = Column(Integer, ForeignKey('devs.id')) + company_id = Column(Integer, ForeignKey('companies.id')) + item_name = Column(String) + value = Column(Integer) + + dev = relationship('Dev', backref='freebies') + company = relationship('Company', backref='freebies')