osx/addressbook/edit_default_address.py

#!/usr/bin/env python
# Copyright 2009 Greg Darke <greg+laptop@tsukasa.net.au>
# Licensed for distribution under the GPL version 2, check COPYING for details
# This program reads in all of your contact details (from AddressBook) and then
# turns them into internation numbers (removes the leading 0, and repaces it
# with +61). This program will also add an area code if it detects one is missing.
import os, sys, time
reload(sys).setdefaultencoding('utf8') # Hacky
import logging
logging.basicConfig(level=logging.DEBUG)
try:
	import AddressBook
except ImportError:
	print >>sys.stderr, 'Seems you do not have pyobjc working for this version of python'
	raise
class dictobject(dict):
	def __getattr__(self, key):
		try:
			return self[key]
		except KeyError:
			raise AttributeError, str(key)
	def __setattr__(self, key, value):
		self[key] = value
def getDisplayName(person):
	firstName = unicode(person.valueForProperty_(AddressBook.kABFirstNameProperty))
	lastName = unicode(person.valueForProperty_(AddressBook.kABLastNameProperty))
	companyName = unicode(person.valueForProperty_(AddressBook.kABOrganizationProperty))
	flags = person.valueForProperty_(AddressBook.kABPersonFlags)
	if flags is None:
		flags = 0
	lastNameFirst = (flags & AddressBook.kABNameOrderingMask) == AddressBook.kABLastNameFirst
	if (flags & AddressBook.kABShowAsMask) == AddressBook.kABShowAsMask:
		if companyName:
			return companyName
	
	if lastName and firstName:
		if lastNameFirst:
			return '%s %s' % (lastName, firstName, )
		else:
			return '%s %s' % (firstName, lastName, )
	
	if lastName:
		return lastName
	return firstName
def getEmailAddresses(person):
	lines = []
	emailAddresses = person.valueForProperty_(AddressBook.kABEmailProperty)
	for i in xrange(emailAddresses.count()):
		email = dictobject(
			address=emailAddresses.valueAtIndex_(i),
			label=emailAddresses.labelAtIndex_(i),
			isDefault=emailAddresses.primaryIdentifier()==emailAddresses.identifierAtIndex_(i),
		)
		line = '%s %2d\t%10s\t%s' % (
			'->' if email.isDefault else '  ',
			i, email.label, email.address,
		)
		lines.append(line)
	
	return '\n'.join(lines)
def changeDefaultEmailAddress(person, index):
	emails = person.valueForProperty_(AddressBook.kABEmailProperty)
	if emails.indexForIdentifier_(emails.primaryIdentifier()) == index:
		return
	emails = emails.mutableCopy()
	emails.setPrimaryIdentifier_(emails.identifierAtIndex_(index))
	person.setValue_forProperty_(emails, AddressBook.kABEmailProperty)
def createSearchQuery(values):
	def _createSingleSearchQuery(prop, value):
		return AddressBook.ABPerson.searchElementForProperty_label_key_value_comparison_(
			prop,  # Property
			None,  # Label
			None,  # Key
			value, # Value
			AddressBook.kABPrefixMatchCaseInsensitive, # Comparison
		)
	PROPS = (
		AddressBook.kABFirstNameProperty,
		AddressBook.kABLastNameProperty,
		AddressBook.kABNicknameProperty,
	)
	innerQuery = []
	for value in values:
		query = AddressBook.ABSearchElement.searchElementForConjunction_children_(
			AddressBook.kABSearchOr, 
			[_createSingleSearchQuery(prop, value) for prop in PROPS],
		)
		innerQuery.append(query)
	return AddressBook.ABSearchElement.searchElementForConjunction_children_(
		AddressBook.kABSearchAnd,
		innerQuery,
	)
def clearScreen():
	sys.stdout.write('\033[H\033[2J')
	sys.stdout.flush()
def main():
	book = AddressBook.ABAddressBook.sharedAddressBook()
	def _searchForUser():
		name = raw_input('Enter name: ')
		if not name:
			sys.exit()
			return None
		nameElements = name.strip().split()
		if not nameElements:
			return None
		query = createSearchQuery(nameElements)
		results = book.recordsMatchingSearchElement_(query)
		if len(results) == 0:
			print 'User not found'
			return None
		if len(results) > 1:
			# Hmm, we need to pick somebody
			return _pickUserFromList(results)
		else:
			return results[0]
	def _pickUserFromList(searchResults):
		personID = -1
		while personID < 0 or personID >= len(searchResults):
			for i, person in enumerate(searchResults):
				print '%2d\t%s' % (i, getDisplayName(person))
			personID = raw_input('Person number (enter to cancel): ')
			if not personID:
				return None
			try:
				personID = int(personID)
			except ValueError:
				pass
		return searchResults[personID]
	def _pickNewDefaultAddress(person):
		emailID = -1
		emailAddresses = person.valueForProperty_(AddressBook.kABEmailProperty).count()
		while emailID < 0 or emailID >= emailAddresses:
			print 'Email addresses for %s:' % getDisplayName(person)
			print getEmailAddresses(person)
			emailID = raw_input('New default address (enter to cancel): ')
			if not emailID:
				return None
			try:
				emailID = int(emailID)
			except ValueError:
				pass
		return emailID
	clearScreen()
	try:
		while True:
			person = _searchForUser()
			clearScreen()
			if person is None:
				continue
			emailIndex = _pickNewDefaultAddress(person)
			clearScreen()
			if emailIndex is None:
				print 'Cancelled change'
				continue
			changeDefaultEmailAddress(person, emailIndex)
	except EOFError:
		pass
	finally:
		book.save()
if __name__ == "__main__":
	main()