revisal/legacy/revisal.py

'''
Created on Apr 27, 2020 @author: Levi
'''

import xml.etree.ElementTree as ET
import os
import zipfile
# import re
from collections import namedtuple

path = 'c:/Users/levi/Documents/DECEL/Revisal Hg500/Rapoarte/'
# working_dir = os.getcwd()
def unzip_rvs(path):
    rvs_files = list(filter(lambda f: f.endswith('.rvs'), os.listdir(path)))
    rvs_files_sorted = sorted(rvs_files, key=lambda x: os.stat(os.path.join(path, x)).st_mtime, reverse=True)
    rvs_file = rvs_files_sorted[0]

    with zipfile.ZipFile(os.path.join(path, rvs_file)) as zip_file:
        zip_file.extractall(path=path)

unzip_rvs(path=path)

xml_files = list(filter(lambda f: f.endswith('.xml'), os.listdir(path)))
xml_files_sorted = sorted(xml_files, key=lambda x: os.stat(os.path.join(path, x)).st_mtime, reverse=True)
print(xml_files_sorted[0])

# tree = ET.parse('S.C. MOZAIK CONSULTING SRL S.R.L. 17259191 Apr 30 2020 13_40_26.xml')
tree = ET.parse(f'{path}{xml_files_sorted[0]}')
root = tree.getroot()

keys = ['Adresa',
    'Cnp',
    'CnpVechi',
    'Cod',
    'DataConsemnare',
    'DataContract',
    'DataInceputContract',
    'DataSfarsitContract',
    'DateContractVechi',
    'Detalii',
    'DetaliiMutare',
    'ExceptieDataSfarsit',
    'NumarContract',
    'NumereContractVechi',
    'Radiat',
    'Salariu',
    'SporuriSalariu',
    'Spor',
    'IsProcent',
    'Tip',
    'Versiune',
    'Valoare',
    'DataIncetareDetasare',
    'DataIncetareSuspendare',
    'StarePrecedenta',
    'DataInceput',
    'DataIncetare',
    'DataSfarsit',
    'Explicatie',
    'TemeiLegal',
    'Durata',
    'IntervalTimp',
    'Norma',
    'Repartizare',
    'TipContract',
    'TipDurata',
    'TipNorma',
    'DetaliiSalariatStrain',
    'Localitate',
    'Mentiuni',
    'Nationalitate',
    'Nume',
    'Prenume',
    'TipActIdentitate']

# roots = ('{http://schemas.datacontract.org/2004/07/Revisal.Entities}Cor',
#          '{http://schemas.datacontract.org/2004/07/Revisal.Entities}StareCurenta',
#          '{http://schemas.datacontract.org/2004/07/Revisal.Entities}TimpMunca')
# d = []
# synt = re.search('(?<=})[A-Za-z]+', roots[0])
# print(synt.group(0))
Contract = namedtuple('Contract', ['codcor', 'datacontract', 'datainceput', 'datasfarsit', 'detalii', 'numar', 'salariu',
                                   'stare', 'inceputsusp', 'sfarsitsusp', 'incetaresusp', 'dataincetare', 'temei'], defaults=[0])

Revisal = namedtuple('Revisal', ['nume', 'prenume', 'adresa', 'cnp', 'nationalitate', 'tara', 'tipact', 'contracts'], defaults=[0])
contracts = []

#--------------------------------------------------------------------------------------------------------------------------------------
# def susp_contracts(nr, root):
#     global contracts
#     for idx in range(nr):
#         for child in root[1]:
#             try:
#                 if child[5][idx][15].attrib['{http://www.w3.org/2001/XMLSchema-instance}type'] == 'ContractStareSuspendare':
#                     contract = Contract(codcor = child[5][idx][1][0].text, datacontract=child[5][idx][3].text,
#                                         datainceput=child[5][idx][4].text,datasfarsit=child[5][idx][5].text,
#                                         detalii=child[5][idx][7].text, numar=child[5][idx][10].text, salariu=child[5][idx][13].text,
#                                         stare=child[5][idx][15].attrib['{http://www.w3.org/2001/XMLSchema-instance}type'],
#                                         inceputsusp=child[5][idx][15][3].text, sfarsitsusp=child[5][idx][15][5].text,
#                                         incetaresusp=child[5][idx][15][1].text, dataincetare=child[5][idx][15][4].text,
#                                         temei=child[5][idx][15].text)
#                     rvs = Revisal(nume=child[10].text, prenume = child[11].text, adresa=child[0].text, cnp=child[3].text,
#                               nationalitate=child[9][0].text, tara=child[12][0].text, tipact=child[13].text, contracts=contract)
#                     contracts.append(rvs)
#---------------------------------------------------------------------------------------------------------------------------------------

