Newer
Older
web-services / affiliations-tools / v1 / corporate / abreviations.py
import csv
import pickle

# fonction : création du dictionnaire depuis un CSV : colonne 1 = key et colonne 2 = value
def transform_to_dict(csv_file):
    my_dict = {}
    with open(csv_file, 'r') as file:
        f = csv.reader(file, delimiter='\t')

        for row in f:
            key = row[0].lower().strip()
            value = row[1].lower().strip()
            my_dict[key] = value

    return my_dict

# fonction : sauvegarde du dictionnaire dans un pickle
def save_dict_to_pickle(my_dict, my_pickel):
    with open(my_pickel, 'wb') as file:
        pickle.dump(my_dict, file)

# appel des fonctions 
fichier_csv = 'abreviations.csv'
resultat = transform_to_dict(fichier_csv)

fichier_pickle = 'abreviations.pickle'
save_dict_to_pickle(resultat, fichier_pickle)