JanusGraphAPI/PrepareZipCodeDB.py at main · justin-napolitano/JanusGraphAPI · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
#prepare_zipcode_db.py
from PandasFunctions import PandasFunctions as PF
from pprint import pprint
import json

def get_file_buffer():
    file_path = 'uscities.csv'
    file_buffer = open(file_path, "r", encoding="utf-8")
    return file_buffer

def get_df_from_csv(file_buffer = 'zip.csv'):
    zip_code_df = PF.Load.csv_to_df(file_buffer)
    pprint(zip_code_df)
    return zip_code_df


def return_zip_code_df_to_zip_code_dict(zip_code_df):
    output_list = []
    zip_code_property_list = PF.Transform.df_to_record_dict(df= zip_code_df,orient = 'records')

    for dictionary in zip_code_property_list:
        dict_string = json.dumps(dictionary)
        output_list.append(dict_string)


    pprint(output_list[0])
    #zip_code_property_list
    return output_list

def add_property_column_to_the_zip_code_df(df, zip_code_property_list):
    df['vertice_property'] = zip_code_property_list
    return df

def write_zip_code_df_to_file(zip_code_df):
    PF.Write.data_frame_to_csv(zip_code_df,'zip_code_db.csv')
    return True


def main():
    #connection_driver = get_janus_graph_connection_driver()
    #traversal = get_janus_graph_traversal(connection_driver)
    #zip_code_csv_file_buffer = get_file_buffer()
    zip_code_df = get_df_from_csv()
    zip_code_property_list = return_zip_code_df_to_zip_code_dict(zip_code_df)
    zip_code_df = add_property_column_to_the_zip_code_df(df = zip_code_df,zip_code_property_list=zip_code_property_list)
    #pprint(zip_code_df)
    written_to_csv = write_zip_code_df_to_file(zip_code_df)


if __name__ == "__main__":
    main()