1 files changed, 167 insertions, 0 deletions
diff --git a/mwlib/zipwiki.py b/mwlib/zipwiki.py
new file mode 100755
index 0000000..c6893b8
--- /dev/null
+++ b/mwlib/zipwiki.py
@@ -0,0 +1,167 @@
+#! /usr/bin/env python
+
+# Copyright (c) 2008, PediaPress GmbH
+# See README.txt for additional licensing information.
+
+import os
+import shutil
+import simplejson
+import tempfile
+from zipfile import ZipFile
+
+from mwlib.metabook import MetaBook
+from mwlib import uparser
+
+class Wiki(object):
+    def __init__(self, zipfile):
+        """
+        @type zipfile: basestring or ZipFile
+        """
+        
+        if isinstance(zipfile, ZipFile):
+            self.zf = zipfile
+        else:
+            self.zf = ZipFile(zipfile)
+        self.metabook = MetaBook()
+        self.metabook.loadJson(self.zf.read("metabook.json"))
+        content = simplejson.loads(self.zf.read('content.json'))
+        self.articles = content['articles']
+        self.templates = content['templates']
+    
+    def _getArticle(self, title, revision=None):
+        try:
+            article = self.articles[title]
+            if revision is None or article['revision'] == revision:
+                return article
+        except KeyError:
+            pass
+        return None
+    
+    def getRawArticle(self, title, revision=None):
+        article = self._getArticle(title, revision=revision)
+        if article:
+            return article['content']
+        return None
+    
+    def getParsedArticle(self, title, revision=None):
+        raw = self.getRawArticle(title, revision=revision)
+        if raw is None:
+            return None
+        a = uparser.parseString(title=title, raw=raw, wikidb=self)
+        return a
+    
+    def getURL(self, title, revision=None):
+        article = self._getArticle(title, revision=revision)
+        if article:
+            return article['url']
+        return None
+    
+    def getAuthors(self, title, revision=None):
+        article = self._getArticle(title, revision=revision)
+        if article:
+            return article.get('authors', [])
+        return None
+    
+    def getTemplate(self, name, followRedirects=True):
+        try:
+            return self.templates[name]['content']
+        except KeyError:
+            pass
+        return None
+    
+
+class ImageDB(object):
+    def __init__(self, zipfile, tmpdir=None):
+        """
+        @type zipfile: basestring or ZipFile
+        """
+        
+        if isinstance(zipfile, ZipFile):
+            self.zf = zipfile
+        else:
+            self.zf = ZipFile(zipfile)
+        content = simplejson.loads(self.zf.read('content.json'))
+        self.images = content['images']
+        self._tmpdir = tmpdir
+        self.diskpaths = {}
+    
+    @property
+    def tmpdir(self):
+        if self._tmpdir is None:
+            self._tmpdir = unicode(tempfile.mkdtemp())
+        return self._tmpdir
+    
+    def getDiskPath(self, name, size=None):
+        try:
+            return self.diskpaths[name]
+        except KeyError:
+            pass
+        try:
+            data = self.zf.read('images/%s' % name.replace("'", '-').encode('utf-8'))
+        except KeyError: # no such file
+            return None
+        
+        try:
+            ext = '.' + name.rsplit('.', 1)[1]
+        except IndexError:
+            ext = ''
+        if ext.lower() == '.svg':
+            ext = '.svg.png'
+        elif ext.lower() == '.gif':
+            ext = '.gif.png'
+        res = os.path.join(self.tmpdir, 'image%04d%s' % (len(self.diskpaths), ext))
+        self.diskpaths[name] = res
+        f=open(res, "wb")
+        f.write(data)
+        f.close()
+        return res
+    
+    def getLicense(self, name):
+        try:
+            return self.images[name]['license']
+        except KeyError:
+            return None
+    
+    def getPath(self):
+        raise NotImplemented('getPath() does not work with zipwiki.ImageDB!')
+    
+    def getURL(self, name):
+        try:
+            return self.images[name]['url']
+        except KeyError:
+            return None
+    
+    def clean(self):
+        if self._tmpdir:
+            shutil.rmtree(self._tmpdir, ignore_errors=True)
+    
+
+
+
+class FakeImageDB(ImageDB):
+
+    imagedata = '\x89PNG\r\n\x1a\n\x00\x00\x00\rIHDR\x00\x00\x03 \x00\x00\x01\xe0\x01\x03\x00\x00\x00g\xc9\x9b\xb6\x00\x00\x00\x01sRGB\x00\xae\xce\x1c\xe9\x00\x00\x00\x06PLTE\xff\xff\xff\x00\x00\x00U\xc2\xd3~\x00\x00\x00\tpHYs\x00\x00\x0b\x13\x00\x00\x0b\x13\x01\x00\x9a\x9c\x18\x00\x00\x00EIDATx\xda\xed\xc1\x01\x01\x00\x00\x00\x82 \xff\xafnH@\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00/\x06\xbd`\x00\x01`<5\x84\x00\x00\x00\x00IEND\xaeB`\x82'
+
+    def __init__(self, tmpdir=None):
+        """
+        @type zipfile: basestring or ZipFile
+        """
+        self._tmpdir = tmpdir        
+    
+    def getDiskPath(self, name, size=None):
+        res = os.path.join(self.tmpdir, 'blank.png')
+        if not os.path.exists(res):
+            open(res, "w").write(self.imagedata)
+        return res
+    
+    def getPath(self):
+        raise NotImplemented('getPath() does not work with zipwiki.FakeImageDB!')
+    
+    def getURL(self, name):
+        raise NotImplemented('getURL() does not work with zipwiki.FakeImageDB!')
+    
+    def getLicense(self, name):
+        raise NotImplemented('getLicense() does not work with zipwiki.FakeImageDB!')
+    
+
+