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

[Slava Bacherikov]

commit:     1141ccb27e72421a606f2e7c5324ce32f7cff426
Author:     Slava Bacherikov <slava <AT> bacher09 <DOT> org>
AuthorDate: Tue Jul  3 20:23:07 2012 +0000
Commit:     Slava Bacherikov <slava <AT> bacherikov <DOT> org <DOT> ua>
CommitDate: Tue Jul  3 21:07:08 2012 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/gentoo-packages.git;a=commit;h=1141ccb2

Add layer for work with portage news

---
 .../libs/package_info/generic_metadata/news.py     |  133 ++++++++++++++++++++
 gpackages/libs/package_info/mixins.py              |   14 ++-
 2 files changed, 146 insertions(+), 1 deletions(-)

diff --git a/gpackages/libs/package_info/generic_metadata/news.py b/gpackages/libs/package_info/generic_metadata/news.py
new file mode 100644
index 0000000..d84ac6b
--- /dev/null
+++ b/gpackages/libs/package_info/generic_metadata/news.py
@@ -0,0 +1,133 @@
+import os
+import os.path
+import re
+import hashlib
+from datetime import datetime
+from email import message_from_string
+from email.utils import getaddresses
+from ..generic import ToStrMixin, toint, file_get_content
+
+NEWS_STR_RE = r'^(?P<date>\d{4,4}-\d{2}-\d{2})-(?P<title>.*)$'
+news_re = re.compile(NEWS_STR_RE)
+
+class News(ToStrMixin):
+    
+    def __init__(self, repo_path = '/usr/portage'):
+        self.news_path = os.path.join(repo_path, 'metadata', 'news')
+        if not os.path.isdir(self.news_path):
+            raise ValueError
+        # For repr
+        self.repo_path = repo_path 
+
+    def iter_news(self):
+        for name in os.listdir(self.news_path):
+            try:
+                i = NewsItem(self.news_path, name)
+            except ValueError:
+                pass
+            else:
+                yield i
+
+    def __iter__(self):
+        return self.iter_news()
+
+    def __unicode__(self):
+        return unicode(self.repo_path)
+
+class NewsItem(ToStrMixin):
+
+    N_ITEM_P = r'^%(name)s\.(?P<lang>[a-z]{2})\.txt$'
+
+    def __init__(self, news_path, news_dir):
+        ndir = os.path.join(news_path, news_dir)
+        if not os.path.isdir(ndir):
+            raise ValueError
+        m = news_re.match(news_dir)
+        if m is None:
+            raise ValueError
+        p_dct =  m.groupdict()
+        try:
+            date = datetime.strptime(p_dct['date'], '%Y-%m-%d')
+        except ValueError:
+            raise
+        else:
+            self.date = date.date()
+
+        self.title = p_dct['title']
+        self.name = news_dir
+        self.news_dir = ndir
+        self._news_dict = {}
+        self._fetch_news()
+
+    def _iter_news_items(self):
+        pattern_str = self.N_ITEM_P % {'name': self.name}
+        pattern = re.compile(pattern_str)
+        for item in os.listdir(self.news_dir):
+            m = pattern.match(item)
+            full_path = os.path.join(self.news_dir, item)
+            if m is not None and os.path.isfile(full_path):
+                lang = m.groupdict()['lang']
+                yield (full_path, lang)
+            
+    def _fetch_news(self):
+        for item, lang in self._iter_news_items():
+            self._news_dict[lang] = NewsItemLang(item,
+                                                 self.date,
+                                                 lang,
+                                                 self.title)
+
+    @property
+    def default_news(self):
+        return self._news_dict['en']
+
+    @property
+    def news(self):
+        return self._news_dict
+
+    def __unicode__(self):
+        return unicode(self.name)
+
+class NewsItemLang(object):
+    
+    def __init__(self, item, date, lang = 'en', name = ''):
+        f = file_get_content(item)
+        self.sha1 = hashlib.sha1(f).hexdigest()
+        self._mes_obj = message_from_string(f)
+        self.date = date
+        self.lang = lang
+        self.name = name
+
+    @property
+    def title(self):
+        return self._mes_obj.get('Title')
+
+    @property
+    def revision(self):
+        return toint(self._mes_obj.get('Revision', 1),1)
+
+    @property
+    def format_ver(self):
+        g = self._mes_obj.get('News-Item-Format', '1.0')
+        try:
+            maj, min = g.split('.')
+        except ValueError:
+            maj, min = 1, 0
+
+        return toint(maj,1), toint(min, 0)
+
+    @property
+    def authors(self):
+        "Returns list of tuples"
+        authors = self._mes_obj.get_all('Author')
+        return getaddresses(authors)
+
+    @property
+    def translators(self):
+        "Returns list of tuples"
+        translators = self._mes_obj.get_all('Translator')
+        return getaddresses(translators)
+
+    @property
+    def message(self):
+        return self._mes_obj.get_payload()
+

diff --git a/gpackages/libs/package_info/mixins.py b/gpackages/libs/package_info/mixins.py
index 923fcc8..0ef9b4d 100644
--- a/gpackages/libs/package_info/mixins.py
+++ b/gpackages/libs/package_info/mixins.py
@@ -13,6 +13,8 @@ from .generic_metadata.category_metadata import CategoryMetadata
 from .generic_metadata.package_metadata import PackageMetaData
 #License group metadata
 from .generic_metadata.license_groups import LicenseGroups
+# News
+from .generic_metadata.news import News
 # Validators
 from .validators import validate_url, validate_email, ValidationError
 #Generic objects
@@ -126,6 +128,15 @@ class PortTreeBaseMixin(ToStrMixin):
     def __unicode__(self):
         return self.name
 
+class PortTreeNewsMixin(object):
+    
+    @cached_property
+    def news(self):
+        try:
+            return News(self.porttree_path)
+        except:
+            return None
+
 class PortTreeIteratorMixin(AutoGeneratorMixin):
     main_iterator = 'iter_categories'
     generator_names = ('iter_packages', 'iter_ebuilds')
@@ -335,7 +346,8 @@ class PackageGenericMixin(PackageBaseMixin, PackageFilesMixin):
 class PortageMixin(PortageGenericMixin, PortageIteratorMixin, AbstractPortage):
     pass
 
-class PortTreeMixin(PortTreeBaseMixin, PortTreeIteratorMixin, AbstractPortTree):
+class PortTreeMixin(PortTreeBaseMixin, PortTreeIteratorMixin, \
+                    PortTreeNewsMixin, AbstractPortTree):
     pass
 
 class CategoryMixin(CategoryBaseMixin, CategoryIteratorMixin, AbstractCategory):