initial wip
This commit is contained in:
37
src/populate.py
Normal file
37
src/populate.py
Normal file
@@ -0,0 +1,37 @@
|
||||
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()
|
||||
Reference in New Issue
Block a user