--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/bin/extract.py Tue Jul 06 14:53:09 2010 +0300
@@ -0,0 +1,173 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+#
+# ============================================================================
+# Name : extract.py
+# Part of : Hb
+# Description : Hb themes script for extracting theme archives
+# Version : %version: %
+#
+# Copyright (c) 2008-2010 Nokia. All rights reserved.
+# This material, including documentation and any related computer
+# programs, is protected by copyright controlled by Nokia. All
+# rights are reserved. Copying, including reproducing, storing,
+# adapting or translating, any or all of this material requires the
+# prior written consent of Nokia. This material also contains
+# confidential information which may not be disclosed to others
+# without the prior written consent of Nokia.
+# ============================================================================
+
+import os
+import re
+import sys
+import shutil
+import zipfile
+import fnmatch
+import optparse
+
+# ============================================================================
+# Globals
+# ============================================================================
+VERBOSE = False
+INPUT_DIR = os.getcwd()
+OUTPUT_DIR = os.getcwd()
+INCLUDE = None
+EXCLUDE = None
+
+# ============================================================================
+# OptionParser
+# ============================================================================
+class OptionParser(optparse.OptionParser):
+ def __init__(self):
+ optparse.OptionParser.__init__(self)
+ self.add_option("-v", "--verbose", action="store_true", dest="verbose",
+ help="print verbose information about each step of the sync process")
+ self.add_option("-q", "--quiet", action="store_false", dest="verbose",
+ help="do not print information about each step of the sync process")
+ self.add_option("-i", "--input", dest="input", metavar="dir",
+ help="specify the input <dir> (default %s)" % INPUT_DIR)
+ self.add_option("-o", "--output", dest="output", metavar="dir",
+ help="specify the output <dir> (default %s)" % OUTPUT_DIR)
+ self.add_option("--include", dest="include", action="append", metavar="pattern",
+ help="specify the include <pattern> (default %s)" % INCLUDE)
+ self.add_option("--exclude", dest="exclude", action="append", metavar="pattern",
+ help="specify the exclude <pattern> (default %s)" % EXCLUDE)
+
+# ============================================================================
+# Utils
+# ============================================================================
+if not hasattr(os.path, "relpath"):
+ def relpath(path, start=os.curdir):
+ abspath = os.path.abspath(path)
+ absstart = os.path.abspath(start)
+ if abspath == absstart:
+ return "."
+ i = len(absstart)
+ if not absstart.endswith(os.path.sep):
+ i += len(os.path.sep)
+ if not abspath.startswith(absstart):
+ i = 0
+ return abspath[i:]
+ os.path.relpath = relpath
+
+def extract(archive):
+ global INPUT_DIR, OUTPUT_DIR
+
+ path, filename = os.path.split(archive)
+ relpath = os.path.relpath(path, INPUT_DIR)
+ outpath = os.path.join(OUTPUT_DIR, relpath)
+
+ # extract
+ zip = zipfile.ZipFile(archive)
+ for name in zip.namelist():
+ data = zip.read(name)
+ outfile = os.path.join(outpath, name)
+ # overwrite only if different size
+ if not os.path.exists(outfile) or os.path.getsize(outfile) != len(data):
+ file = open(outfile, "w")
+ file.write(data)
+ file.close()
+
+def include_exclude(filepath):
+ global INCLUDE, EXCLUDE
+ result = True
+ if INCLUDE != None:
+ for pattern in INCLUDE:
+ if not fnmatch.fnmatch(filepath, pattern):
+ result = False
+ if EXCLUDE != None:
+ for pattern in EXCLUDE:
+ if fnmatch.fnmatch(filepath, pattern):
+ result = False
+ return result
+
+# ============================================================================
+# main()
+# ============================================================================
+def main():
+ global VERBOSE, INPUT_DIR, OUTPUT_DIR, INCLUDE, EXCLUDE
+
+ parser = OptionParser()
+ (options, args) = parser.parse_args()
+
+ if options.verbose != None:
+ VERBOSE = options.verbose
+ if options.input != None:
+ INPUT_DIR = options.input
+ if options.output != None:
+ OUTPUT_DIR = options.output
+ if options.include != None:
+ INCLUDE = options.include
+ if options.exclude != None:
+ EXCLUDE = options.exclude
+
+ extracted = 0
+ copied = 0
+ omitted = 0
+ workpath = None
+ newline = False
+ sys.stdout.write("Processing: ")
+ for root, dirs, files in os.walk(INPUT_DIR):
+ for file in files:
+ filepath = os.path.join(root, file)
+ extension = os.path.splitext(file)[1]
+ if include_exclude(filepath):
+ # ensure that output dir exists
+ relfilepath = os.path.relpath(filepath, INPUT_DIR)
+ relpath = os.path.split(relfilepath)[0]
+ outpath = os.path.join(OUTPUT_DIR, relpath)
+ if not os.path.exists(outpath):
+ os.makedirs(outpath)
+
+ # output processing dir info
+ tmppath = os.path.split(filepath)[0]
+ if tmppath != workpath:
+ if workpath != None:
+ newline = True
+ sys.stdout.write("\n ")
+ workpath = tmppath
+ sys.stdout.write(os.path.relpath(workpath, INPUT_DIR).replace("\\", "/"))
+ sys.stdout.write(".")
+ else:
+ sys.stdout.write(".")
+
+ # extract zips
+ if extension == ".zip":
+ extracted += 1
+ extract(filepath)
+ # copy others
+ else:
+ copied += 1
+ shutil.copy(filepath, outpath)
+ else:
+ omitted += 1
+ if newline:
+ sys.stdout.write("\n")
+ print " ==> %s archives(s) extracted" % extracted
+ print " ==> %s file(s) copied" % copied
+ print " ==> %s file(s) omitted" % omitted
+
+ return 0
+
+if __name__ == "__main__":
+ sys.exit(main())