24 lines
525 B
Python
24 lines
525 B
Python
|
"""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")
|