- initial import

5 files changed, 341 insertions, 0 deletions

diff --git a/chitin/__init__.py b/chitin/__init__.py
new file mode 100644
index 0000000..792d600
--- /dev/null
+++ b/chitin/__init__.py
@@ -0,0 +1 @@
+#
diff --git a/chitin/__init__.pyo b/chitin/__init__.pyo
new file mode 100644
index 0000000..2f4bca1
--- /dev/null
+++ b/chitin/__init__.pyo
diff --git a/chitin/parser.py b/chitin/parser.py
new file mode 100644
index 0000000..27e79cd
--- /dev/null
+++ b/chitin/parser.py
@@ -0,0 +1,311 @@
+#!/usr/bin/python
+
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU Library General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+#
+# Copyright Red Hat Inc. 2007, 2008
+#
+# Author: Izhar Firdaus <izhar@fedoraproject.com>
+# Examples:
+#
+# yum oci-install oneclickinstall-XML-metadata-file
+# yum oci-query details oneclickinstall-XML-metadata-file
+# yum oci-query repositories oneclickinstall-XML-metadata-file
+# yum oci-query packages oneclickinstall-XML-metadata-file
+#
+
+
+from xml.etree.ElementTree import ElementTree,Element
+import re
+
+from yum.plugins import PluginYumExit,TYPE_CORE,TYPE_INTERACTIVE
+from yum import logginglevels
+import yum.i18n, tempfile
+_ = yum.i18n._
+
+ocins='http://opensuse.org/Standards/One_Click_Install'
+#distro='fedora'
+#release='9'
+
+## Use this for now
+distro='openSUSE'
+release='Factory'
+
+
+requires_api_version = '2.5'
+plugin_type = (TYPE_INTERACTIVE,)
+
+
+def attrsfilter(elements,keywords,operation='or'):
+   """
+   Function to filter elements by its attributes
+
+   @param   elements      A list of elements to be filtered	
+   @param   keywords      A dictionary containing required tags and its values
+   @param   operation     A string, either 'or' or 'and', for operation selection
+
+   @return              A list of filtered elements by attributes
+   """
+   retval = []
+   if not keywords: return elements
+   if operation == 'or':
+       for element in elements:
+           for key in keywords:
+               if element.attrib.has_key(key):
+                  if element.attrib[key] == keywords[key]:
+                     retval.append(element)
+                     break
+   elif operation == 'and':
+       for element in elements:
+           for key in keywords:
+               if not element.attrib.has_key(key): break
+               if element.attrib[key] != keywords[key]: break
+           retval.append(element)
+   return retval
+
+def get_element_tag(element):
+    """
+    Function to get the element tag without namespace
+
+    @param   element    An element which will be extracted the tag name
+
+    @return             A string, containing the tag name without namespace
+    """
+    tag = re.match('{.*?}(.*)',element.tag)
+    if tag:
+       return tag.group(1)
+    else:
+       return tag
+
+def get_child_value(element,childname):
+    """
+    Function to get the value of an element's child element
+
+    @param  element    An element to be queried
+    @param  childname  A string, containing the tagname of the child
+
+    @return            A string containing child's text value if child is a leaf element, 
+                       or a ElementTree element if child is a parent to 1 or more elements
+    """
+    for c in element.getchildren():
+        if get_element_tag(c) == childname:
+           if c.getchildren():
+             return c
+           else:
+             return c.text
+
+
+def add_tmp_repo(base,name,repourls):
+    """
+    Function to add temporary repository
+
+    @param  base      The YumBase object for the operation
+    @param  name      The repository name
+    @param  repourls  A list of urls for the repository
+
+    @return           None
+    """
+    tfo = tempfile.NamedTemporaryFile()
+    repoid = name
+    repoconfig = """
+[%s]
+name=%s
+failovermethod=priority
+baseurl=%s
+enabled=1
+""" % (name,name,'\n'.join(repourls))
+    tfo.file.write(repoconfig)
+    tfo.file.close()
+    base.getReposFromConfigFile(tfo.name)
+
+class OCIMetapackage:
+   """
+   Parser class for OCI Metapackage XML.
+   """
+   def __init__(self,element,distversion):
+       """
+       @param   element       ElementTree object which stores the XML file
+       @param   distversion   A string containing '$distro $release'
+
+       @return                None
+       """
+       self.element = element
+       self.distversion = distversion
+
+   def getGroup(self):
+       """
+       Function to get Group element from OCI Metadata
+
+       @return    'group' Element of the current distversion
+       """
+       for group in self.element.getiterator('{%s}%s' % (ocins,'group')):
+           if group.attrib['distversion'] == self.distversion:
+              return group
+
+   def getRepositories(self,*args,**kwargs):
+       """
+       Function to get 'repository' elements from the current distversion group
+
+       @kwargs     Keyword arguments for filtering results
+
+       @return     A list containing a filtered list of 'repository' Elements
+       """
+       repos = self.getGroup().getiterator('{%s}%s' % (ocins,'repository'))
+       return attrsfilter(repos,kwargs)
+
+   def getProducts(self,*args,**kwargs):
+       """
+       Function to get 'product' elements from the current distversion group
+
+       @kwargs     Keyword arguments for filtering results
+
+       @return     A list containing a filtered list of 'product' Elements
+       """
+       products = self.getGroup().getiterator('{%s}%s' % (ocins,'product'))
+       return attrsfilter(products,kwargs)
+
+   def getRepoURL(self,reponame):
+       """
+       Function to get 'url' values from current distversion group, and requested reponame
+
+       @param   reponame   The repository name
+
+       @return             A list of URLs from the requested repository, sorted higher score first
+       """
+       repos = self.getRepositories()
+       urls = []
+       for r in repos:
+           if get_child_value(r,'name') == reponame:
+              for url in r.getiterator('{%s}%s' % (ocins,'url')):
+                  priority = 0
+                  if url.attrib.has_key('score'): priority = int(url.attrib['score'])
+                  urls.append((url.text.strip(),priority))
+       def sorturl(x,y):
+           if x[1] <= y[1]:
+              return +1
+           else:
+              return -1
+       urls.sort(sorturl)
+       return [ url[0] for url in urls ]
+
+class OCIQuery:
+    def __init__(self,distro,release):
+        self.distro = distro
+        self.release = release
+
+    def getNames(self):
+        return ['oci-query']
+
+    def getUsage(self):
+        return "[details|repositories|packages] [OCI file]"
+
+    def getSummary(self):
+        return "Query data from OneClickInstall metadata"
+
+    def doCheck(self, base, basecmd, extcmds):
+        pass
+
+    def getRepos(self): # so we can act as a "conduit"
+        return self.repos
+
+    def show_pkg(self, msg, pkg, md, disp=None):
+        print msg, pkg, md
+
+    def show_pkg_exit(self):
+        pass
+
+    def doCommand(self, base, basecmd, extcmds):
+        action = extcmds[0]
+        param = extcmds[1:]
+        metapackages = []
+        for mp in param:
+            et = ElementTree(Element('metapackage'),open(mp))
+            metapackages.append(OCIMetapackage(et,'%s %s' % (self.distro,self.release)))
+
+        if action == 'details':
+           for oci in metapackages:
+               print '======================='
+               print 'Metadata'
+               print '======================='
+               for c in oci.getGroup().getchildren():
+                   tagname = get_element_tag(c)
+                   if tagname in ['name','summary','description']:
+                      print tagname,':',c.text
+        if action == 'repositories':
+           for oci in metapackages:
+               print '========================'
+               print 'Repositories'
+               print '========================'
+               for r in oci.getRepositories():
+                   print get_child_value(r,'name')
+                   print oci.getRepoURL(get_child_value(r,'name'))
+        if action == 'packages':
+           for oci in metapackages:
+               print '========================'
+               print 'Packages'
+               print '========================'
+               for p in oci.getProducts():
+                   print p.getchildren()[0].text
+        print ''
+        return 0, ['']
+
+class OCIInstall:
+    def __init__(self,distro,release):
+        self.distro = distro
+        self.release = release
+
+    def getNames(self):
+        return ['oci-install']
+
+    def getUsage(self):
+        return "[OCI file]"
+
+    def getSummary(self):
+        return "Install packages from OneClickInstall metadata"
+
+    def doCheck(self, base, basecmd, extcmds):
+        pass
+
+    def getRepos(self): # so we can act as a "conduit"
+        return self.repos
+
+    def show_pkg(self, msg, pkg, md, disp=None):
+        print msg, pkg, md
+
+    def show_pkg_exit(self):
+        pass
+
+    def doCommand(self, base, basecmd, extcmds):
+        packages = []
+        repositories = []
+        for mp in extcmds:
+            et = ElementTree(Element('metapackage'),open(mp))
+            oci = OCIMetapackage(et,'%s %s' % (self.distro,self.release))
+            for p in oci.getProducts():
+                packages.append(get_child_value(p,'name'))
+            for r in oci.getRepositories():
+                repositories.append((get_child_value(r,'name'),oci.getRepoURL(get_child_value(r,'name'))))
+        print packages
+        print repositories
+
+        base.verbose_logger.log(logginglevels.INFO_2,
+                _("Setting up Install Process"))
+        try:
+            return base.installPkgs(packages)
+        except yum.Errors.YumBaseError, e:
+            return 1, [str(e)]
+
+def config_hook(conduit):
+    conduit.registerCommand(OCIQuery(distro,release))
+    conduit.registerCommand(OCIInstall(distro,release))
diff --git a/setup.cfg b/setup.cfg
new file mode 100644
index 0000000..01bb954
--- /dev/null
+++ b/setup.cfg
@@ -0,0 +1,3 @@
+[egg_info]
+tag_build = dev
+tag_svn_revision = true
diff --git a/setup.py b/setup.py
new file mode 100644
index 0000000..1e98c7c
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,26 @@
+from setuptools import setup, find_packages
+import sys, os
+
+version = '0.0.1'
+
+setup(name='chitin',
+      version=version,
+      description="OpenSuSE One Click Install metadata parsing library for python",
+      long_description="""\
+""",
+      classifiers=[], # Get strings from http://pypi.python.org/pypi?%3Aaction=list_classifiers
+      keywords='',
+      author='Mohd Izhar Firdaus Ismail',
+      author_email='izhar@fedoraproject.org',
+      url='http://izhar.fedorapeople.org/projects/chitin',
+      license='GPL',
+      packages=find_packages(exclude=['ez_setup', 'examples', 'tests']),
+      include_package_data=True,
+      zip_safe=False,
+      install_requires=[
+          # -*- Extra requirements: -*-
+      ],
+      entry_points="""
+      # -*- Entry points: -*-
+      """,
+      )