From 1772ee0fc68eaea14f88f732c5f3524dc6a44b0d Mon Sep 17 00:00:00 2001 From: + <+> Date: Tue, 12 Sep 2023 14:13:15 +0300 Subject: [PATCH] completed assignment --- lib/seed.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/lib/seed.py b/lib/seed.py index b16becbbb..6f7eedfdf 100644 --- a/lib/seed.py +++ b/lib/seed.py @@ -1,3 +1,22 @@ #!/usr/bin/env python3 # Script goes here! + +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker +from models import Base, Company, Dev, Freebies + +engine = create_engine('sqlite:///database.db') +Base.metadata.bind = engine +DBSession = sessionmaker(bind=engine) +session = DBSession() + +# Seed some sample data +company1 = Company(name="Company A", founding_year=1990) +company2 = Company(name="Company B", founding_year=2000) + +dev1 = Dev(name="Developer 1") +dev2 = Dev(name="Developer 2") + +session.add_all([company1, company2, dev1, dev2]) +session.commit()