38 lines
980 B
Python
38 lines
980 B
Python
from sqlmodel import Session
|
|
|
|
from .model.util import engine
|
|
from src.model import count_rows, Status, Contact
|
|
|
|
|
|
def populate_contact(session):
|
|
if count_rows(session, Contact) == 0:
|
|
for kwargs in (
|
|
dict(address="Active"),
|
|
dict(id="I", name="Inactive"),
|
|
dict(id="N", name="New"),
|
|
dict(id="P", name="Prepared"),
|
|
dict(id="X", name="eXcluded"),
|
|
):
|
|
session.add(Status(**kwargs))
|
|
|
|
|
|
|
|
def populate_status(session):
|
|
if count_rows(session, Status) == 0:
|
|
for kwargs in (
|
|
dict(id="A", name="Active"),
|
|
dict(id="I", name="Inactive"),
|
|
dict(id="N", name="New"),
|
|
dict(id="P", name="Prepared"),
|
|
dict(id="X", name="eXcluded"),
|
|
):
|
|
session.add(Status(**kwargs))
|
|
|
|
|
|
def populate():
|
|
with Session(engine) as session:
|
|
|
|
populate_status(session)
|
|
|
|
session.commit()
|