[gentoo-commits] proj/gentoo-packages:master commit in: gpackages/libs/

[Slava Bacherikov]

commit:     50d5fdbe69872b2bf509cf9c8a9efb4845bd924c
Author:     Slava Bacherikov <slava <AT> bacher09 <DOT> org>
AuthorDate: Sat Jun  2 19:24:17 2012 +0000
Commit:     Slava Bacherikov <slava <AT> bacherikov <DOT> org <DOT> ua>
CommitDate: Sat Jun  2 19:24:17 2012 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/gentoo-packages.git;a=commit;h=50d5fdbe

Changes in libs for working with portage tree.

---
 gpackages/libs/generic.py  |    8 ++++++++
 gpackages/libs/herds.py    |   42 ++++++++++++++++++++++++++++++++++++++----
 gpackages/libs/porttree.py |    8 +-------
 3 files changed, 47 insertions(+), 11 deletions(-)

diff --git a/gpackages/libs/generic.py b/gpackages/libs/generic.py
new file mode 100644
index 0000000..7fe9cc8
--- /dev/null
+++ b/gpackages/libs/generic.py
@@ -0,0 +1,8 @@
+
+class ToStrMixin(object):
+    def __str__(self):
+        return unicode(self).encode('utf-8')
+
+    def __repr__(self):
+        return '<%s %s>' % (type(self).__name__, self.__str__())
+

diff --git a/gpackages/libs/herds.py b/gpackages/libs/herds.py
index c123814..b1e596b 100644
--- a/gpackages/libs/herds.py
+++ b/gpackages/libs/herds.py
@@ -3,6 +3,8 @@ try:
 except (ImportError, SystemError):
     import xml.etree.ElementTree as etree
 
+from collections import defaultdict
+from generic import ToStrMixin
 
 def _gen_func(name):
     return lambda self: getattr(self, name)
@@ -24,10 +26,22 @@ class AbstarctXmlObject(object):
             if obj_xml is not None:
                 setattr(self, '_' + val, obj_xml.text)
 
-class Maintainer(AbstarctXmlObject):
+class Maintainer(AbstarctXmlObject, ToStrMixin):
     attrs = ('name', 'email', 'role')
 
-class Herd(AbstarctXmlObject):
+    def __eq__(self, other):
+        return self.email == other.email
+
+    def __ne__(self, other):
+        return self.email != other.email
+
+    def __hash__(self):
+        return hash(self.email)
+
+    def __unicode__(self):
+        return self.email
+
+class Herd(AbstarctXmlObject, ToStrMixin):
     # create name, email and description property
     attrs = ('name', 'email', 'description')
 
@@ -39,13 +53,33 @@ class Herd(AbstarctXmlObject):
         for maintainer_tree in self._iter_maintainer:
             yield Maintainer(maintainer_tree)
 
+    def __unicode__(self):
+        return self.name
+
 
 class Herds(object):
     def __init__(self, herd_path='/usr/portage/metadata/herds.xml'):
         self._herd_path = herd_path
         self._herd_parse = etree.parse(herd_path)
-        self._iter_herd = self._herd_parse.iterfind('herd')
+        self._herds_dict = None
 
     def iter_herd(self):
-        for herd_tree in self._iter_herd:
+        for herd_tree in self._herd_parse.iterfind('herd'):
             yield Herd(herd_tree)
+
+    def get_herds_indict(self):
+        if self._herds_dict is not None:
+            return self._herds_dict
+        res = {}
+        for herd in self.iter_herd():
+            res[herd.name] = herd
+        self._herds_dict = res
+        return res
+
+    def get_maintainers_with_hers(self):
+        herd_dict = self.get_herds_indict()
+        maintainer_dict = defaultdict(list)
+        for herd in herd_dict.itervalues():
+            for maintainer in herd.iter_mainteiner():
+                maintainer_dict[maintainer].append(herd.name)
+        return maintainer_dict

diff --git a/gpackages/libs/porttree.py b/gpackages/libs/porttree.py
index 0a5b0fc..3f1f1f6 100644
--- a/gpackages/libs/porttree.py
+++ b/gpackages/libs/porttree.py
@@ -6,6 +6,7 @@ from portage.exception import PortageException, FileNotFound, InvalidAtom, \
 
 from gentoolkit.package import Package as PackageInfo
 from gentoolkit.metadata import MetaData
+from generic import ToStrMixin
 from datetime import datetime
 import hashlib
 import os
@@ -47,13 +48,6 @@ def file_mtime(file_path):
     else:
         return None
 
-class ToStrMixin(object):
-    def __str__(self):
-        return unicode(self).encode('utf-8')
-
-    def __repr__(self):
-        return '<%s %s>' % (type(self).__name__, self.__str__())
-
 class Use(ToStrMixin):
     __slots__ = ('name',)