scripts-pde/wincan2ctrl.py

'''
    Met à jour la base controle avec les données d'un chantier Wincan

    usage: wincan2ctrl [-c <idchantier>] [-i <inspectioname>] [-o]

    Options:
      -c <idchantier>         Passe le code chantier
      -i <inspectioname>      Passe le nom de l'inpection wincan
      -o                      Ne demande pas confirmation avant de lancer le traitement
      -h, --help              Affiche l'aide

    Attention: Si un des arguments comporte des espaces, pensez à le mettre entre guillemets

    @olivier.massot, mai 2018
'''
import logging
import sys

from docopt import docopt
from path import Path

from core import logconf
from core.pde import ControlesDb, WincanDb, InspectionTronconWincan, \
    InterventionITV, PiracaDb
from core.sqlformatter import SqlFormatter


logger = logging.getLogger("wincan2ctrl")
logconf.start("wincan2ctrl", logging.DEBUG)


# # POUR TESTER, décommenter les lignes suivantes
# > Lancer le script /resources/test_wincan2ctrl.py pour reinitialiser les données de la base de test
##-----------------------------------------------

ControlesDb._path = Path(r"\\h2o\local\4-transversal\BDD\mdb_test\cg67Parc_data.mdb")
WincanDb._path = Path(r"\\h2o\local\4-transversal\BDD\mdb_test\Wincan\parc_2007\DB\PARC_2007.mdb")
PiracaDb._path = Path(r"\\h2o\local\4-transversal\BDD\mdb_test\db_Piraca.mdb")
logger.handlers = [h for h in logger.handlers if (type(h) == logging.StreamHandler)]
logger.warning("<<<<<<<<<<<<<<   Mode TEST   >>>>>>>>>>>>>>>>>")

##-----------------------------------------------

logger.info("Initialisation...")

Sql = SqlFormatter()

# Connexion à Controles
controles_db = ControlesDb(autocommit=False)

# Connexion à Wincan
wincan_db = WincanDb(autocommit=False)

# Connexion à CommunDb
piraca_db = PiracaDb(autocommit=False)


def get_lib(lib_id, tbl_id):
    row = piraca_db.first(Sql.format("""SELECT LIB_ID, LIB_LIB, TLB_ID
                                            FROM LISTE_LIBELLE_WINCAN
                                            WHERE TLB_ID='{}' AND LIB_ID='{}'""", tbl_id, lib_id))
    if not row:
        logger.error("Aucun libelle trouvé dans la table LISTE_LIBELLE_WINCAN pour %s, id=%s", tbl_id, lib_id)
        raise KeyError()
    return row.LIB_LIB


def get_materiau(wincan_id):
    return controles_db.first(Sql.format("""SELECT strMateriauCourt
                                            FROM tblMateriaux
                                            WHERE strWincanId='{}'""", wincan_id)).strMateriauCourt

# Parse les arguments
args = docopt(__doc__, help=False)
chantier_id = args["-c"] if args["-c"] else 0
inspname = args["-i"] if args["-i"] else ""

# Demande à l'utilisateur de saisir les informations qui n'ont pas été passées en argument
if not inspname:
    # Propose une liste d'inspections possibles
    sql = """SELECT SI_InspName
                FROM SI_T
                WHERE SI_Spare1 Is Null OR Len([SI_Spare1])=0
                GROUP BY SI_InspName
            """
    candidats = wincan_db.read_all(sql)

    print("Veuillez choisir une inspection Wincan à traiter en saisissant son numéro:")
    for i, candidat in enumerate(candidats):
        print("[{}] - {}".format(i, candidat.SI_InspName))

    while not inspname:
        reponse = input("> ")
        try:
            inspname = candidats[int(reponse)].SI_InspName
        except ValueError:
            print("Valeur invalide!")

if not chantier_id:
    try:
        default = int(inspname[:6])
        chantier_id = input("Veuillez saisir le code chantier [{}]: ".format(default))
        if not chantier_id:
            chantier_id = default
    except ValueError:
        chantier_id = input("Veuillez saisir le code chantier: ")
        if not chantier_id:
            logger.error("Code chantier invalide")
            sys.exit(1)

# Calcul du numéro d'intervention
last_interv_id = controles_db.first("SELECT max(bytIntervId) as last_interv_id FROM tblVideoIntervs WHERE lngChantierId={}".format(chantier_id)).last_interv_id
interv_id = last_interv_id + 1 if last_interv_id else 1

