File size: 670 Bytes
7872230 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | from sqlalchemy.orm import Session
from app.db.session import Base, engine
from app.models import RuleMaster
from app.services.seed_rules import seed_all_product_rules
def create_database() -> None:
Base.metadata.create_all(bind=engine)
def seed_database(db: Session) -> int:
return seed_all_product_rules(db)
if __name__ == "__main__":
from app.db.session import SessionLocal
create_database()
session = SessionLocal()
try:
inserted = seed_database(session)
session.commit()
print(f"Seeded {inserted} rules")
print(f"Total rules: {session.query(RuleMaster).count()}")
finally:
session.close()
|