From 0fd2c14276d85218bf56b0446135656c95c6a50c Mon Sep 17 00:00:00 2001 From: + Date: Wed, 6 Sep 2023 01:14:35 +0300 Subject: [PATCH] seed.py --- lib/seed.py | 27 ++++++++++++++++++++++++++- 1 file changed, 26 insertions(+), 1 deletion(-) diff --git a/lib/seed.py b/lib/seed.py index b16becbbb..dbd8af699 100644 --- a/lib/seed.py +++ b/lib/seed.py @@ -1,3 +1,28 @@ #!/usr/bin/env python3 -# Script goes here! +from sqlalchemy import create_engine +from sqlalchemy.orm import sessionmaker +from models import Company, Dev + + + +Session = sessionmaker(bind=engine) +session = Session() + +# Create sample data +company1 = Company(name='Company A', founding_year=2000) +company2 = Company(name='Company B', founding_year=2010) +company3 = Company(name='Company C', founding_year=2020) + +dev1 = Dev(name='Dev X') +dev2 = Dev(name='Dev Y') +dev3 = Dev(name='Dev Z') + + +session.add_all([company1, company2, company3, dev1, dev2, dev3]) + + +session.commit() + + +session.close()