1 files changed, 144 insertions, 0 deletions
diff --git a/setup.py b/setup.py
new file mode 100755
index 0000000..77c1b6e
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,144 @@
+#!/usr/bin/env python
+#
+# Copyright (C) 2007, 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
+#
+from sugar.activity import bundlebuilder
+import sys
+import os
+import shutil
+import urllib2
+import subprocess
+import zipfile
+import ConfigParser
+import codecs
+
+
+def download_file(url):
+    web_file = urllib2.urlopen(url)
+    local_file_name = 'cache/' + url.split('/')[-1]
+    local_file = open(local_file_name, 'w')
+    chunk = 4096
+    while 1:
+        data = web_file.read(chunk)
+        if not data:
+            print "done."
+            break
+        local_file.write(data)
+        print ".",
+    web_file.close()
+    local_file.close()
+    return local_file_name
+
+if len(sys.argv) == 1:
+    print "Use ./setup prepare lang"
+    exit()
+
+prepare_ok = True
+if sys.argv[1] == 'prepare':
+    if len(sys.argv) < 3:
+        print 'You must select a language. For example: ./setup.py prepare es'
+        prepare_ok = False
+    else:
+        prepare_ok = False
+        language = sys.argv[2]
+        print "Preparing", language
+        config = ConfigParser.ConfigParser()
+        config.readfp(open('data_repository.cfg'))
+        data_repository = config.get('repository', 'url')
+
+        last_version = config.get('last_versions', language)
+        print "Check data", last_version
+        already_downloaded = False
+        if not os.path.exists('cache'):
+            os.mkdir('cache')
+
+        flag_file_name = 'download_complete.' + language
+        if os.path.exists(flag_file_name):
+            flag_file = open(flag_file_name, 'r')
+            for line in flag_file:
+                if line == last_version:
+                    already_downloaded = True
+            flag_file.close()
+        if not already_downloaded:
+            url_data = data_repository + last_version
+            url_md5 = url_data.replace('.zip', '.md5sum')
+            print "Downloading data", url_data
+            print "md5 data", url_md5
+            local_md5_file_name = download_file(url_md5)
+            local_data_file_name = download_file(url_data)
+        else:
+            local_data_file_name = 'cache/' + last_version.split('/')[-1]
+            local_md5_file_name = local_data_file_name.replace('.zip', \
+                    '.md5sum')
+
+        real_md5 = subprocess.check_output(['/usr/bin/md5sum', \
+                local_data_file_name])
+        real_md5_value = real_md5.split(' ')[0]
+        print "md5sum %s = %s" % (local_data_file_name, real_md5_value)
+        md5_ok = False
+        for line in open(local_md5_file_name):
+            if line.find(real_md5_value) == 0:
+                md5_ok = True
+        if md5_ok:
+            print "MD5 Ok"
+            # Unzip data files
+            zf = zipfile.ZipFile(local_data_file_name, 'r')
+            list_data_files = zf.namelist()
+            root_data_directory = list_data_files[0]
+            # Create temporary directory
+            tmp_directory = 'TMP'
+            if not os.path.exists(tmp_directory):
+                os.mkdir(tmp_directory)
+            zf.extractall(tmp_directory)
+            zf.close()
+            if os.path.exists(root_data_directory):
+                shutil.rmtree(root_data_directory)
+            shutil.move(tmp_directory + '/' + root_data_directory, root_data_directory)
+            shutil.rmtree(tmp_directory)
+            # Create flag file
+            flag_file = open(flag_file_name, 'w')
+            flag_file.write(last_version)
+            flag_file.close()
+
+            # Copy activity.info file
+            print "Create activity.info for", language
+            shutil.copyfile('activity/activity.info.' + language,
+                'activity/activity.info')
+
+            # Create MANIFEST
+            print "Create MANIFEST"
+            list_git_files = subprocess.check_output(['git', 'ls-files'])
+            manifest_file = codecs.open('MANIFEST', 'w', encoding='utf-8')
+            manifest_file.write('activity/activity.info\n')
+            for name_file in list_git_files:
+                manifest_file.write(name_file)
+            for name_file in list_data_files:
+                if not os.path.isdir(name_file):
+                    manifest_file.write(name_file + '\n')
+            manifest_file.close()
+
+            prepare_ok = True
+        else:
+            print "MD5 check error"
+
+
+if sys.argv[1] == 'fix_manifest':
+        print "Don't use fix_manifest. The MANIFEST is build in prepare stage"
+        prepare_ok = False
+
+if prepare_ok:
+    bundlebuilder.start('Wikipedia')