import time

import streamlit as st

from queries.process_all_db import process_all_tech_db
from queries.process_gsm import process_gsm_data_to_excel
from queries.process_lte import process_lte_data_to_excel
from queries.process_neighbors import (
    process_neighbors_data,
    process_neighbors_data_to_excel,
)
from queries.process_wcdma import process_wcdma_data_to_excel
from utils.check_sheet_exist import Technology, execute_checks_sheets_exist
from utils.utils_vars import UtilsVars

st.title("Database processing")

uploaded_file = st.file_uploader("Upload updated dump file", type="xlsb")


def process_database(process_func, database_type):
    if uploaded_file is not None:
        start_time = time.time()
        process_func(uploaded_file)
        execution_time = time.time() - start_time
        st.write(
            f"{database_type} database is generated. Execution time: {execution_time:.2f} seconds"
        )
        download_button(database_type)


def download_button(database_type):
    if database_type == "2G":
        data = UtilsVars.final_gsm_database
        file_name = f"2G database_{time.time()}.xlsx"
    elif database_type == "3G":
        data = UtilsVars.final_wcdma_database
        file_name = f"3G database_{time.time()}.xlsx"
    elif database_type == "LTE":
        # data = UtilsVars.final_lte_database
        data = UtilsVars.final_lte_database
        file_name = f"LTE database_{time.time()}.xlsx"
    elif database_type == "All":
        data = UtilsVars.final_all_database
        file_name = f"All database_{time.time()}.xlsx"
    elif database_type == "NEI":
        data = UtilsVars.neighbors_database
        file_name = f"Neighbors database_{time.time()}.xlsx"
    st.download_button(
        type="primary",
        label=f"Download {database_type} Database File",
        data=data,
        file_name=file_name,
        mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
    )


def execute_process_all_tech_db(uploaded_file):
    if uploaded_file is not None:
        start_time = time.time()
        process_all_tech_db(uploaded_file)
        execution_time = time.time() - start_time
        st.write(
            f"All databases are generated. Execution time: {execution_time:.2f} seconds"
        )
        download_button("All")


col1, col2, col3, col4 = st.columns(4)
col5, col6, col7, col8 = st.columns(4)
if uploaded_file is not None:
    # UtilsVars.file_path = uploaded_file
    try:
        execute_checks_sheets_exist(uploaded_file)
        if (
            Technology.gsm == False
            and Technology.wcdma == False
            and Technology.lte == False
        ):
            st.error(
                """
                     Uploaded file does not contain required sheets for any technology.
                    "gsm": ["BTS", "BCF", "TRX"],
                    "wcdma": ["WCEL", "WBTS", "WNCEL"],
                    "lte": ["LNBTS", "LNCEL", "LNCEL_FDD", "LNCEL_TDD"],
                    "neighbors": ["ADCE", "ADJS", "ADJI", "ADJG", "ADJW", "BTS", "WCEL"],
                """
            )

        if Technology.gsm == True:
            with col1:
                st.button(
                    "Generate 2G DB",
                    on_click=lambda: process_database(process_gsm_data_to_excel, "2G"),
                )
        if Technology.wcdma == True:
            with col2:
                st.button(
                    "Generate 3G DB",
                    on_click=lambda: process_database(
                        process_wcdma_data_to_excel, "3G"
                    ),
                )
        if Technology.lte == True:
            with col3:
                st.button(
                    "Generate LTE DB",
                    on_click=lambda: process_database(process_lte_data_to_excel, "LTE"),
                )
        if Technology.gsm == True or Technology.wcdma == True or Technology.lte == True:
            with col4:
                st.button(
                    "Generate All DBs",
                    on_click=lambda: execute_process_all_tech_db(uploaded_file),
                )
        if Technology.neighbors == True:
            with col5:
                st.button(
                    "Generate NEI DB",
                    on_click=lambda: process_database(
                        process_neighbors_data_to_excel, "NEI"
                    ),
                    # on_click=lambda: process_neighbors_data(uploaded_file),
                )

    except Exception as e:
        st.error(f"Error: {e}")