Revert BaseTools: PYTHON3 migration

[mirror_edk2.git] / BaseTools / Source / Python / GenFds / Capsule.py

diff --git a/BaseTools/Source/Python/GenFds/Capsule.py b/BaseTools/Source/Python/GenFds/Capsule.py
index 1c673949fa8009b8b6cd621aac6f290a1e898764..baa6e1dfa2ab8bc31f61aaaca0b8c0bc501ef82f 100644 (file)
--- a/BaseTools/Source/Python/GenFds/Capsule.py
+++ b/BaseTools/Source/Python/GenFds/Capsule.py
@@ -15,13 +15,13 @@
 ##\r
 # Import Modules\r
 #\r
+from __future__ import absolute_import\r
 from .GenFdsGlobalVariable import GenFdsGlobalVariable\r
 from .GenFdsGlobalVariable import FindExtendTool\r
 from CommonDataClass.FdfClass import CapsuleClassObject\r
 import Common.LongFilePathOs as os\r
 import subprocess\r
 from io import BytesIO\r
-from io import StringIO\r
 from Common.Misc import SaveFileOnChange\r
 from Common.Misc import PackRegistryFormatGuid\r
 import uuid\r
@@ -185,7 +185,7 @@ class Capsule (CapsuleClassObject) :
         #\r
         # The real capsule header structure is 28 bytes\r
         #\r
-        Header.write(b'\x00'*(HdrSize-28))\r
+        Header.write('\x00'*(HdrSize-28))\r
         Header.write(FwMgrHdr.getvalue())\r
         Header.write(Content.getvalue())\r
         #\r
@@ -247,7 +247,7 @@ class Capsule (CapsuleClassObject) :
     def GenCapInf(self):\r
         self.CapInfFileName = os.path.join(GenFdsGlobalVariable.FvDir,\r
                                    self.UiCapsuleName +  "_Cap" + '.inf')\r
-        CapInfFile = StringIO() #open (self.CapInfFileName , 'w+')\r
+        CapInfFile = BytesIO() #open (self.CapInfFileName , 'w+')\r
 \r
         CapInfFile.writelines("[options]" + T_CHAR_LF)\r
 \r