"""
Inicializacao de banco de dados.
Cria tabelas e executa seed inicial em ambos os bancos.
"""

from app.db.database import Base, engine
from app.db.mock_database import MockBase, mock_engine
from app.db.models import Tool
from app.db.mock_models import Customer, Order, ReviewSchedule, Vehicle
from app.db.mock_seed import seed_mock_data
from app.db.tool_seed import seed_tools


def init_db():
    """Cria tabelas e executa seed inicial em ambos os bancos."""
    print("Inicializando bancos...")

    try:
        print("Criando tabelas PostgreSQL (tools)...")
        Base.metadata.create_all(bind=engine)
        print("Populando tools iniciais...")
        seed_tools()
        print("PostgreSQL OK.")
    except Exception as exc:
        print(f"Aviso: falha no PostgreSQL (tools): {exc}")

    try:
        print("Criando tabelas MySQL (dados ficticios)...")
        MockBase.metadata.create_all(bind=mock_engine)
        print("Populando dados ficticios iniciais...")
        seed_mock_data()
        print("MySQL mock OK.")
    except Exception as exc:
        print(f"Aviso: falha no MySQL mock: {exc}")

    print("Bancos inicializados com sucesso!")


if __name__ == "__main__":
    init_db()