kslob.py

#!/usr/bin/env python
import csv
import json
import lob
import sys

class ParseKickstarterAddresses:
    def __init__(self, filename):
        self.items = []
        with open(filename, 'r') as csvfile:
            reader = csv.DictReader(csvfile)
            for row in reader:
                self.items.append(row)

def addr_identifier(addr):
    # Use the line below if you sent postcards before Kickstarter changed their CSV format and need to send some more.
    # return u"{name}|{address_line1}|{address_line2}|{address_city}|{address_zip}".format(**addr).upper() 
    return u"{name}|{address_line1}|{address_line2}|{address_city}|{address_state}|{address_zip}|{address_country}".format(**addr).upper() 

def kickstarter_dict_to_lob_dict(dictionary):
    ks_to_lob = {'Shipping Name': 'name',
                 'Shipping Address 1': 'address_line1',
                 'Shipping Address 2': 'address_line2',
                 'Shipping City': 'address_city',
                 'Shipping State': 'address_state',
                 'Shipping Postal Code': 'address_zip',
                 'Shipping Country Code': 'address_country'}
    address_dict = {}
    for key in ks_to_lob.keys():
        address_dict[ks_to_lob[key]] = dictionary[key]
    return address_dict

def main():
    filename = sys.argv[1]
    config = json.load(open("config.json"))
    lob.api_key = config['api_key']

    postcard_from_address = config['postcard_from_address']
    postcard_message = config['postcard_message']
    postcard_front = config['postcard_front']
    postcard_name = config['postcard_name']

    print("Fetching list of any postcards already sent...")
    processed_addrs = set()
    postcards = []
    postcards_result = lob.Postcard.list(count=100)
    while len(postcards_result):
        postcards.extend(postcards_result)
        postcards_result = lob.Postcard.list(count=100, offset=len(postcards))

    postcards_sent = []
    for postcard in postcards:
        if postcard.name == postcard_name:
            postcards_sent.append(postcard)

    print("...found {} previously sent postcards.".format(len(postcards_sent)))
    for processed in postcards_sent:
        identifier = addr_identifier(processed.to.to_dict())
        processed_addrs.add(identifier)

    addresses = ParseKickstarterAddresses(filename)

    to_send = []
    already_sent = []

    print("Verifying addresses of backers...")
    for line in addresses.items:
        to_person = line['Shipping Name']
        to_address = kickstarter_dict_to_lob_dict(line)
        if not to_person:
            msg = ("Warning: No address found for '{}'. "
                   "Skipping this backer.")
            print(msg.format(line['Email']))
            continue
        try:
            to_name = to_address['name']
            to_address = lob.AddressVerify.verify(**to_address).to_dict()['address']
            to_address['name'] = to_name
        except lob.exceptions.LobError, e:
            msg = "Warning: Address verification failed for '{}'. "
            msg = msg.format(line['Email'])
            error = e[0]
            if error:
                error_msg = error[0]['message']
                msg += "Error: '{}'. Skipping this backer.".format(error_msg)
                print(msg)
                continue
            else:
                msg += "Mailing postcard anyway."
                print(msg)
                print("Address: {}".format(to_address))

        if addr_identifier(to_address) in processed_addrs:
            already_sent.append(to_address)
        else:
            to_send.append(to_address)

    nbackers = len(addresses.items)
    print("Already sent postcards to {} of {} backers".format(len(already_sent), nbackers))
    if not to_send:
        print("SUCCESS: All backers with verified addresses have been processed, you're done!")
        return

    query = "Send to {} unsent backers now? [y/N]: ".format(len(to_send), nbackers)
    if raw_input(query).lower() == "y":
        successes = failures = 0
        for to_address in to_send:
            try:
                rv = lob.Postcard.create(to=to_address,
                                         name=postcard_name,
                                         from_address=postcard_from_address,
                                         front=postcard_front,
                                         full_bleed=1,
                                         message=postcard_message)
                print("Postcard sent to {}! ({})".format(to_address['name'], rv.id))
                successes += 1
            except lob.exceptions.LobError:
                msg = 'Error: Failed to send postcard to Lob.com'
                print("{} for {}".format(msg, to_address['name']))
                failures += 1
        print("Successfully sent to {} backers with {} failures".format(successes, failures))
    else:
        print("Okay, not sending to unsent backers.")


if __name__ == "__main__":
    main()