Fix a memory leak in the "no imap" account prefs label.
[claws.git] / tools / vcard2xml.py
1 #!/usr/bin/env python
2 # -*- coding: latin-1 -*-
3 """
4
5 Copyright © 2003 Bogdan Sumanariu <zarrok@yahoo.com>
6
7   This file is free software; you can redistribute it and/or modify it
8   under the terms of the GNU General Public License as published by
9   the Free Software Foundation; either version 3 of the License, or
10   (at your option) any later version.
11    
12   This program is distributed in the hope that it will be useful, but
13   WITHOUT ANY WARRANTY; without even the implied warranty of
14   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15   General Public License for more details.
16
17   You should have received a copy of the GNU General Public License
18   along with this program; if not, write to the Free Software
19   Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
20
21   script name : evolutionvcard2claws.py
22
23  script purpose : convert an evolution addressbook VCARD file 
24  into a Claws Mail addressbook
25
26  tested with evolution 1.2.x, and 1.4.x
27
28 """
29
30 import string
31 import sys
32 import time
33 import os
34 import StringIO
35
36 keywds = ('x-evolution-file-as','fn', 'n','email;internet','nickname', 'url', 'org')
37
38 def normalizeLongLines(file):
39         """
40         Skip line breaks after 72 chars
41         """
42         buf = ''
43         
44         line = file.readline()
45         while line:
46                 if line[0] == ' ':
47                         buf = buf.rstrip('\n')
48                         line = line.lstrip();
49                         buf += line
50                 else:
51                         buf += line
52                 line = file.readline()
53         
54         return buf
55
56 def getEmailAddress(vcard):
57         """
58         Get email address.
59         Supported formats:
60                 - email;something
61                 - email;type=something
62         something := (internet,work,home, other)
63         """
64         
65         for key in vcard:
66                 items = key.split(';')
67                 if len(items) == 2:
68                         if items[0].lower() == 'email':
69                                 list = vcard[key]
70                                 return list[0]
71                 else:
72                         if key.lower() == 'email':
73                                 list = vcard[key]
74                                 return list[0]
75         
76         return ""
77
78 def findName(vcard):
79         """
80         Find a version 3.0 name
81         """
82         for key in vcard:
83                 items = key.split(';')
84                 if len(items) == 2:
85                         if items[0].lower() == 'n':
86                                 return  vcard[key]
87                 else:
88                         if key.lower() == 'n':
89                                 return  vcard[key]
90         
91         return None
92
93 ################################################################################
94 ##  reads a vcard and stores as hash pairs key/value where value is a list    ##
95 ################################################################################
96
97 def readVCARD (buffer) :
98
99         """
100
101         skips fom <file> until a 'begin' tag from VCARD is encountered.
102         from this point starts constructing a map (key, [values] ) 
103         VCARD entry format -> tag:value 
104         
105                 key <- tag
106                 [values] <- list with the values of <tag> if there are more tags with the same name
107
108         """
109         r=' '
110         bgn,end = -1, -1;
111         d = dict()
112         while r and bgn < 0 :
113                 r = buffer.readline()
114                 if len (r)  == 0 : return dict()
115                 if string.find('begin',string.lower(string.strip(r))) :
116                         bgn = 1
117         while r and end < 0 :
118                 r = buffer.readline()
119                 s = string.split(string.lower(string.strip(r)),':')
120                 if s[0] <> '' :
121                         if d.has_key(s[0]) :
122                                 d[s[0]].append(s[1])
123                         elif len(s) > 1:
124                                 d[s[0]] = [s[1]]        
125                         else :
126                                 d[s[0]] = ['']
127                         if s[0] == 'end' : end = 1      
128         return d
129
130 ##################################################################################
131
132 ###############################################################################################
133 ## writes on a given file an xml representation for claws-mail addressbook received as a hash  ##
134 ###############################################################################################
135
136 def writeXMLREPR (vcard,file,uid) :
137
138         """
139         based on <vcard> and <uid> writes only recognized tags (the ones defined in <keywds> list)
140         NOTE: <url> and <org> tag will be written as attributes (there are such tags in claws-mail's
141               XML schema)
142         """
143         if len (vcard.keys()) == 0 : return
144         item = vcard.get(keywds[2]);
145         if item:
146                 name = string.split(item[0],';')
147         else:
148                 """ version 3.0 n ?"""
149                 name = findName(vcard)
150                 if not name:
151                         return
152
153         fn, ln, nick, cn, a = '', '', '', '', ''
154
155         if len(name) >= 2 :
156                 fn = name[0]
157                 ln = name[1]
158         elif len(name) ==1 :
159                 fn = name[0]
160         
161         if vcard.has_key(keywds[4]) :
162                 nick = vcard.get(keywds[4])[0]
163         if len(vcard.get(keywds[1])[0]) :
164                 cn = vcard.get(keywds[1])[0]
165         else :
166                 cn = vcard.get(keywds[0])[0];
167
168         a += str('\n<person uid=\"' + str(uid[0]) + '\" first-name=\"' + fn + '\" last-name=\"' + ln
169                 + '\" nick-name=\"' + nick + '\" cn=\"' + cn + '\" >\n')
170         a += '\t<address-list>\n'
171         if vcard.get(keywds[3]) :
172                 for c in vcard.get(keywds[3]) :
173                         uid[0] = uid[0] + 1
174                         a += '\t\t<address uid=\"' + str(uid[0]) + '\" alias=\"' + nick  + '\" email=\"' + c + '\" remarks=\"\" />\n'
175         else :
176                 email = getEmailAddress(vcard)
177                 uid[0] = uid[0]+1
178                 a += '\t\t<address uid=\"' + str(uid[0]) + '\" alias=\"' +  nick + '\" email=\"' + email + '\" remarks=\"\" />\n'
179         a += '\t</address-list>\n'
180         a += '\t<attribute-list>\n'
181         for key in keywds[5:] :
182                 if vcard.get(key) :
183                         for c in vcard.get(key) :
184                                 uid[0] = uid[0] + 1
185                                 a += '\t\t<attribute uid=\"' + str(uid[0]) + '\" name=\"' + key +'\">'+c+'</attribute>\n'
186         a += '\t</attribute-list>\n'
187         a += '</person>\n'
188         file.write(a)
189         file.flush()
190                 
191 ###################################################################################################
192
193 def convert (in_f, o_f, name='INBOX') :
194         d = {'d':1}
195         uid = [int(time.time())]
196         
197         try :
198                 print 'proccessing...\n'
199                 o_f.write('<?xml version="1.0" encoding="ISO-8859-1" ?>\n<address-book name="'+name+'" >\n');
200                 
201                 buf = normalizeLongLines(in_f)
202                 buffer = StringIO.StringIO(buf)
203                 while len(d.keys()) > 0 :
204                         d = readVCARD(buffer)
205                         writeXMLREPR (d, o_f, uid)
206                         uid[0] = uid [0]+1
207                 
208                 o_f.write('\n</address-book>')
209                 print 'finished processing...\n'
210         except IOError, err :
211                 print 'Caught an IOError : ',err,'\t ABORTING!!!'
212                 raise err
213
214 #################################################################################################
215
216 def execute () :
217         if len(sys.argv) <> 3 and len(sys.argv) <> 2 :
218                 print str("\nUsage: vcard2xml.py  source_file [destination_file]\n\n" +
219                 '\tWhen only <source_file> is specified will overwrite the existing addressbook.\n'+
220                 '\tWhen both arguments are suplied will create a new additional addressbook named \n\tas the destination file.'+'\n\tNOTE: in both cases the Claws Mail must be closed and ran at least once.\n\n')
221                 sys.exit(1)
222
223         in_file = None
224         out_file = None
225         path_to_out = os.environ['HOME']+'/.claws-mail/addrbook/'
226         adr_idx = 'addrbook--index.xml'
227         adr_idx_file = None
228         tmp_adr_idx_file= None
229         got_ex = 0
230
231         try :
232                 in_file = open(sys.argv[1])
233         except IOError, e:
234                 print 'Could not open input file <',sys.argv[1],'>  ABORTING'
235                 sys.exit(1)
236
237         if len(sys.argv) == 2 :
238                 try :
239                         dlist = os.listdir(path_to_out);
240                         flist=[]
241                         for l in dlist :
242                                 if l.find('addrbook') == 0 and l.find("addrbook--index.xml") < 0 and l.find('bak') < 0 :
243                                         flist.append(l)
244                         flist.sort()
245                         out_file = flist.pop()
246                         os.rename(path_to_out+out_file, path_to_out+out_file+'.tmp')
247                         out_file = open(path_to_out+out_file,'w')
248                         convert(in_file, out_file)
249                 except Exception, e:
250                         got_ex = 1
251                         print 'got exception: ', e
252         else :
253                 try :
254                         os.rename(path_to_out+adr_idx, path_to_out+adr_idx+'.tmp')
255                         tmp_adr_idx_file = open(path_to_out+adr_idx+'.tmp')
256                         adr_idx_file = open(path_to_out+adr_idx,'w')
257                 except Exception, e :
258                         print 'Could not open <', path_to_out+adr_idx,'> file. Make sure you started Claws Mail at least once.'
259                         sys.exit(1)
260                 try :
261                         out_file = open(path_to_out+sys.argv[2],'w')
262                         convert(in_file, out_file, sys.argv[2].split('.xml')[0])
263                         l = tmp_adr_idx_file.readline()
264                         while l :
265                                 if l.strip() == '</book_list>' :
266                                         adr_idx_file.write('\t<book name="'+sys.argv[2].split('.xml')[0] +'" file="'+sys.argv[2]+'" />\n')
267                                         adr_idx_file.write(l)
268                                 else :
269                                         adr_idx_file.write(l)
270                                 l = tmp_adr_idx_file.readline()
271                 except Exception, e:
272                         got_ex = 1
273                         print 'got exception: ', e
274         
275
276         if got_ex :
277                 #clean up the mess
278                 print 'got exception, cleaning up the mess... changed files will be restored...\n'
279                 if adr_idx_file :
280                         adr_idx_file.close()
281                 if out_file :
282                         out_file.close()
283                 if len(sys.argv) == 2 :
284                         os.rename(out_file.name+'.tmp', out_file.name)
285                 else :
286                         os.remove(out_file.name)
287                         os.rename(path_to_out+adr_idx+'.tmp', path_to_out+adr_idx)
288                 if tmp_adr_idx_file :
289                         tmp_adr_idx_file.close()
290                                 
291         else :
292                 #closing all and moving temporary data into place
293                 print 'closing open files...\n'
294                 in_file.close()
295                 out_file.close()        
296                 if len(sys.argv) == 3 :
297                         os.rename(path_to_out+adr_idx+'.tmp',path_to_out+adr_idx+'.bak' )
298                 if len(sys.argv) == 2 :
299                         os.rename(out_file.name+'.tmp', out_file.name+'.bak')
300                 if adr_idx_file :
301                         adr_idx_file.close()
302                 if tmp_adr_idx_file :
303                         tmp_adr_idx_file.close()
304                 print 'done!'
305                 
306
307 if __name__ == '__main__':
308     execute ()
309
310