Blum / backend /app /services /bootstrap.py
Italianhype's picture
Upload folder using huggingface_hub
2deb2c5 verified
from sqlalchemy import select
from sqlalchemy.orm import Session
from app.core.database import Base, engine
from app.data.seed_assets import SEED_ASSETS
from app.models import Asset
def create_schema() -> None:
Base.metadata.create_all(bind=engine)
def seed_asset_universe(db: Session) -> int:
inserted = 0
existing = set(db.scalars(select(Asset.ticker)).all())
for item in SEED_ASSETS:
if item["ticker"] in existing:
continue
db.add(Asset(**item))
inserted += 1
db.commit()
return inserted
def bootstrap_database(db: Session) -> dict:
create_schema()
inserted = seed_asset_universe(db)
return {"schema_ready": True, "seeded_assets": inserted}