BaseTools: Replace StringIO.StringIO with io.BytesIO

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

diff --git a/BaseTools/Source/Python/GenFds/CapsuleData.py b/BaseTools/Source/Python/GenFds/CapsuleData.py
index 9dc55e5dbf7b1df4fc4df74df6a97ba56f2dfbb6..83b2731110bcead8a4ecb97c74b1336dba64adda 100644 (file)
--- a/BaseTools/Source/Python/GenFds/CapsuleData.py
+++ b/BaseTools/Source/Python/GenFds/CapsuleData.py
@@ -17,7 +17,7 @@
 #\r
 import Ffs\r
 from GenFdsGlobalVariable import GenFdsGlobalVariable\r
 #\r
 import Ffs\r
 from GenFdsGlobalVariable import GenFdsGlobalVariable\r

-import StringIO\r
+from io import BytesIO\r

 from struct import pack\r
 import os\r
 from Common.Misc import SaveFileOnChange\r
 from struct import pack\r
 import os\r
 from Common.Misc import SaveFileOnChange\r


@@ -82,7 +82,7 @@ class CapsuleFv (CapsuleData):
         if self.FvName.find('.fv') == -1:\r
             if self.FvName.upper() in GenFdsGlobalVariable.FdfParser.Profile.FvDict:\r
                 FvObj = GenFdsGlobalVariable.FdfParser.Profile.FvDict[self.FvName.upper()]\r
         if self.FvName.find('.fv') == -1:\r
             if self.FvName.upper() in GenFdsGlobalVariable.FdfParser.Profile.FvDict:\r
                 FvObj = GenFdsGlobalVariable.FdfParser.Profile.FvDict[self.FvName.upper()]\r

-                FdBuffer = StringIO.StringIO('')\r
+                FdBuffer = BytesIO('')\r

                 FvObj.CapsuleName = self.CapsuleName\r
                 FvFile = FvObj.AddToBuffer(FdBuffer)\r
                 FvObj.CapsuleName = None\r
                 FvObj.CapsuleName = self.CapsuleName\r
                 FvFile = FvObj.AddToBuffer(FdBuffer)\r
                 FvObj.CapsuleName = None\r