From 0e6702e069aeabcd995b8968d0ebbc1593d59bf6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ren=C3=A9=20=27Necoro=27=20Neumann?= <necoro@necoro.net>
Date: Fri, 14 Aug 2009 15:54:22 +0200
Subject: Add header-object

---
 portato/eix/__init__.py |  7 ++++++-
 portato/eix/parser.py   | 44 +++++++++++++++++++++++++++++++++++---------
 2 files changed, 41 insertions(+), 10 deletions(-)

(limited to 'portato/eix')

diff --git a/portato/eix/__init__.py b/portato/eix/__init__.py
index 3c1958e..6768ca7 100644
--- a/portato/eix/__init__.py
+++ b/portato/eix/__init__.py
@@ -18,7 +18,7 @@ from . import exceptions as ex
 from ..helper import debug
 
 class EixReader(object):
-    supported_version = (28,)
+    supported_versions = (28,)
         
     def __init__ (self, filename):
         self.filename = filename
@@ -31,10 +31,15 @@ class EixReader(object):
                 raise ex.UnsupportedVersionError(self.version)
 
             debug("Started EixReader for version %s.", self.version)
+
+            self.file.seek(0)
         except:
             self.close()
             raise
 
+    def header (self):
+        return parser.header(self.file)
+
     def close (self):
         self.file.close()
         debug("EixReader closed.")
diff --git a/portato/eix/parser.py b/portato/eix/parser.py
index 676cdd5..90c569b 100644
--- a/portato/eix/parser.py
+++ b/portato/eix/parser.py
@@ -15,7 +15,8 @@ from __future__ import absolute_import, with_statement
 import os
 import struct
 
-from .helper import debug
+from ..helper import debug
+from functools import partial
 
 from . import exceptions as ex
 
@@ -30,7 +31,7 @@ def _get_bytes (file, length, expect_list = False):
     else:
         return struct.unpack("%sB" % length, s)
 
-def number (file):
+def number (file, skip = False):
     n = _get_bytes(file, 1)
 
     if n < 0xFF:
@@ -49,10 +50,16 @@ def number (file):
         value = n << (count*8)
 
         if count > 0:
-            rest = _get_bytes(file, count, expect_list = True)
 
-            for i, r in enumerate(rest):
-                value += r << ((count - i - 1)*8)
+            if skip:
+                file.seek(count, os.SEEK_CUR)
+                return
+            
+            else:
+                rest = _get_bytes(file, count, expect_list = True)
+
+                for i, r in enumerate(rest):
+                    value += r << ((count - i - 1)*8)
         
     return value
 
@@ -65,11 +72,15 @@ def vector (file, get_type, skip = False):
     else:
         return [get_type(file) for i in range(nelems)]
 
+def typed_vector(type):
+    return partial(vector, get_type = type)
+
 def string (file, skip = False):
     nelems = number(file)
 
     if skip:
         file.seek(nelems, os.SEEK_CUR)
+        return
     else:
         s = file.read(nelems)
 
@@ -92,17 +103,32 @@ class LazyElement (object):
         self._value = None
 
         self.pos = file.tell()
-        get_type(skip=True) # skip it for the moment
+        get_type(file, skip=True) # skip it for the moment
 
     @property
     def value (self):
         if self._value is None:
             old_pos = self.file.tell()
-            self.file.seek(self.pos)
-            self._value = self.get_type(skip = False)
-            self.file.seek(old_pos)
+            self.file.seek(self.pos, os.SEEK_SET)
+            self._value = self.get_type(self.file, skip = False)
+            self.file.seek(old_pos, os.SEEK_SET)
         
         return self._value
 
     def __call__ (self):
         return self.value
+
+class header (object):
+    def __init__ (self, file):
+        def LE (t):
+            return LazyElement(t, file)
+
+        self.version = LE(number)
+        self.ncats = LE(number)
+        self.overlays = LE(typed_vector(overlay))
+        self.provide = LE(typed_vector(string))
+        self.licenses = LE(typed_vector(string))
+        self.keywords = LE(typed_vector(string))
+        self.useflags = LE(typed_vector(string))
+        self.slots = LE(typed_vector(string))
+        self.sets = LE(typed_vector(string))
-- 
cgit v1.2.3