#!/usr/bin/python
 
import sys, pinyin
from Foundation import *
from ScriptingBridge import *
 
reset = False
 
def is_cjk_char(x):
    # Punct & Radicals
    if x >= 0x2e80 and x <= 0x33ff:
        return 1
 
    # Fullwidth Latin Characters
    if x >= 0xff00 and x <= 0xffef:
        return 1
 
    # CJK Unified Ideographs &
    # CJK Unified Ideographs Extension A
    if x >= 0x4e00 and x <= 0x9fbb:
        return 1
    # CJK Compatibility Ideographs
    if x >= 0xf900 and x <= 0xfad9:
        return 1
 
    # CJK Unified Ideographs Extension B
    if x >= 0x20000 and x <= 0x2a6d6:
        return 1
 
    # CJK Compatibility Supplement
    if x >= 0x2f8000 and x <= 0x2fa1d:
        return 1
 
    return 0
 
def contain_cjk_char(line):
    # print "@: %s" % (line.encode("utf-8")) 
 
    for ch in line:
       if is_cjk_char(ord(ch)):
           return 1
 
    return 0
 
def assign_pinyin(getName, getPhoneticName, setPhoneticName):
    name = getName()
 
    if name and contain_cjk_char(name):
        pname = getPhoneticName()
        if not reset and pname:
            return None
 
        name_py = pinyin.hanzi2pinyin(name).capitalize()
        setPhoneticName(name_py)
 
        return (name, name_py)
 
    return None
 
ab = SBApplication.applicationWithBundleIdentifier_("com.apple.AddressBook")
 
if len(sys.argv) > 1 and sys.argv[0] == '-r':
    reset = True
 
for person in ab.people():
    fname_pair = assign_pinyin(person.firstName, 
                               person.phoneticFirstName,
                               person.setPhoneticFirstName_)
    lname_pair = assign_pinyin(person.lastName, 
                               person.phoneticLastName,
                               person.setPhoneticLastName_)
 
    if fname_pair or lname_pair:
        print "%s%s (%s%s%s)" % ((fname_pair and fname_pair[0]) or "",
                                 (lname_pair and lname_pair[0]) or "",
                                 (lname_pair and fname_pair and " ") or "",
                                 (fname_pair and fname_pair[1]) or "",
                                 (lname_pair and lname_pair[1]) or "")
 
print "Done."