configurationengine/source/testautomation/testautomation/zip_dir.py
changeset 0 2e8eeb919028
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/configurationengine/source/testautomation/testautomation/zip_dir.py	Thu Mar 11 17:04:37 2010 +0200
@@ -0,0 +1,76 @@
+#
+# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
+# All rights reserved.
+# This component and the accompanying materials are made available
+# under the terms of "Eclipse Public License v1.0"
+# which accompanies this distribution, and is available
+# at the URL "http://www.eclipse.org/legal/epl-v10.html".
+#
+# Initial Contributors:
+# Nokia Corporation - initial contribution.
+#
+# Contributors:
+#
+# Description:
+#
+
+import os, re, zipfile
+
+SVN_IGNORE_PATTERN = r'(.*/|.*\\|^)\.svn(/.*|\\.*|$)'
+
+def zip_dir(source_dir, target_file, ignore_patterns=[]):
+    """
+    Archive the contents of a directory into a zip file.
+    """
+    if not os.path.isdir(source_dir):
+        raise RuntimeError("'%s' does not exist or is not a directory" % dir)
+
+    target_dir = os.path.dirname(target_file)
+    if target_dir != '' and not os.path.exists(target_dir):
+        os.makedirs(target_dir)
+    
+    def is_ignored(entry_name):
+        for pat in ignore_patterns:
+            if re.match(pat, entry_name) != None:
+                return True
+        return False
+    
+    zf = zipfile.ZipFile(target_file, 'w', compression=zipfile.ZIP_DEFLATED)
+    try:
+        source_dir = os.path.abspath(source_dir).replace('\\', '/')
+        if not source_dir.endswith('/'): source_dir += '/'
+        
+        empty_dirs = []
+        for root, dirs, files in os.walk(source_dir):
+            # Construct the ZIP entry name of the current directory
+            root_dir_zip_name = root.replace('\\', '/')[len(source_dir):] + '/'
+            
+            if is_ignored(root_dir_zip_name):
+                continue
+            
+            # Create a dictionary of the files that should be written to the
+            # ZIP file
+            filtered_files = {}
+            for fname in files:
+                if root_dir_zip_name == '/':    zip_name = fname
+                else:                           zip_name = root_dir_zip_name + fname
+                if not is_ignored(zip_name):
+                    filtered_files[zip_name] = os.path.join(root, fname)
+            
+            # If the current directory is a sub-directory of an empty directory candidate,
+            # remove the base directory from the sub-directory candidates
+            empty_dirs = filter(lambda n: not root_dir_zip_name.startswith(n), empty_dirs)
+            
+            if len(filtered_files) == 0:
+                # If there are no files, the directory is possibly an empty directory
+                # (though it can still contain sub-directories that contain something)
+                empty_dirs.append(root_dir_zip_name)
+            else:
+                for zip_name, path in filtered_files.iteritems():
+                    zf.write(path, zip_name, zipfile.ZIP_DEFLATED)
+            
+        # Write empty directories
+        for zip_name in empty_dirs:
+            zf.writestr(zip_name, '')
+    finally:
+        zf.close()
\ No newline at end of file