"""Seed DB for documents, e.g. with doc types"""

from sqlmodel import Session

from nrsk import DOCS_ROOT
from nrsk.db import get_engine
from nrsk.documents.validate import validate_doc_types


def seed_doc_types():
    engine = get_engine()
    doc_types = validate_doc_types(DOCS_ROOT / "_data" / "doc-types.yaml")

    with Session(engine) as session:
        for dtype in doc_types:
            session.add(dtype)

        session.commit()


if __name__ == "__main__":
    seed_doc_types()
    print("seeded doc types")