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:
parent
f7ee72a66b
commit
8c73123862
8 changed files with 355 additions and 77 deletions
23
src/nrsk/documents/seed_doc_db.py
Normal file
23
src/nrsk/documents/seed_doc_db.py
Normal 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")
|
||||
Loading…
Add table
Add a link
Reference in a new issue