from pycen import con_gn
import numpy as np
import pandas as pd

# Liste des CD_NOM en entrée
taxlist = pd.read_excel('/media/colas/SRV/FICHIERS/TRANSFERTS-EQUIPE/CG/BOCA_CD_nom.xlsx')

sql = """
    SELECT 
        t.cd_nom,
        t.cd_ref,
        t.regne,
        t.phylum,
        t.classe,
        t.ordre,
        t.famille,
        t.group1_inpn,
        t.group2_inpn,
        t.nom_vern,
        t.nom_complet,
        t.nom_valide,
        t.lb_nom,
        --s.*
        s.code_statut,
        s.cd_type_statut,
        s.label_statut
    FROM taxonomie.v_taxref_all_listes t
    JOIN taxonomie.v_bdc_status s USING (cd_nom)
    WHERE t.cd_nom IN {cd_nom}
;""".format(cd_nom = tuple(taxlist.CD_NOM.astype(str)) )
df = pd.read_sql_query(sql,con_gn)
df.to_csv('/media/colas/SRV/FICHIERS/TRANSFERTS-EQUIPE/LC/BOCA_CD_NOM_STATUS.csv')

pivot = pd.pivot_table(
    df,
    values='code_statut',
    index=['cd_nom', 'cd_ref','lb_nom'#,'niveau_admin','lb_adm_tr'
    ],
    columns=['cd_type_statut'],
    aggfunc=list,fill_value=None)

for c in pivot.columns:
    pivot[c] = [x[0] if x is not np.NaN and len(x)==1 else x for x in pivot[c]]
pivot['DH'] = [','.join(x) if (x is not np.NaN) and (len(x)==2) else x for x in pivot['DH']]
pivot.DH.replace({'CDH':''},regex=True,inplace=True)


pivlib = pd.pivot_table(
    df,
    values='label_statut',
    index=['cd_nom', 'cd_ref','lb_nom'#,'niveau_admin','lb_adm_tr'
    ],
    columns=['cd_type_statut'],
    aggfunc=list,fill_value=None)
for c in pivlib.columns:
    pivlib[c] = [x[0] if x is not np.NaN and len(x)==1 else x for x in pivlib[c]]
pivlib['DH'] = [','.join(x) if (x is not np.NaN) and (len(x)==2) else x for x in pivlib['DH']]
pivlib.DH.replace({'CDH':''},regex=True,inplace=True)


print('INIT writer')
NAME_OUT = '/media/colas/SRV/FICHIERS/TRANSFERTS-EQUIPE/LC/BOCA_CD_NOM_STATUS.xlsx'
with pd.ExcelWriter(NAME_OUT) as writer:
    df.to_excel(
        writer,sheet_name='v_bdc_status'
        )
    # writer.save()
    print('v_bdc_status OK !')
    pivot.to_excel(
        writer,sheet_name='pivot_table'
        )
    # writer.save()
    print('pivot_table OK !')
    pivlib.to_excel(
        writer,sheet_name='pivot_libel'
        )
    writer.save()
    print('pivot_libel OK !')