Add postgresql persistence of data in data dict

Moved back to sqlmodel because we do need some way
of putting this info into a database.
This commit is contained in:
Nick Touran 2026-01-05 12:04:22 -05:00
parent f7ee72a66b
commit 8c73123862
8 changed files with 355 additions and 77 deletions

View file

@ -0,0 +1,23 @@
"""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")