Skip to content

Commit a55a702

Browse files
fix : minor edit in functions
1 parent f703876 commit a55a702

2 files changed

Lines changed: 27 additions & 15 deletions

File tree

csv2vcf/VCF.py

Lines changed: 27 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,8 @@
33
from tkinter.filedialog import askopenfilename
44
from tkinter import messagebox
55

6+
name_dict={}
7+
unknown_index=0
68
def zero_insert(input_string):
79
'''
810
This function get a string as input if input is one digit add a zero
@@ -75,12 +77,34 @@ def VCF_creator(folder_name,first_name,last_name,tel_mobile,tel_home,tel_work,em
7577
VCF_adr(file,adr_work,adr_home)
7678
VCF_website(file,website_url)
7779

80+
def name_dict_update(name):
81+
global name_dict
82+
if name not in name_dict.keys():
83+
name_dict[name] = 0
84+
else:
85+
name_dict[name] = name_dict[name] + 1
86+
87+
def VCF_write(temp,name_dict,foldername):
88+
name = temp[0] + "," + temp[1]
89+
name_dict_update(name)
90+
global unknown_index
91+
if len(temp[0]) == 0 and len(temp[1]) == 0:
92+
unknown_index += 1
93+
VCF_creator(foldername, str(unknown_index),"Unknown ", temp[2], temp[3], temp[4], temp[5], temp[6], temp[7],
94+
temp[8], temp[9], temp[10])
95+
else:
96+
if name_dict[name] != 0:
97+
VCF_creator(foldername, temp[0] + "_" + str(name_dict[name]), temp[1], temp[2], temp[3], temp[4], temp[5],
98+
temp[6], temp[7], temp[8], temp[9], temp[10])
99+
else:
100+
VCF_creator(foldername, temp[0], temp[1], temp[2], temp[3], temp[4], temp[5], temp[6], temp[7], temp[8],
101+
temp[9], temp[10])
78102
def csv_reader(file_name):
79103
try:
80104
file=open(file_name,"r")
81-
unknown_index=0
105+
82106
vcf_counter=0
83-
name_dict={}
107+
84108
foldername=VCF_Folder(file_name)
85109
for index,line in enumerate(file):
86110
if index>0:
@@ -90,19 +114,7 @@ def csv_reader(file_name):
90114
print("[Warning] CSV File Line "+str(index)+" Bad Format")
91115
continue
92116
else:
93-
name=temp[0]+","+temp[1]
94-
if name not in name_dict.keys():
95-
name_dict[name]=0
96-
else:
97-
name_dict[name]=name_dict[name]+1
98-
if len(temp[0])==0 and len(temp[1])==0:
99-
unknown_index+=1
100-
VCF_creator(foldername,"Unknown ",str(unknown_index),temp[2],temp[3],temp[4],temp[5],temp[6],temp[7],temp[8],temp[9],temp[10])
101-
else:
102-
if name_dict[name]!=0:
103-
VCF_creator(foldername,temp[0]+"_"+str(name_dict[name]),temp[1],temp[2],temp[3],temp[4],temp[5],temp[6],temp[7],temp[8],temp[9],temp[10])
104-
else:
105-
VCF_creator(foldername, temp[0], temp[1], temp[2], temp[3],temp[4], temp[5], temp[6], temp[7], temp[8], temp[9], temp[10])
117+
VCF_write(temp,name_dict,foldername)
106118
vcf_counter+=1
107119
return vcf_counter
108120

4.88 KB
Binary file not shown.

0 commit comments

Comments
 (0)