def __init__(self, db):
self.my_connector = Connector(db)
self.gd_client = gdata.contacts.client.ContactsClient(source='GoogleInc-ContactsPythonSample-1')
self.people = []
self.db = db
class GoogleContactsConnector:
def __init__(self, db):
self.my_connector = Connector(db)
self.gd_client = gdata.contacts.client.ContactsClient(source='GoogleInc-ContactsPythonSample-1')
self.people = []
self.db = db
def run(self):
print "Starting the Google Contacts Connector"
self.my_connector.request_credentials()
try:
self.import_contacts()
self.print_contacts()
self.add_contacts_to_db()
except gdata.client.BadAuthentication:
print 'Invalid user credentials given.'
return
def import_contacts(self):
self.login()
feed = self.gd_client.GetContacts()
self.process_feed(feed, self.create_person)
def login(self):
username = self.my_connector.username
password = self.my_connector.password
self.gd_client.ClientLogin(username, password, self.gd_client.source)
def process_feed(self, feed, processing_method):
ctr = 0
while feed:
# Process the feed
ctr = processing_method(feed=feed, ctr=ctr)
# Prepare for next feed iteration
next_feed = feed.GetNextLink()
feed = None
if next_feed:
feed = self.gd_client.GetContacts(uri=next_feed.href)
def create_person(self, feed, ctr):
for contact in feed.entry:
print "Processing contact number %s..." % (ctr)
p = Person.Person()
get_name_strategy = GetNameFromGoogleContactsStrategy(contact)
p.add_name_from_service(get_name_strategy)
#no birthdays stored in google contacts
self.add_note(contact, p)
self.add_email_addresses(contact, p)
self.add_phone_numbers(contact, p)
self.people.append(p)
ctr += 1
return ctr
def add_note(self, contact, p):
if contact.content is not None:
content = str(contact.content)
content_text = ET.XML(content).text
if content_text is not None:
p.note = content_text
def add_email_addresses(self, contact, p):
if contact.email is not None:
for email in contact.email:
e = EmailMessage.EmailMessage()
e.address = email.address
p.emails.append(e)
#print "Email: {0}".format(email.address)
def add_phone_numbers(self, contact, p):
if contact.phone_number is not None:
for phone in contact.phone_number:
if phone is not None:
my_phone = Phone.Phone()
if phone.label is not None:
my_phone.type = phone.label
my_phone.number = phone.text
#print "LABEL - {0}: {1}".format(phone.label, phone.text)
elif phone.rel is not None or phone.rel != "None":
formatted_rel = self.format_rel(phone.rel)
my_phone.type = formatted_rel
my_phone.number = phone.text
#print "REL - {0}: {1}".format(formatted_rel, phone.text)
else:
my_phone.number = phone.text
#print "Phone: {0}".format(phone.text)
p.phones.append(my_phone)
def format_rel(self, rel):
index_of_hashtag = rel.find("#") + 1
return rel[index_of_hashtag:]
def print_contacts(self):
for person in self.people:
person.print_out()
print ""
def add_contacts_to_db(self):
entity_to_database = PersonToDatabase(self.people, self.db)
entity_to_database.add_standard_attribute_table_setters()
entity_to_database.add_people_to_database()