# Récupération du numéro de commande en cours
commande_id = controles_db.first("SELECT bytCommandeId FROM tblVideoBases WHERE lngChantierId={}".format(chantier_id)).bytCommandeId


# Affiche les infos et demande confirmation avant ed lancer le traitement
logger.info("## Traitement de l'inspection Wincan")
logger.info("Nom chantier: %s", inspname)
logger.info("Code chantier: %s", chantier_id)
logger.info("Numero d'intervention: %s", interv_id)
logger.info("Numero de commande: %s", commande_id)
if not args["-o"]:
    if not input("Voulez-vous continuer? (o/n)") == 'o':
        if input("Etes-vous sûr de vouloir annuler l'opération? (o/n)") == 'o':
            logger.info("Opération annulée par l'utilisateur")
            sys.exit(1)


# Recuperation des données de l'intervention
sql = """SELECT SI_T.SI_InspName, Sum(S_T.S_Sectionlength) AS Long_insp, Min(SI_T.SI_Date) AS mindate, Max(SI_T.SI_Date) AS maxdate,
        First(SI_T.SI_Operator) AS Equipe, First(SI_T.SI_Vehicle) AS Mat, First(SI_T.SI_InspMethod) AS InspMethod,
        First(SI_T.SI_ReasonOfInspection) AS SI_ReasonOfInspection, First(S_T.S_SectionPurpose) AS nature,
        First(S_T.S_SectionUse) AS Fonction, First(S_T.S_SectionType) AS Type
        FROM S_T INNER JOIN SI_T ON S_T.S_ID = SI_T.SI_Section_ID
        WHERE (((SI_T.[SI_InspName])='{}'))
        GROUP BY SI_T.SI_InspName
        HAVING Sum(S_T.S_Sectionlength) Is Not Null AND First(SI_T.SI_Operator) Is Not Null AND First(SI_T.SI_Vehicle) Is Not Null AND First(SI_T.SI_InspMethod) Is Not Null
        """.format(inspname)
try:
    inspection = next(wincan_db.read(sql))
except StopIteration:
    logger.error("Aucune inspection trouvée pour le nom '%s'", inspname)
    raise

# Insère l'inspection dans la table tblVideoIntervs de Controles
logger.info("Création de l'intervention dans tblVideoIntervs")

interv = InterventionITV()

interv.lngChantierId = chantier_id
interv.bytIntervId = interv_id
interv.dtmIntervDu = inspection.mindate
interv.dtmIntervAu = inspection.maxdate
interv.strEquipeId = inspection.Equipe
interv.intlMaterielID = inspection.Mat
interv.bytCommandeId = commande_id
interv.lngTroncon = inspection.Long_insp
interv.SI_InspMethod = inspection.InspMethod
interv.SI_ReasonOfInspection = get_lib(inspection.SI_ReasonOfInspection, "SI_REASONOfINSPECTION")

sql = Sql.format("""INSERT INTO tblVideoIntervs ( lngChantierId, bytIntervId, dtmIntervDu, dtmIntervAu, strEquipeId,
                                       intlMaterielID, bytCommandeId, lngTroncon, SI_InspMethod, SI_ReasonOfInspection )
                    VALUES ({interv.lngChantierId}, {interv.bytIntervId}, {interv.dtmIntervDu:date}, {interv.dtmIntervAu:date}, {interv.strEquipeId:text},
                            {interv.intlMaterielID}, {interv.bytCommandeId}, {interv.lngTroncon}, {interv.SI_InspMethod:text}, {interv.SI_ReasonOfInspection:text})
                    """, interv=interv)
controles_db.execute(sql)

# Met a jour les champs SI_Spare1 et SI_Spare2 de la table Wincan en retour
logger.info("Mise à jour en retour de SI_T")
sql = """UPDATE SI_T
         SET SI_Spare1='{}', SI_Spare2='{}'
         WHERE SI_InspName='{}'
         """.format(chantier_id, interv_id, inspection.SI_InspName)
wincan_db.execute(sql)

# Met a jour la table tbl_so_rate de Controles
logger.info("Traitement des inspections")

# Extrait les données des tronçons

