BaseTools/UPT: Support Multiple Installation

[mirror_edk2.git] / BaseTools / Source / Python / UPT / ReplacePkg.py

diff --git a/BaseTools/Source/Python/UPT/ReplacePkg.py b/BaseTools/Source/Python/UPT/ReplacePkg.py
index 9f231f9bcd5ba6b391ef52bdac5478d7b5d7a866..efbf68a4ecc638a6b350d1c5268be9dfd672fe27 100644 (file)
--- a/BaseTools/Source/Python/UPT/ReplacePkg.py
+++ b/BaseTools/Source/Python/UPT/ReplacePkg.py
@@ -1,7 +1,7 @@
 ## @file\r
 # Replace distribution package.\r
 #\r
-# Copyright (c) 2014, Intel Corporation. All rights reserved.<BR>\r
+# Copyright (c) 2014 - 2017, Intel Corporation. All rights reserved.<BR>\r
 #\r
 # This program and the accompanying materials are licensed and made available \r
 # under the terms and conditions of the BSD License which accompanies this \r
@@ -99,9 +99,9 @@ def Main(Options = None):
             DistFile.Close()\r
         if ContentZipFile:\r
             ContentZipFile.Close()\r
-        if GlobalData.gUNPACK_DIR:\r
-            rmtree(GlobalData.gUNPACK_DIR)\r
-            GlobalData.gUNPACK_DIR = None\r
+        for TempDir in GlobalData.gUNPACK_DIR:\r
+            rmtree(TempDir)\r
+        GlobalData.gUNPACK_DIR = []\r
         Logger.Quiet(ST.MSG_REMOVE_TEMP_FILE_DONE)        \r
 \r
     if ReturnCode == 0:\r