initial implementation

pypcaxis.py

@@ -0,0 +1,79 @@
+import re
+from itertools import product
+from operator import mul
+
+
+class Dimension(object):
+
+    def __init__(self, title, values):
+        self.title = title
+        self.values = values
+
+    def __len__(self):
+        return len(self.values)
+
+
+class Table(object):
+
+    def __init__(self):
+        self.dimensions = []
+        self.data = None
+
+    def add_dimension(self, dimension):
+        self.dimensions.append(dimension)
+
+    def get_by(self, title, value):
+        #FIXME does not work!!!
+        title_index = [dim.title for dim in self.dimensions].index(title)
+        dims = [dim.values for dim in self.dimensions]
+        dims[title_index] = [value]
+        table = Table()
+        table.dimension = [d for d in self.dimensions if d.title != title]
+        table.data = [self.get(*criteria) for criteria in reversed(list(product(*dims)))]
+        return table
+
+    def get(self, *criteria):
+        dim_lenghts = [len(dim) for dim in self.dimensions]
+        dim_indices = [dim.values.index(c) for (dim, c)
+                       in zip(self.dimensions, criteria)]
+        return self.data[sum(reduce(mul, dim_lenghts[i+1:], 1) * index
+                         for i, index in enumerate(dim_indices))]
+
+
+def parse(path):
+    data = read_data(path)
+    value_regex = re.compile(r'VALUES\(\"(.*)\"\)')
+    table = Table()
+    for item in data:
+        if not item:
+            continue
+        name, values = [t.strip() for t in item.split('=', 1)]
+        value_match = value_regex.match(name)
+        if value_match:
+            title = value_match.group(1)
+            table.add_dimension(create_dimension(title, values))
+        if name == 'DATA':
+            table.data = [i.strip() for i in values.split(' ')]
+    return table
+
+
+def read_data(path):
+    return [t.strip() for t in
+            open(path).read().decode('ISO-8859-1').split(';')]
+
+
+def create_dimension(title, values):
+    # values are defined like: "foo","bar","zap"
+    values = values.replace('\r\n', '')[1:-1].split('","')
+    return Dimension(title, values)
+
+
+if __name__ == '__main__':
+    table = parse('examples/tulot.px')
+    print table.get('2008', 'Tuusula - Tusby', 'Veronalaiset tulot, mediaani')
+    print table.get('2009', 'Tuusula - Tusby', 'Veronalaiset tulot, mediaani')
+    print table.get('2007', u'Hyvink\xe4\xe4 - Hyvinge', 'Tulonsaajia')
+    print table.get_by('Vuosi', '2007').get(u'Hyvink\xe4\xe4 - Hyvinge', 'Tulonsaajia')
+    print table.get('2008', 'Tuusula - Tusby', 'Veronalaiset tulot, mediaani')
+    table = parse('examples/vaalit.px')
+    print table.get('Uudenmaan vaalipiiri', 'VIHR', u'Yhteens\xe4', u'78 vuotta')