configurationengine/source/testautomation/testautomation/unzip_file.py
author terytkon
Thu, 11 Mar 2010 17:04:37 +0200
changeset 0 2e8eeb919028
permissions -rw-r--r--
Adding EPL version of configurationengine.
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     1
#
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     2
# Copyright (c) 2009 Nokia Corporation and/or its subsidiary(-ies).
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     3
# All rights reserved.
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     4
# This component and the accompanying materials are made available
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     5
# under the terms of "Eclipse Public License v1.0"
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     6
# which accompanies this distribution, and is available
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     7
# at the URL "http://www.eclipse.org/legal/epl-v10.html".
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     8
#
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
     9
# Initial Contributors:
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    10
# Nokia Corporation - initial contribution.
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    11
#
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    12
# Contributors:
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    13
#
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    14
# Description: 
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    15
#
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    16
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    17
import os, zipfile, shutil
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    18
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    19
def unzip_file(file, dir, delete_if_exists=False):
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    20
    if os.path.exists(dir):
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    21
        if delete_if_exists == True:
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    22
            shutil.rmtree(dir)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    23
        else:
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    24
            raise RuntimeError("Directory '%s' already exists" % dir)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    25
    if not os.path.exists(file):
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    26
        raise RuntimeError("File '%s' does not exist" % file)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    27
    
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    28
    os.makedirs(dir)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    29
    zf = zipfile.ZipFile(file)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    30
    
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    31
    try:
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    32
        for name in zf.namelist():
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    33
            if name.endswith('/'):
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    34
                path = os.path.join(dir, name)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    35
                if not os.path.exists(path):
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    36
                    os.makedirs(path)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    37
            else:
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    38
                path = os.path.join(dir, name)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    39
                dirname = os.path.dirname(path)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    40
                if dirname != '' and not os.path.exists(dirname):
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    41
                    os.makedirs(dirname)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    42
                f = open(path, 'wb')
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    43
                try:        f.write(zf.read(name))
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    44
                finally:    f.close()
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    45
    finally:
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    46
        zf.close()
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    47
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    48
def unzip_file_if_newer(file, dir):
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    49
    if not os.path.exists(file):
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    50
        raise RuntimeError("File '%s' does not exist" % file)
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    51
    
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    52
    if not os.path.exists(dir) or os.stat(file)[8] > os.stat(dir)[8]:
2e8eeb919028 Adding EPL version of configurationengine.
terytkon
parents:
diff changeset
    53
        unzip_file(file, dir, True)