diff --git a/lib/debug.py b/lib/debug.py index 4f922eb69..9a38a0160 100644 --- a/lib/debug.py +++ b/lib/debug.py @@ -2,8 +2,9 @@ from sqlalchemy import create_engine -from models import Company, Dev +from models import Company, Dev, Freebie if __name__ == '__main__': engine = create_engine('sqlite:///freebies.db') - import ipdb; ipdb.set_trace() + import ipdb + ipdb.set_trace() \ No newline at end of file diff --git a/lib/seed.py b/lib/seed.py index b16becbbb..0c3591b12 100644 --- a/lib/seed.py +++ b/lib/seed.py @@ -1,3 +1,32 @@ #!/usr/bin/env python3 # Script goes here! + +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker +from models import Base, Company, Dev, Freebie + +engine = create_engine('sqlite:///freebies.db') + +Base.metadata.bind = engine + +DBSession = sessionmaker(bind=engine) +session = DBSession() + +if 'session' not in locals(): + session = DBSession() + +freebie1 = Freebie(item_name="keyboard", value=10) +freebie2 = Freebie(item_name="mouse", value=20) + +company = session.query(Company).filter_by(id=1).first() +dev = session.query(Dev).filter_by(id=1).first() + +freebie1.company = company +freebie2.dev = dev + +session.add(freebie1) +session.add(freebie2) + +session.commit() +session.close() \ No newline at end of file