addon_generator.py

#!/usr/bin/env python
""" addons.xml generator """

import os
import md5
import sys


class Generator:
    """
        Generates a new addons.xml file from each addons addon.xml file
        and a new addons.xml.md5 hash file. Must be run from the root of
        the checked-out repo. Only handles single depth folder structure.
    """
    def __init__(self, outdir='out'):
        # generate files
        self._skip_kodiro = True
        self._outdir = outdir
        self._generate_addons_file()
        self._generate_md5_file()
        # notify user
        print "Finished updating addons xml and md5 files"

    def _generate_addons_file(self):
        # addon list
        addons = os.listdir(".")
        # final addons text
        addons_xml = u"<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n<addons>\n"
        # loop thru and add each addons addon.xml file
        for addon in addons:
            try:
                # skip any file or .svn folder
                if not os.path.isdir(addon) or addon in (".svn", ".git"):
                    continue
                # create path
                _path = os.path.join(addon, "addon.xml")
                # skip non addon directories
                if not os.path.exists(_path):
                    continue
                # special case, skip another repos in filmkodi
                if self._skip_kodiro and self._outdir.endswith('/filmkodi') and 'repository.kodiro' in addon:
                    continue
                # split lines for stripping
                xml_lines = open(_path, "r").read().splitlines()
                # new addon
                addon_xml = ""
                # loop thru cleaning each line
                for line in xml_lines:
                    # skip encoding format line
                    if line.find("<?xml") == -1:
                        # add line
                        addon_xml += unicode(line.rstrip() + "\n", "UTF-8")
                # we succeeded so add to our final addons.xml text
                addons_xml += addon_xml.rstrip() + "\n\n"
            except Exception as e:
                # missing or poorly formatted addon.xml
                print "Excluding %s for %s" % (_path, e,)
        # clean and add closing tag
        addons_xml = addons_xml.strip() + u"\n</addons>\n"
        # save file
        self._save_file(addons_xml.encode("UTF-8"), filename="addons.xml")

    def _generate_md5_file(self):
        try:
            # create a new md5 hash
            with open(os.path.join(self._outdir, "addons.xml")) as f:
                m = md5.new(f.read()).hexdigest()
            # save file
            self._save_file(m, filename="addons.xml.md5")
        except Exception, e:
            # oops
            print "An error occurred creating addons.xml.md5 file!\n%s" % (e,)

    def _save_file(self, data, filename):
        try:
            # write data to the file
            with open(os.path.join(self._outdir, filename), "w") as f:
                f.write(data)
        except Exception, e:
            # oops
            print "An error occurred saving %s file!\n%s" % (file, e,)


if __name__ == "__main__":
    # start
    repo = sys.argv[1] if len(sys.argv) > 1 else "filmkodi"
    Generator(outdir=os.path.join('out', repo))