1 files changed, 83 insertions, 0 deletions
diff --git a/mwlib/recorddb.py b/mwlib/recorddb.py
new file mode 100755
index 0000000..fe895cf
--- /dev/null
+++ b/mwlib/recorddb.py
@@ -0,0 +1,83 @@
+#! /usr/bin/env python
+
+# Copyright (c) 2007-2008 PediaPress GmbH
+# See README.txt for additional licensing information.
+
+import simplejson
+import zipfile
+from mwlib import uparser, parser
+import mwlib.log
+log = mwlib.log.Log("zip")
+
+
+class RecordDB(object):
+    def __init__(self, db):
+        assert db is not None, "db must not be None"
+        self.db = db
+        self.articles = {}
+        self.templates = {}
+        
+    def getRawArticle(self, name, revision=None):
+        r = self.db.getRawArticle(name, revision=revision)
+        self.articles[name] = {
+            'revision': revision,
+            'content-type': 'text/x-wiki',
+            'content': r,
+            'url': self.db.getURL(name, revision=revision),
+            'authors': self.db.getAuthors(name, revision=revision),
+        }
+        return r
+    
+    def getTemplate(self, name, followRedirects=False):
+        r = self.db.getTemplate(name, followRedirects=followRedirects)
+        self.templates[name] = {
+            'content-type': 'text/x-wiki',
+            'content': r,
+        }
+        return r
+    
+
+class ZipfileCreator(object):
+    def __init__(self, zf, wikidb=None, imgdb=None):
+        self.zf = zf
+        self.db = RecordDB(wikidb)
+        self.images = {}
+        self.imgdb = imgdb
+
+    def addObject(self, name, value):
+        """
+        @type name: unicode
+        
+        @type value: str
+        """
+        
+        self.zf.writestr(name.encode('utf-8'), value)
+    
+    def addArticle(self, title, revision=None):
+        a = uparser.parseString(title, revision=revision, wikidb=self.db)
+        for x in a.allchildren():
+            if isinstance(x, parser.ImageLink):
+                name = x.target
+                self.images[name] = {}
+    
+    def writeImages(self, size=None):
+        if self.imgdb is None:
+            return
+        
+        for name in sorted(self.images.keys()):
+            dp = self.imgdb.getDiskPath(name, size=size)
+            if dp is None:
+                continue
+            self.zf.write(dp, (u"images/%s" % name.replace("'", '-')).encode("utf-8"))
+            self.images[name]['url'] = self.imgdb.getURL(name, size=size)
+            license = self.imgdb.getLicense(name)
+            if license:
+                self.images[name]['license'] = license
+    
+    def writeContent(self):
+        self.addObject('content.json', simplejson.dumps(dict(
+            articles=self.db.articles,
+            templates=self.db.templates,
+            images=self.images,
+        )))
+