import time

from utils import *

# Параметры подключения к базе данных
conn_params = {
    "dbname": "db_univer",
    "user": "postgres",
    "password": "123",
    "host": "localhost",
}


if __name__ == "__main__":
    delete_all_rows_from_all_tables(conn_params)

    start_time = time.perf_counter()

    # Заполняем таблицы первого уровня
    load_csv_to_db(conn_params, "division", "division.csv")
    load_csv_to_db(conn_params, "organizer", "organizer.csv")
    load_csv_to_db(conn_params, "competition_stage", "competition_stage.csv")
    load_csv_to_db(conn_params, "sportsman", "sportsman.csv", date_columns=[4])
    load_csv_to_db(conn_params, "judge", "judge.csv")

    # Второй уровень
    generate_competitions(conn_params, 200)

    # Третий уровень
    generate_protocols(conn_params)
    generate_judge_requests(conn_params)
    generate_participant_requests(conn_params)
    generate_division_in_competitions(conn_params)

    # Четвёртый уровень
    generate_result_in_stage(conn_params)

    # Пятый уровень
    generate_shot_series(conn_params)

    print_rows_count_summary(conn_params)

    print(f"База данных заполнена за {time.perf_counter() - start_time:.3}с")