carddav2asterisk.py

#!/usr/bin/env python3
#-*- coding: utf-8 -*-

import sys
import requests
import vobject
import re
import asyncio
from panoramisk import Manager
from requests.auth import HTTPBasicAuth
from lxml import etree
from urllib.parse import urlparse
import argparse
import configparser


# get list with links to all available vcards
def getAllVcardLinks(url, auth):
  baseurl = urlparse(url).scheme+'://' + urlparse(url).netloc

  r = requests.request('PROPFIND', url, auth = auth)
  if r.status_code != 207:
    raise RuntimeError('error in response from %s: %r' % (url, r))

  root = etree.XML(r.text)
  vcardUrlList = []
  for record in root.xpath(".//d:response", namespaces = {"d" : "DAV:"}):
    type = record.xpath(".//d:getcontenttype", namespaces = {"d" : "DAV:"})
    if (type) and type[0].text.startswith("text/vcard"):
      vcardlinks = record.xpath(".//d:href", namespaces = {"d" : "DAV:"})
      for link in vcardlinks:
        vcardUrlList.append(baseurl + link.text)
  return vcardUrlList


def tidyPhoneNumber(config, num):
  num = re.sub("^\+", "00", num)    # +39 -> 0039
  num = re.sub("\D", "", num)       # remove all non-digits
  if 'phone' in config:
    if 'nationalprefix' in config['phone']:
      num = re.sub("^" + config['phone']['nationalprefix'] + "0*", "0", num)    # strip own national prefix
    if 'domesticprefix' in config['phone']:
      num = re.sub("^[^0]", "0" + config['phone']['domesticprefix'], num)       # add domestic prefix, if missing
  return num


def main():
  parser = argparse.ArgumentParser()
  parser.add_argument('ini_file')
  parser.add_argument("--no-update", help="Don't write to asterisk", action='store_true', default=False)
  args = parser.parse_args()

  config = configparser.RawConfigParser()
  config.read(args.ini_file)

  loop = asyncio.get_event_loop()
  loop.run_until_complete(putCids(loop, args, config))
  loop.close()


def putCids(lp, args, config):
  auth = HTTPBasicAuth(config['carddav']['user'], config['carddav']['pass'])
  url = config['carddav']['url']

  # connect to asterisk
  ami = Manager(host = config['ami']['host'],
                  port = config['ami']['port'],
                  username = config['ami']['user'],
                  secret = config['ami']['pass'])
  yield from ami.connect()

  # get phone numbers from vcard
  for vurl in getAllVcardLinks(url, auth):
    r = requests.request("GET", vurl, auth=auth)
    vcard = vobject.readOne(r.text)
    if "tel" in vcard.contents:
      for telno in vcard.contents['tel']:
        num = tidyPhoneNumber(config, telno.value)
        if "fn" in vcard.contents:
          name = vcard.fn.value
          print("Adding/updating Number: %s Name: %s" % (num, name), end="... ")
          if not args.no_update:
            ami_result = yield from ami.send_action({"Action": "DBPut", "Family": "cidname", "Key": num, "Val": name})
            print(ami_result.Response)
          else:
            print("no-update")
  ami.close()

if __name__ == "__main__":
  main()