"""DB seed operations."""

from manualadjustment.connectors import mysql
from manualadjustment.models.currencies import Currency
from manualadjustment.models.manual_adjustment import ManualAdjustment
from manualadjustment.models.manual_adjustment_category import (
    ManualAdjustmentCategory)
from manualadjustment.models.orchadmin_users import OrchadminUser
from manualadjustment.models.period import Period
from manualadjustment.models.vendor import Vendor
from utils.tables import currencies
from utils.tables import manual_adjustment
from utils.tables import manual_adjustment_category
from utils.tables import orchadmin_users
from utils.tables import period
from utils.tables import vendor


ar_session = mysql.art_relations_session()


def create_tables():
    """Reset and create initial tables."""
    mysql.ArtRelationsBaseModel.metadata.drop_all(mysql.ar_engine)
    mysql.ArtRelationsBaseModel.metadata.create_all(mysql.ar_engine)


def seed_art_relations():
    """Seed initial tables with data."""
    ar_session.execute(
        ManualAdjustment.__table__.insert().values(manual_adjustment.data))
    ar_session.execute(
        Vendor.__table__.insert().values(vendor.data))
    ar_session.execute(
        Period.__table__.insert().values(period.data))
    ar_session.execute(
        ManualAdjustmentCategory.__table__.insert().values(
            manual_adjustment_category.data))
    ar_session.execute(
        OrchadminUser.__table__.insert().values(orchadmin_users.data))
    ar_session.execute(
        Currency.__table__.insert().values(currencies.data))
    ar_session.commit()
    ar_session.close()