311 lines
8.6 KiB
Python
Executable File
311 lines
8.6 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# -*- coding: latin-1 -*-
|
|
"""
|
|
|
|
Copyright © 2003 Bogdan Sumanariu <zarrok@yahoo.com>
|
|
|
|
This file is free software; you can redistribute it and/or modify it
|
|
under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful, but
|
|
WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
|
|
script name : evolutionvcard2claws.py
|
|
|
|
script purpose : convert an evolution addressbook VCARD file
|
|
into a Claws Mail addressbook
|
|
|
|
tested with evolution 1.2.x, and 1.4.x
|
|
|
|
"""
|
|
|
|
import string
|
|
import sys
|
|
import time
|
|
import os
|
|
from io import StringIO
|
|
|
|
keywds = ('x-evolution-file-as','fn', 'n','email;internet','nickname', 'url', 'org')
|
|
|
|
def normalizeLongLines(file):
|
|
"""
|
|
Skip line breaks after 72 chars
|
|
"""
|
|
buf = ''
|
|
|
|
line = file.readline()
|
|
while line:
|
|
if line[0] == ' ':
|
|
buf = buf.rstrip('\n')
|
|
line = line.lstrip();
|
|
buf += line
|
|
else:
|
|
buf += line
|
|
line = file.readline()
|
|
|
|
return buf
|
|
|
|
def getEmailAddress(vcard):
|
|
"""
|
|
Get email address.
|
|
Supported formats:
|
|
- email;something
|
|
- email;type=something
|
|
something := (internet,work,home, other)
|
|
"""
|
|
|
|
for key in vcard:
|
|
items = key.split(';')
|
|
if len(items) == 2:
|
|
if items[0].lower() == 'email':
|
|
list = vcard[key]
|
|
return list[0]
|
|
else:
|
|
if key.lower() == 'email':
|
|
list = vcard[key]
|
|
return list[0]
|
|
|
|
return ""
|
|
|
|
def findName(vcard):
|
|
"""
|
|
Find a version 3.0 name
|
|
"""
|
|
for key in vcard:
|
|
items = key.split(';')
|
|
if len(items) == 2:
|
|
if items[0].lower() == 'n':
|
|
return vcard[key]
|
|
else:
|
|
if key.lower() == 'n':
|
|
return vcard[key]
|
|
|
|
return None
|
|
|
|
################################################################################
|
|
## reads a vcard and stores as hash pairs key/value where value is a list ##
|
|
################################################################################
|
|
|
|
def readVCARD (buffer) :
|
|
|
|
"""
|
|
|
|
skips fom <file> until a 'begin' tag from VCARD is encountered.
|
|
from this point starts constructing a map (key, [values] )
|
|
VCARD entry format -> tag:value
|
|
|
|
key <- tag
|
|
[values] <- list with the values of <tag> if there are more tags with the same name
|
|
|
|
"""
|
|
r=' '
|
|
bgn,end = -1, -1;
|
|
d = dict()
|
|
while r and bgn < 0 :
|
|
r = buffer.readline()
|
|
if len (r) == 0 : return dict()
|
|
if r.strip().lower().find('begin') :
|
|
bgn = 1
|
|
while r and end < 0 :
|
|
r = buffer.readline()
|
|
s = r.strip().lower().split(':')
|
|
if s[0] != '' :
|
|
if s[0] in d :
|
|
d[s[0]].append(s[1])
|
|
elif len(s) > 1:
|
|
d[s[0]] = [s[1]]
|
|
else :
|
|
d[s[0]] = ['']
|
|
if s[0] == 'end' : end = 1
|
|
return d
|
|
|
|
##################################################################################
|
|
|
|
###############################################################################################
|
|
## writes on a given file an xml representation for claws-mail addressbook received as a hash ##
|
|
###############################################################################################
|
|
|
|
def writeXMLREPR (vcard,file,uid) :
|
|
|
|
"""
|
|
based on <vcard> and <uid> writes only recognized tags (the ones defined in <keywds> list)
|
|
NOTE: <url> and <org> tag will be written as attributes (there are such tags in claws-mail's
|
|
XML schema)
|
|
"""
|
|
if len (vcard.keys()) == 0 : return
|
|
item = vcard.get(keywds[2]);
|
|
if item:
|
|
name = item[0].split(';')
|
|
else:
|
|
""" version 3.0 n ?"""
|
|
name = findName(vcard)
|
|
if not name:
|
|
return
|
|
|
|
fn, ln, nick, cn, a = '', '', '', '', ''
|
|
|
|
if len(name) >= 2 :
|
|
fn = name[0]
|
|
ln = name[1]
|
|
elif len(name) ==1 :
|
|
fn = name[0]
|
|
|
|
if keywds[4] in vcard :
|
|
nick = vcard.get(keywds[4])[0]
|
|
if len(vcard.get(keywds[1])[0]) :
|
|
cn = vcard.get(keywds[1])[0]
|
|
else :
|
|
cn = vcard.get(keywds[0])[0];
|
|
|
|
a += str('\n<person uid=\"' + str(uid[0]) + '\" first-name=\"' + fn + '\" last-name=\"' + ln
|
|
+ '\" nick-name=\"' + nick + '\" cn=\"' + cn + '\" >\n')
|
|
a += '\t<address-list>\n'
|
|
if vcard.get(keywds[3]) :
|
|
for c in vcard.get(keywds[3]) :
|
|
uid[0] = uid[0] + 1
|
|
a += '\t\t<address uid=\"' + str(uid[0]) + '\" alias=\"' + nick + '\" email=\"' + c + '\" remarks=\"\" />\n'
|
|
else :
|
|
email = getEmailAddress(vcard)
|
|
uid[0] = uid[0]+1
|
|
a += '\t\t<address uid=\"' + str(uid[0]) + '\" alias=\"' + nick + '\" email=\"' + email + '\" remarks=\"\" />\n'
|
|
a += '\t</address-list>\n'
|
|
a += '\t<attribute-list>\n'
|
|
for key in keywds[5:] :
|
|
if vcard.get(key) :
|
|
for c in vcard.get(key) :
|
|
uid[0] = uid[0] + 1
|
|
a += '\t\t<attribute uid=\"' + str(uid[0]) + '\" name=\"' + key +'\">'+c+'</attribute>\n'
|
|
a += '\t</attribute-list>\n'
|
|
a += '</person>\n'
|
|
file.write(a)
|
|
file.flush()
|
|
|
|
###################################################################################################
|
|
|
|
def convert (in_f, o_f, name='INBOX') :
|
|
d = {'d':1}
|
|
uid = [int(time.time())]
|
|
|
|
try :
|
|
print('proccessing...\n')
|
|
o_f.write('<?xml version="1.0" encoding="ISO-8859-1" ?>\n<address-book name="'+name+'" >\n');
|
|
|
|
buf = normalizeLongLines(in_f)
|
|
buffer = StringIO(buf)
|
|
while len(d.keys()) > 0 :
|
|
d = readVCARD(buffer)
|
|
writeXMLREPR (d, o_f, uid)
|
|
uid[0] = uid [0]+1
|
|
|
|
o_f.write('\n</address-book>')
|
|
print('finished processing...\n')
|
|
except IOError as err :
|
|
print('Caught an IOError : ',err,'\t ABORTING!!!')
|
|
raise err
|
|
|
|
#################################################################################################
|
|
|
|
def execute () :
|
|
if len(sys.argv) != 3 and len(sys.argv) != 2 :
|
|
print(str("\nUsage: vcard2xml.py source_file [destination_file]\n\n" +
|
|
'\tWhen only <source_file> is specified will overwrite the existing addressbook.\n'+
|
|
'\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'))
|
|
sys.exit(1)
|
|
|
|
in_file = None
|
|
out_file = None
|
|
path_to_out = os.environ['HOME']+'/.claws-mail/addrbook/'
|
|
adr_idx = 'addrbook--index.xml'
|
|
adr_idx_file = None
|
|
tmp_adr_idx_file= None
|
|
got_ex = 0
|
|
|
|
try :
|
|
in_file = open(sys.argv[1])
|
|
except IOError as e:
|
|
print('Could not open input file <',sys.argv[1],'> ABORTING')
|
|
sys.exit(1)
|
|
|
|
if len(sys.argv) == 2 :
|
|
try :
|
|
dlist = os.listdir(path_to_out);
|
|
flist=[]
|
|
for l in dlist :
|
|
if l.find('addrbook') == 0 and l.find("addrbook--index.xml") < 0 and l.find('bak') < 0 :
|
|
flist.append(l)
|
|
flist.sort()
|
|
out_file = flist.pop()
|
|
os.rename(path_to_out+out_file, path_to_out+out_file+'.tmp')
|
|
out_file = open(path_to_out+out_file,'w')
|
|
convert(in_file, out_file)
|
|
except Exception as e:
|
|
got_ex = 1
|
|
print('got exception: ', e)
|
|
else :
|
|
try :
|
|
os.rename(path_to_out+adr_idx, path_to_out+adr_idx+'.tmp')
|
|
tmp_adr_idx_file = open(path_to_out+adr_idx+'.tmp')
|
|
adr_idx_file = open(path_to_out+adr_idx,'w')
|
|
except Exception as e :
|
|
print('Could not open <', path_to_out+adr_idx,'> file. Make sure you started Claws Mail at least once.')
|
|
sys.exit(1)
|
|
try :
|
|
out_file = open(path_to_out+sys.argv[2],'w')
|
|
convert(in_file, out_file, sys.argv[2].split('.xml')[0])
|
|
l = tmp_adr_idx_file.readline()
|
|
while l :
|
|
if l.strip() == '</book_list>' :
|
|
adr_idx_file.write('\t<book name="'+sys.argv[2].split('.xml')[0] +'" file="'+sys.argv[2]+'" />\n')
|
|
adr_idx_file.write(l)
|
|
else :
|
|
adr_idx_file.write(l)
|
|
l = tmp_adr_idx_file.readline()
|
|
except Exception as e:
|
|
got_ex = 1
|
|
print('got exception: ', e)
|
|
|
|
|
|
if got_ex :
|
|
#clean up the mess
|
|
print('got exception, cleaning up the mess... changed files will be restored...\n')
|
|
if adr_idx_file :
|
|
adr_idx_file.close()
|
|
if out_file :
|
|
out_file.close()
|
|
if len(sys.argv) == 2 :
|
|
os.rename(out_file.name+'.tmp', out_file.name)
|
|
else :
|
|
os.remove(out_file.name)
|
|
os.rename(path_to_out+adr_idx+'.tmp', path_to_out+adr_idx)
|
|
if tmp_adr_idx_file :
|
|
tmp_adr_idx_file.close()
|
|
|
|
else :
|
|
#closing all and moving temporary data into place
|
|
print('closing open files...\n')
|
|
in_file.close()
|
|
out_file.close()
|
|
if len(sys.argv) == 3 :
|
|
os.rename(path_to_out+adr_idx+'.tmp',path_to_out+adr_idx+'.bak' )
|
|
if len(sys.argv) == 2 :
|
|
os.rename(out_file.name+'.tmp', out_file.name+'.bak')
|
|
if adr_idx_file :
|
|
adr_idx_file.close()
|
|
if tmp_adr_idx_file :
|
|
tmp_adr_idx_file.close()
|
|
print('done!')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
execute ()
|
|
|
|
|