csv2FritzBox_phonebook/csv2FritzBox.py at master · c7h/csv2FritzBox_phonebook · 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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
'''
Created on 19.04.2014


@author: Christoph Gerneth
'''

from csv import DictReader
import xml.etree.ElementTree as et
import time

from os.path import expanduser
## optionsparser
from optparse import OptionParser
op = OptionParser(usage="convert your Contacts to FritzBox-Phonebook xml")
op.add_option("-i", dest="filename_in", help="input file (Outlook CSV format)")
op.add_option("-o", dest="filename_out", help="output file (xml)")
op.add_option("-d", dest="debug", action='store_true', help="debug mode")
(options, args) = op.parse_args()

if options.filename_in and options.filename_out:
    filename_in = expanduser(options.filename_in)
    filename_out = expanduser(options.filename_out)
else:
    op.error("i need input AND output (-h is for help)")


trans_number = {"Home Phone": "home",
                "Mobile Phone": "mobile",
                "Business Phone": "work",
                "Home Fax": "fax_work",
                }


with open(filename_in, "rb") as f:
    reader = DictReader(f)
    phonebooks = et.Element("phonebooks")
    phonebook = et.SubElement(phonebooks, tag="phonebook")
    contact_count = 1
    for cr in reader:

        contact = et.SubElement(phonebook, "contact")

        category = et.SubElement(contact, "category")
        category.text = "0"

        person = et.SubElement(contact, "person")
        realName = et.SubElement(person, "realName")
        realName.text = "%s %s" % (cr["Last Name"].decode('utf-8'), cr["First Name"].decode('utf-8'))

        telephony = et.SubElement(contact, "telephony")
        number_counter = 0
        for key, value in trans_number.iteritems():
            if key in cr:
                if cr[key] != "":
                    phonenumber = et.SubElement(telephony,
                                  "number",
                                  {"id": str(number_counter),
                                   "prio": "0",
                                   "type": value,
                                   }
                                  )
                    phonenumber.text = cr[key].decode('utf-8')
                    number_counter += 1
        telephony.set("nid", str(number_counter+1))

        service = et.SubElement(contact, "services", {"nid": "1"})
        mail = et.SubElement(service, "email", {"classifier": "private", "id": "0"})
        mail.text = cr["E-mail Address"].decode('utf-8')

        et.SubElement(contact, "setup")

        mod_time = et.SubElement(contact, "mod_time")
        mod_time.text = str(int(time.time()))

        uid = et.SubElement(contact, "uniqueid")
        uid.text = str(contact_count)
        contact_count += 1
    elemTree = et.ElementTree(element=phonebooks)

    if options.debug == True:
        root = elemTree.getroot()
        et.dump(root)

    print "[parsed %i contacts]" % contact_count
    print "[writing to '%s']" % filename_out
    elemTree.write(filename_out)