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()