scripts-pde/pda2suiviactivite.py

'''

    Script d'export des données de configuration des applications SuiviDesActivtes et SuiviActivitePC

    Les données d'origine sont issues de la base \\h2o\LOCAL\4-transversal\BDD\mdb\PDA\db_PDA.mdb

    Les fichiers résultats sont les fichiers *.xml du répertoire \\h2o\LOCAL\4-transversal\BDD\mdb\PDA\Fichiers_PDA\

'''
import logging

from lxml import builder  # @UnresolvedImport
from lxml import etree  # @UnresolvedImport
from path import Path  # @UnusedImport

from core import logconf
from core.pde import PdaDb, PDA_FILES_DEST


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


# # POUR TESTER, décommenter les lignes suivantes
##-----------------------------------------------

# PDA_FILES_DEST = Path(r"\\h2o\local\4-transversal\BDD\mdb_test\PDA\Fichiers_PDA")
# PdaDb._path = Path(r"\\h2o\local\4-transversal\BDD\mdb_test\PDA\db_PDA.mdb")
# logger.handlers = [h for h in logger.handlers if (type(h) == logging.StreamHandler)]
# logger.warning("Mode TEST")

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

def main():
    logger.info("Initialization")

    # Connect to db_PDA.mdb
    pda_db = PdaDb(autocommit=True)

    class PdaXmlFile():
        def __init__(self):
            self.source_table = ""
            self.fields_mapping = {}
            self.root_node = ""
            self.item_node = ""
            self.file_name = ""
            self.order_by = []
            self.nsmap = {'xsd': "http://www.w3.org/2001/XMLSchema",
                          'xsi': "http://www.w3.org/2001/XMLSchema-instance"}

        def create(self):
            maker = builder.ElementMaker(nsmap=self.nsmap)

            nodes = []

            sql = "SELECT {mapping} FROM {tbl} ORDER BY {order}".format(
                                                                    mapping=",".join(["[{}].[{}] as {}".format(self.source_table, key, val) for key, val in self.fields_mapping.items()]),
                                                                    tbl=self.source_table,
                                                                    order=",".join(self.order_by)
                                                                    )
            data = [record._asdict() for record in pda_db.read_all(sql)]

            for record in data:
                node = maker.__call__(self.item_node, *[maker.__call__(field, str(value)) for field, value in record.items()])
                nodes.append(node)

            root = maker.__call__(self.root_node, *nodes)

            with open(PDA_FILES_DEST / self.file_name, "wb") as f:
                f.write(etree.tostring(root, xml_declaration=True, encoding='utf-8', pretty_print=True))
            logger.info("> Créé: {}".format(self.file_name))


    xml = PdaXmlFile()
    xml.source_table = "pdaEngin"
    xml.fields_mapping = {"strEnginId": "Id", "strEnginLibelleLong": "Nom"}
    xml.root_node = "ArrayOfAttelage"
    xml.item_node = "Attelage"
    xml.file_name = "attelages.xml"
    xml.order_by = ["strEnginLibelleLong"]
    xml.create()

    xml = PdaXmlFile()
    xml.source_table = "pdaorigine"
    xml.fields_mapping = {"lngorigine": "Id", "strorigine": "Nom"}
    xml.root_node = "ArrayOfDepart"
    xml.item_node = "Depart"
    xml.file_name = "depart.xml"
    xml.order_by = ["strorigine"]
    xml.create()

    xml = PdaXmlFile()
    xml.source_table = "pdaHeures"
    xml.fields_mapping = {"annee": "annee", "mois": "mois", "heures": "heures"}
    xml.root_node = "dataroot"
    xml.item_node = "pdaHeures"
    xml.file_name = "heures.xml"
    xml.order_by = ["annee", "mois"]
    xml.create()

    xml = PdaXmlFile()
    xml.source_table = "pdaEquip"
    xml.fields_mapping = {"strEquipesId": "Id", "strEquipesLibelle": "Nom"}
    xml.root_node = "ArrayOfEquipe"
    xml.item_node = "Equipe"
    xml.file_name = "equipes.xml"
    xml.order_by = ["strEquipesLibelle"]
    xml.create()

    xml = PdaXmlFile()
    xml.source_table = "pdalocalisation"
    xml.fields_mapping = {"lngTiersId": "Id", "strTiersMnemo": "Nom"}
    xml.root_node = "ArrayOfLocalisation"
    xml.item_node = "Localisation"
    xml.file_name = "localisation.xml"
    xml.order_by = ["strTiersMnemo"]
    xml.create()

    xml = PdaXmlFile()
    xml.source_table = "pdaNatInterv"
    xml.fields_mapping = {"strCategorieInterventioinId": "Id", "strCategorieInterventioinLibelle": "Nom"}
    xml.root_node = "ArrayOfNatureRealisation"
    xml.item_node = "NatureRealisation"
    xml.file_name = "naturesinterventions.xml"
    xml.order_by = ["strCategorieInterventioinLibelle"]
    xml.create()


if __name__ == "__main__":
    main()
    logger.info("-- Fin --")