def get_revisal(deep=5):
    global root
    for idx in range(deep):
        for child in root[1]:
            try:
                if child[5][idx][15].attrib['{http://www.w3.org/2001/XMLSchema-instance}type'] == 'ContractStareSuspendare':
                    contract = Contract(codcor = child[5][idx][1][0].text, datacontract=child[5][idx][3].text,
                                        datainceput=child[5][idx][4].text,datasfarsit=child[5][idx][5].text,
                                        detalii=child[5][idx][7].text, numar=child[5][idx][10].text, salariu=child[5][idx][13].text,
                                        stare=child[5][idx][15].attrib['{http://www.w3.org/2001/XMLSchema-instance}type'],
                                        inceputsusp=child[5][idx][15][3].text, sfarsitsusp=child[5][idx][15][5].text,
                                        incetaresusp=child[5][idx][15][1].text, dataincetare=child[5][idx][15][4].text,
                                        temei=child[5][idx][15].text)
                    rvs = Revisal(nume=child[10].text, prenume = child[11].text, adresa=child[0].text, cnp=child[3].text,
                              nationalitate=child[9][0].text, tara=child[12][0].text, tipact=child[13].text, contracts=contract)
                    contracts.append(rvs)
                elif child[5][idx][15].attrib['{http://www.w3.org/2001/XMLSchema-instance}type'] == 'ContractStareIncetare':
                    contract = Contract(codcor=child[5][idx][1][0].text, datacontract=child[5][idx][3].text,
                                        datainceput=child[5][idx][4].text,datasfarsit=child[5][idx][5].text,
                                        detalii=child[5][idx][7].text, numar=child[5][idx][10].text, salariu=child[5][idx][13].text,
                                        stare=child[5][idx][15].attrib['{http://www.w3.org/2001/XMLSchema-instance}type'],
                                        dataincetare=child[5][idx][15][3].text, temei=child[5][idx][15][5].text,
                                        inceputsusp='', sfarsitsusp='', incetaresusp='')
                    rvs = Revisal(nume=child[10].text, prenume = child[11].text, adresa=child[0].text, cnp=child[3].text,
                              nationalitate=child[9][0].text, tara=child[12][0].text, tipact=child[13].text, contracts=contract)
                    contracts.append(rvs)
                elif child[5][idx][15].attrib['{http://www.w3.org/2001/XMLSchema-instance}type'] == 'ContractStareActiv':
                    contract = Contract(codcor=child[5][idx][1][0].text, datacontract=child[5][idx][3].text,
                                        datainceput=child[5][idx][4].text,datasfarsit=child[5][idx][5].text,
                                        detalii=child[5][idx][7].text, numar=child[5][idx][10].text, salariu=child[5][idx][13].text,
                                        stare=child[5][idx][15].attrib['{http://www.w3.org/2001/XMLSchema-instance}type'],
                                        inceputsusp='', sfarsitsusp='', incetaresusp='', dataincetare='')
                    rvs = Revisal(nume=child[10].text, prenume = child[11].text, adresa=child[0].text, cnp=child[3].text,
                              nationalitate=child[9][0].text, tara=child[12][0].text, tipact=child[13].text, contracts=contract)
                    contracts.append(rvs)
            except IndexError:
                continue
    return contracts

if __name__ == '__main__':
    for i in get_revisal(deep=5):
        print(i.nume, i.prenume)