1 files changed, 126 insertions, 0 deletions
diff --git a/setup_new_wiki.py b/setup_new_wiki.py
new file mode 100755
index 0000000..b52c24a
--- /dev/null
+++ b/setup_new_wiki.py
@@ -0,0 +1,126 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2011, One Laptop Per Child
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
+#
+import sys
+import os
+import shutil
+import zipfile
+from fnmatch import fnmatch
+from sugar.activity import bundlebuilder
+
+INCLUDE_DIRS = ['activity', 'binarylibs', 'icons', 'locale', 'bin',
+                'mwlib', 'po', 'seek-bzip2', 'static', 'tools2']
+IGNORE_FILES = ['.gitignore', 'MANIFEST', '*.pyc', '*~', '*.bak', 'pseudo.po']
+
+
+class WikiXOPackager(bundlebuilder.XOPackager):
+
+    def __init__(self, builder, data_file):
+        bundlebuilder.XOPackager.__init__(self, builder)
+        self.data_file = data_file
+
+    def package(self):
+        bundle_zip = zipfile.ZipFile(self.package_path, 'w',
+                                     zipfile.ZIP_DEFLATED)
+
+        for f in self.list_files(self.config.source_dir, True):
+            if os.path.exists(os.path.join(self.config.source_dir, f)):
+                bundle_zip.write(os.path.join(self.config.source_dir, f),
+                                 os.path.join(self.config.bundle_root_dir, f))
+
+        if self.data_file is not None:
+            # Add the data files
+            needed_sufix = ['.processed.bz2', '.processed.bz2t']
+            for sufix in needed_sufix:
+                data_file = self.data_file + sufix
+                print "Add %s" % data_file
+                bundle_zip.write(data_file,
+                                 os.path.join(self.config.bundle_root_dir,
+                                              data_file))
+
+            data_path = os.path.dirname(self.data_file)
+
+            index_file = os.path.join(data_path, 'search.db')
+            print "Add %s" % index_file
+            bundle_zip.write(index_file,
+                         os.path.join(self.config.bundle_root_dir, index_file))
+
+            # add images
+            images_path = os.path.join(data_path, 'images')
+            if os.path.exists(images_path):
+                print "Adding images"
+                for f in self.list_files(images_path):
+                    bundle_zip.write(os.path.join(images_path, f),
+                        os.path.join(self.config.bundle_root_dir,
+                            images_path, f))
+
+        bundle_zip.close()
+
+    def list_files(self, base_dir, filter_directories=False):
+        if filter_directories:
+            include_dirs = INCLUDE_DIRS
+        else:
+            include_dirs = None
+
+        ignore_files = IGNORE_FILES
+        result = []
+
+        base_dir = os.path.abspath(base_dir)
+
+        for root, dirs, files in os.walk(base_dir):
+
+            if ignore_files:
+                for pattern in ignore_files:
+                    files = [f for f in files if not fnmatch(f, pattern)]
+
+            rel_path = root[len(base_dir) + 1:]
+            for f in files:
+                result.append(os.path.join(rel_path, f))
+
+            if root == base_dir:
+                n = 0
+                while n < len(dirs):
+                    directory = dirs[n]
+                    if include_dirs is not None and \
+                        not directory in include_dirs:
+                        print "** Ignoring directory", directory
+                        dirs.remove(directory)
+                    else:
+                        n = n + 1
+        return result
+
+
+if __name__ == '__main__':
+
+    if len(sys.argv) < 2:
+        data_file = None
+        lang = 'base'
+    else:
+        data_file = sys.argv[1]
+
+        lang = data_file[:data_file.find('/')]
+
+    print "Lang:", lang
+    # copy activty/activity.info.lang as activty/activity.info
+    f = 'activity/activity.info.' + lang
+    if os.path.exists(f):
+        shutil.copyfile(f, 'activity/activity.info')
+
+    config = bundlebuilder.Config()
+    packager = WikiXOPackager(bundlebuilder.Builder(config), data_file)
+    packager.package()