profile.py

#!/usr/bin/python
#
# Script to pull provisioning profiles from Apple's developer portal.
# By Alexey Baj <alexey.baj@gmail.com> for Rob Terrell / TouchCentric
#
# Copyright 2010 TouchCentric LLC. All rights reserved.

import sys, re, urllib, urllib2, urlparse, cookielib, gzip, StringIO, optparse, os

# Apple cookie issue workaround 
# http://bugs.python.org/issue3924
class MyCookieJar(cookielib.CookieJar):
    def _cookie_from_cookie_tuple(self, tup, request):
        name, value, standard, rest = tup
        version = standard.get('version', None)
        if version is not None:
            version = version.replace('"', '')
            standard["version"] = version
        return cookielib.CookieJar._cookie_from_cookie_tuple(self, tup, request)

def fetch(url, data=None):
    data, headers = fetch_with_headers(url, data)
    return data

def bail_out(msg):
    print >>sys.stderr, msg
    sys.exit(1)

def fetch_with_headers(url, data=None):
    
    if data:
        data = urllib.urlencode(data)
    
    req = urllib2.Request(url, data)
    req.add_header("User-Agent", "Mozilla/5.0 (Windows; U; Windows NT 5.1; en; rv:1.9.2) Gecko/20100115 Firefox/3.6 (.NET CLR 3.5.30729)")
    req.add_header("Accept-encoding", "gzip")

    x = urllib2.urlopen(req)
    data = x.read()
    if x.headers.get('content-encoding', None) == 'gzip':
        data = gzip.GzipFile(fileobj=StringIO.StringIO(data)).read()

    return data, x.headers

def extract_helper(regex, data):
    match = re.search(regex, data)
    if match:
        return match.group(1)
    raise Exception('Unexpected web page structure')

# handle cookies 
opener = urllib2.build_opener(urllib2.HTTPCookieProcessor(MyCookieJar()))
urllib2.install_opener(opener)

# process options
p = optparse.OptionParser(description='A script to fetch provision profiles from Apple site', prog='profile.py', \
    version='1.0', usage='usage: %prog file -a <appleID> -p <password> -t <team> -P <profile_name> -o')

p.add_option('-a', "--account", dest='appleID', help='Account\'s Apple ID')
p.add_option('-p', "--password", dest='password', help='Password')
p.add_option('-t', "--team", dest='team', help='Team')
p.add_option('-P', "--profile", dest='profile', help='Profile Name')
p.add_option('-o', "--open", action="store_true", default=False, dest='openAfterSaving', help='open after saving')
options, arguments = p.parse_args()

if not options.appleID or not options.password:
    bail_out('Both Apple ID and password have to be specified')

initial_url = 'https://developer.apple.com/iphone/index.action'
# determine login url
data = fetch(initial_url)
login_url = extract_helper('href="(.+?login.+)"', data)
# fetch login page
data = fetch(login_url)
form_action = urlparse.urljoin(login_url, extract_helper('action="(.+DSAuthWeb\.woa.+)"' , data))
# perform login
data = fetch(form_action, {'theAccountName':options.appleID, 'theAccountPW':options.password})
if data.find('DSAuthWeb') != -1:
    bail_out("Provided credentials are not authorized.")
    
data = fetch(initial_url)
if data.find('saveTeamSelection.action') != -1:
    # additional team selection step
    match = re.search('value="(.+)">%s</option>' % options.team, data)
    if not match:
        bail_out('Team %s is not found.' % options.team)
    form_action = urlparse.urljoin(initial_url, '/devcenter/saveTeamSelection.action')
    data = fetch(form_action, {'memberDisplayId':match.group(1), 'action:saveTeamSelection!save':'Continue'})
    
if data.find('/iphone/my/') == -1:
    provision_url = 'https://developer.apple.com/ios/manage/provisioningprofiles/index.action'
else:
    provision_url = 'http://developer.apple.com/iphone/my/provision/index.action'

# fetching list of profiles
data = fetch(provision_url)

downloadable_links = re.findall('href="(.+blobId=(.+))">', data)
if not downloadable_links:
	bail_out('There are no downloadable profiles.')
    
if options.profile:
    # if profile was specified, leave only it in a list
    match = re.findall('href=".+view\.action\?provDisplayId=(.+)">(<span>|)(.+?)</', data)    
    match = [x[0] for x in match if x[2] == options.profile]
    if not match:
        bail_out('There is no %s profile.' % options.profile)
    downloadable_links = [x for x in downloadable_links if x[1] == match[0]]
    if not downloadable_links:    
        bail_out('Profile %s is present but not downloadable.' % options.profile)

for link in downloadable_links:
    data, headers = fetch_with_headers(urlparse.urljoin(provision_url, link[0]))
    fname = extract_helper('filename=(.+)', headers['Content-Disposition'])
    with open(fname, 'wb') as f:
        f.write(data)
    if options.openAfterSaving:
	    os.system('open "'+fname+'"')