My current progress of an attempt of updating autodmg-rss.py for https://github.com/MagerValp/AutoDMGUpdateProfiles

autodmg_rss.py

#!/usr/bin/env python
# _*_ coding:utf-8 _*_
"""Helper tool to generate AutoDMG profiles."""

import xml.etree.ElementTree as ET
import sys
import re
import argparse
import urllib2
import plistlib

FEED_URL = u'http://rss.support.apple.com/?channel=DOWNLOADS'

INCLUDED_UPDATES = ['Update',
                    'Safari',
                    'iTunes',]

EXCLUDED_UPDATES = ['Windows',
                    'Firmware',
                    'Combo',]

def get_download_url(link):
    """Follows HTTP 302 redirects to fetch the final url of a download."""
    base_url = "https://support.apple.com"
    article_number = link.split('/')[-1]
    download_url = ("{base_url}/downloads/{article_number}/en_US/".format(
        base_url=base_url,
        article_number=article_number))
    try:
        request = urllib2.Request(download_url)
        response = urllib2.urlopen(request)
    except BaseException as e:
        sys.exit("Can't download %s: %s" % (download_url, e))

    filename = response.geturl().split('/')[-1]
    return "{}{}".format(download_url, filename)

def get_itunes_download_url(title):
    """Takes the title of an iTunes update and attempts to find a suitble
    download link from the iTunes download page.
    """
    re_version = re.compile(ur"(?P<version>(?:(\d+)\.)?(?:(\d+)\.)?(\d+)(.*))")
    itunes_base_url = "https://swdlp.apple.com/iframes/82/en_us/82_en_us.html"

    # Figure out which version we need a url for
    version_match = re_version.search(title)
    if version_match:
        version = version_match.group('version')

    # Build a regex to include the version number
    re_url = re.compile(ur"value='(https://.*?iTunes{}.dmg)'".format(version))

    try:
        html = urllib2.urlopen(itunes_base_url).read()
    except urllib2.URLError as e:
        sys.exit(u"Feed fetching failed: %s" % unicode(e))

    url_match = re_url.search(html)
    if url_match:
        return url_match.group(1)

def main():
    """Where all the magic happens"""
    try:
        rss_feed = urllib2.urlopen(FEED_URL)
        rss_feed = ET.parse(rss_feed)
    except urllib2.URLError as e:
        sys.exit(u"Feed fetching failed: %s" % unicode(e))
    except ET.ParseError as e:
        sys.exit(u"Feed parsing failed: %s" % unicode(e))

    updates_found = {"Updates": {}}

    for item in rss_feed.findall(".//item"):
        title = item.find('title').text
        # Only includes updates with 'INCLUDED_UPDATES' in the title.
        # Filters out 'EXCLUDED_UPDATES' for things we don't care about.
        if (not any(update in title for update in INCLUDED_UPDATES) or
                any(update in title for update in EXCLUDED_UPDATES)):
            continue

        link = item.find(u'link').text
        url = get_download_url(link)

        if (".dmg" or ".pkg") not in url:
            if "iTunes" in title:
                url = get_itunes_download_url(title)

        print "{}\n{}".format(title, url)
        update = {"name": title,
                  "url": url,
                 }

        updates_found["Updates"][link.split('/')[-1]] = update

    plistlib.writePlist(updates_found, "updates.plist")

if __name__ == '__main__':
    main()