# -*- coding: utf-8 -*-
"""
/***************************************************************************
Name                 : Mascaret
Description          : Pre and Postprocessing for Mascaret for QGIS
Date                 : June,2017
copyright            : (C) 2017 by Artelia
email                :
***************************************************************************/

/***************************************************************************
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 3 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 ***************************************************************************/
 """
import os
from datetime import datetime
from qgis.PyQt.QtCore import *
from qgis.PyQt.uic import *
from qgis.core import *
from qgis.gui import *

from .ui.custom_control import ClassWarningBox

if int(qVersion()[0]) < 5:  # qt4

    from qgis.PyQt.QtGui import *
else:  # qt4
    from qgis.PyQt.QtWidgets import *


class ClassDeletrunDialog(QDialog):
    """
    Class allow to delete run
    """

    def __init__(self, mgis, iface):
        QDialog.__init__(self)
        self.mgis = mgis
        self.mdb = self.mgis.mdb
        self.iface = iface
        self.ui = loadUi(os.path.join(self.mgis.masplugPath, 'ui/ui_delete.ui'),
                         self)
        self.box = ClassWarningBox()
        self.listeRuns = []
        self.listeScen = {}
        self.cond_com = None
        self.tree = None
        self.parent = {}
        self.child = {}

        self.init_gui()

    def init_gui(self):
        """
            initialisation GUI
        """
        liste_col = self.mdb.list_columns('runs')
        self.cond_com = ('comments' in liste_col)

        dico = self.mdb.select("runs", "", "date")

        if self.cond_com:
            for run, scen, date, comments in zip(dico["run"], dico["scenario"],
                                                 dico["date"],
                                                 dico["comments"]):
                if run not in self.listeRuns:
                    self.listeRuns.append(run)
                    self.listeScen[run] = []
                self.listeScen[run].append((scen, date, comments))
        else:
            for run, scen, date in zip(dico["run"], dico["scenario"],
                                       dico["date"]):
                if run not in self.listeRuns:
                    self.listeRuns.append(run)
                    self.listeScen[run] = []
                self.listeScen[run].append((scen, date))

        if len(self.listeRuns) > 0:
            self.tree = self.ui.treeWidget

            self.parent = {}
            self.child = {}

            for run in self.listeRuns:
                self.parent[run] = QTreeWidgetItem(self.tree)
                self.parent[run].setText(0, run)
                self.parent[run].setFlags(self.parent[run].flags() |
                                          Qt.ItemIsTristate |
                                          Qt.ItemIsUserCheckable)

                lbl = QLabel('')
                self.tree.setItemWidget(self.parent[run], 2, lbl)

                self.child[run] = {}
                maxi = datetime(1900, 1, 1, 0, 0)
                if self.cond_com:
                    for scen, date, comments in self.listeScen[run]:
                        self.child[run][scen] = QTreeWidgetItem(
                            self.parent[run])
                        self.child[run][scen].setFlags(
                            self.child[run][scen].flags() |
                            Qt.ItemIsUserCheckable)
                        self.child[run][scen].setText(0, scen)

                        self.child[run][scen].setCheckState(0, Qt.Unchecked)

                        lbl = QLabel("{:%d/%m/%Y %H:%M}".format(date))
                        self.tree.setItemWidget(self.child[run][scen], 1, lbl)

                        maxi = max(maxi, date)
                        lbl = QLabel(comments)
                        self.tree.setItemWidget(self.child[run][scen], 2, lbl)
                else:
                    for scen, date in self.listeScen[run]:
                        self.child[run][scen] = QTreeWidgetItem(
                            self.parent[run])
                        self.child[run][scen].setFlags(
                            self.child[run][scen].flags() |
                            Qt.ItemIsUserCheckable)
                        self.child[run][scen].setText(0, scen)

                        self.child[run][scen].setCheckState(0, Qt.Unchecked)

                        lbl = QLabel("{:%d/%m/%Y %H:%M}".format(date))
                        self.tree.setItemWidget(self.child[run][scen], 1, lbl)

                        maxi = max(maxi, date)

                lbl = QLabel("{:%d/%m/%Y %H:%M}".format(maxi))
                self.tree.setItemWidget(self.parent[run], 1, lbl)
        else:
            self.ui.b_delete.setDisabled(True)

        self.ui.b_delete.clicked.connect(self.lancement)
        self.ui.b_cancel.clicked.connect(self.annule)

    def lancement(self):
        """ Delete selection function"""
        selection = {}
        for run in self.listeRuns:
            if self.parent[run].checkState(0) > 0:
                selection[run] = []
                if self.cond_com:
                    for scen, date, comments in self.listeScen[run]:
                        if self.child[run][scen].checkState(0) > 1:
                            selection[run].append("'{}'".format(scen))
                else:
                    for scen, date in self.listeScen[run]:
                        if self.child[run][scen].checkState(0) > 1:
                            selection[run].append("'{}'".format(scen))

        self.close()

        self.iface.messageBar().clearWidgets()
        progress_message_bar = self.iface.messageBar()
        progress = QProgressBar()
        progress.setMaximum(100)
        progress_message_bar.pushWidget(progress)

        n = len(selection.keys())
        ok = self.box.yes_no_q('Do you want to delete ?')

        if ok:
            for i, (run, scenarios) in enumerate(selection.items()):

                lst_tab = self.mdb.list_tables()
                sql = "run = '{0}' AND scenario IN ({1})".format(run,
                                                                 ",".join(
                                                                     scenarios))
                if "resultats" in lst_tab:
                    self.mdb.delete("resultats", sql)
                    self.mdb.delete("resultats_basin", sql)
                    self.mdb.delete("resultats_links", sql)

                self.mdb.delete("runs", sql)
                self.delete_useless_data()

                if self.mgis.DEBUG:
                    self.mgis.add_info(
                        "Deletion of {0} scenario for {1} is done".format(
                            scenarios, run))

                progress.setValue(i / float(n) * 100)

        self.iface.messageBar().clearWidgets()

    def annule(self):
        """"Cancel """
        self.close()

    def delete_useless_data(self):
        # delete var transport_pur
        sql = "DELETE FROM {0}.results_var WHERE id IN(" \
              "SELECT id FROM {0}.results_var " \
              "WHERE id  IN (SELECT DISTINCT var FROM {0}.results " \
              "WHERE id_runs IN (SELECT DISTINCT id_runs FROM {0}.results " \
              "WHERE id_runs NOT IN (SELECT id FROM {0}.runs ))) " \
              "and type_res= 'tracer_TRANSPORT_PUR');"
        self.mdb.run_query(sql.format(self.mdb.SCHEMA))

        # delete results
        sql = "DELETE  FROM {0}.results WHERE id_runs IN " \
              "(SELECT DISTINCT id_runs FROM {0}.results " \
              "where id_runs not in (SELECT id FROM {0}.runs));"
        self.mdb.run_query(sql.format(self.mdb.SCHEMA))

        # delete results_sect
        sql = "DELETE  FROM {0}.results_sect WHERE id_runs IN " \
              "(SELECT DISTINCT id_runs FROM {0}.results_sect " \
              "where id_runs not in (SELECT id FROM {0}.runs));"
        self.mdb.run_query(sql.format(self.mdb.SCHEMA))

        # delete run_graph
        sql = "DELETE  FROM {0}.runs_graph WHERE id_runs IN " \
              "(SELECT DISTINCT id_runs FROM {0}.runs_graph " \
              "where id_runs not in (SELECT id FROM {0}.runs));"
        self.mdb.run_query(sql.format(self.mdb.SCHEMA))