sql = """SELECT SI_T.SI_ID, S_T.S_ID, SI_T.SI_InspName, SI_T.SI_Spare1, SI_T.SI_Spare2, SI_T.SI_AutoNumber, SI_T.SI_MediaNumber1, S_T.S_SectionFlow,
            S_T.S_StartNode, S_T.S_EndNode, S_T.S_StartNodeType, S_T.S_EndNodeType, S_T.S_StartNodeCoord_Z, S_T.S_EndNodeCoord_Z,
            S_T.S_Sectionlength, S_T.S_SectionPurpose, S_T.S_SectionUse, S_T.S_SectionType, S_T.S_PipeMaterial, S_T.S_YearLayed,
            S_T.S_PipeDia, S_T.S_Situation, S_T.S_Spare1, S_T.S_PipeShape, S_T.S_Pipelength, S_T.S_Spare3,
            SI_T.SI_Spare0, SI_T.SI_Date
        FROM S_T INNER JOIN SI_T ON S_T.S_ID = SI_T.SI_Section_ID
        WHERE SI_T.SI_Spare1='{}' AND SI_T.SI_Spare2='{}'
        """.format(chantier_id, interv_id)

for data in wincan_db.read(sql):

    logger.info("* Traitement de %s (S_ID: %s)", data.SI_InspName, data.S_ID)

    inspection = InspectionTronconWincan()

    inspection.s_guid = data.S_ID[2:-1] if (data.S_ID[:2] == "b'" and data.S_ID[-1:] == "'") else data.S_ID
    inspection.si_guid = data.SI_ID[2:-1] if (data.SI_ID[:2] == "b'" and data.SI_ID[-1:] == "'") else data.SI_ID
    inspection.nom_chantier = data.SI_InspName
    inspection.lng_chantier_id = data.SI_Spare1
    inspection.byt_interv_id = data.SI_Spare2
    inspection.si_autonumber = data.SI_AutoNumber
    inspection.classement_troncons = data.SI_MediaNumber1
    inspection.nom_troncon = "{r1}-{r2}".format(r1=data.S_StartNode, r2=data.S_EndNode) if data.S_SectionFlow == '2' else "{r2}-{r1}".format(r1=data.S_StartNode, r2=data.S_EndNode)
    inspection.startnode_type = get_lib(data.S_StartNodeType, "S_StartNodeType")
    inspection.endnode_type = get_lib(data.S_StartNodeType, "S_EndNodeType")
    inspection.sens_ecoul = ">>" if data.S_SectionFlow == '1' else ('<<' if data.S_SectionFlow == '2' else '')
    inspection.startnode_z = data.S_StartNodeCoord_Z
    inspection.endnode_z = data.S_EndNodeCoord_Z
    inspection.section_length = data.S_Sectionlength
    inspection.section_purpose = data.S_SectionPurpose
    inspection.section_use = data.S_SectionUse
    inspection.section_type = data.S_SectionType
    inspection.materiau = get_materiau(data.S_PipeMaterial)
    inspection.annee_pose = data.S_YearLayed
    inspection.diametre = get_lib(int(data.S_PipeDia), "S_PipeDia")
    inspection.route = get_lib(data.S_Situation, "S_Situation")
    inspection.n_route = data.S_Spare1
    inspection.pipe_shape = data.S_PipeShape
    inspection.pipe_length = data.S_Pipelength
    if data.S_Spare3:
        inspection.arbres = get_lib(data.S_Spare3, "S_Spare3")
    inspection.test_ecoulement = data.SI_Spare0
    inspection.si_date = data.SI_Date

    sql = """ SELECT SO_T.SO_ID, SO_T.SO_Rate, SO_T.SO_Photonumber1, SO_T.SO_Photonumber2
                FROM SO_T
                WHERE SO_T.SO_Inspecs_ID='{}'
            """.format(inspection.si_guid)

    # Parcours les opérations réalisées au cours de l'inspection du tronçon
    for opdata in wincan_db.read(sql):
        inspection.nb_ops += 1

        if opdata.SO_Rate == 1:
            inspection.rate_1 = inspection.rate_1 + 1 if inspection.rate_1 else 1
        elif opdata.SO_Rate == 2:
            inspection.rate_2 = inspection.rate_2 + 1 if inspection.rate_2 else 1
        elif opdata.SO_Rate == 3:
            inspection.rate_3 = inspection.rate_3 + 1 if inspection.rate_3 else 1
        elif opdata.SO_Rate == 4:
            inspection.rate_4 = inspection.rate_4 + 1 if inspection.rate_4 else 1
        elif opdata.SO_Rate == 5:
            inspection.rate_5 = inspection.rate_5 + 1 if inspection.rate_5 else 1
        elif not opdata.SO_Rate:
            pass
        elif opdata.SO_Rate >= 6:
            logger.error("Attention: une valeur de [SO_Rate] supérieure à 5 a été enregistrée (SO_ID: %s)", opdata.SO_ID)

        if opdata.SO_Photonumber1:
            inspection.nb_photos += 1
        if opdata.SO_Photonumber2:
            inspection.nb_photos += 1

    if not any([inspection.rate_1, inspection.rate_2, inspection.rate_3, inspection.rate_4, inspection.rate_5]):
        inspection.DG = 'ABS'  # Absence de défauts
    elif not any([inspection.rate_1, inspection.rate_2, inspection.rate_3]):
        inspection.DG = 'ACC'  # Défauts acceptables
    else:
        inspection.DG = 'INT'  # Défauts non-acceptables

    logger.info("\t- Mise à jour de tblso_Rate_Analyse")
    sql = Sql.format("""INSERT INTO tblso_Rate_Analyse (
                    lngChantierId, bytIntervId, SI_InspName, SI_AutoNumber,
                    [Classement tronons], Nom_troncon, S_StartNodeType,
                    Sens_ecoul, S_EndNodeType, S_PipeShape, MateriauCourt,
                    SI_Date, nb_Arbres, ANNEE_POSE,
                    Route, NRoute, Test_ecoulement, MaxDeS_StartNodeCoord_Z, MaxDeS_EndNodeCoord_Z,
                    MaxDeS_Sectionlength, MaxDeS_Pipelength, MaxDeDiametre, cpt_Photos, [Total de SO_ID],
                    1, 2, 3, 4, 5, DG
                    )
            VALUES ({inspection.lng_chantier_id}, {inspection.byt_interv_id}, {inspection.nom_chantier:text}, {inspection.si_autonumber},
                    {inspection.classement_troncons:text}, {inspection.nom_troncon:text}, {inspection.startnode_type:text},
                    {inspection.sens_ecoul:text}, {inspection.endnode_type:text}, {inspection.pipe_shape:text}, {inspection.materiau:text},
                    {inspection.si_date:date}, {inspection.arbres:text}, {inspection.annee_pose},
                    {inspection.route:text}, {inspection.n_route:text}, {inspection.test_ecoulement:text}, {inspection.startnode_z}, {inspection.endnode_z},
                    {inspection.section_length}, {inspection.pipe_length}, {inspection.diametre:text}, {inspection.nb_photos}, {inspection.nb_ops},
                    {inspection.rate_1}, {inspection.rate_2}, {inspection.rate_3}, {inspection.rate_4}, {inspection.rate_5}, {inspection.DG:text}
                    )
            """, inspection=inspection)
    controles_db.execute(sql)

    # Met à jour tblvideointervs.strResGlobal avec le resultat global
    #  le resultat global vaut '-' si un de ces trois champs n'est pas nul: tbl_so_rate.1, tbl_so_rate.2, tbl_so_rate.3
    # >> On peut peut-être rassembler cette partie et l'insertion dans cette table au debut?
    logger.info("\t- Mise à jour du resultat global dans tblVideoIntervs")
    sql = """UPDATE tblVideoIntervs
             SET strResGlobal='{}'
             WHERE lngChantierId= {} AND bytIntervId= {}
            """.format('-' if inspection.DG == 'INT' else '+',
                       chantier_id,
                       interv_id)
    controles_db.execute(sql)

    # Met à jour la table tblVideoBases pour marquer le chantier comme traité
    logger.info("\t- Mise à jour de tblVideoBases")
    sql = """UPDATE tblVideoBases
             SET blnWincan=True,bytNbInterv={}
             WHERE lngChantierId={}
             """.format(inspection.byt_interv_id,
                        inspection.lng_chantier_id)
    controles_db.execute(sql)

    # Met à jour les données du réseau dans tblChantiers
    logger.info("\t- Mise à jour des données du réseau dans tblChantiers")
    sql = """UPDATE tblChantiers
             SET  bytFoncReseauId ={} , bytNatureReseauId={}, bytTypeReseauId={}
             WHERE lngChantierId={}
             """.format(inspection.section_use,
                        inspection.section_purpose,
                        inspection.section_type,
                        inspection.lng_chantier_id)
    controles_db.execute(sql)



logger.info("Commit des modifications")
controles_db.commit()
wincan_db.commit()