Revert BaseTools: PYTHON3 migration

[mirror_edk2.git] / BaseTools / Source / Python / Eot / InfParserLite.py

diff --git a/BaseTools/Source/Python/Eot/InfParserLite.py b/BaseTools/Source/Python/Eot/InfParserLite.py
index 7de88990e4536c929e53953522d9a640e90df5e5..88d7e7d58e0bbe022240c83382c8693ccdd2544a 100644 (file)
--- a/BaseTools/Source/Python/Eot/InfParserLite.py
+++ b/BaseTools/Source/Python/Eot/InfParserLite.py
@@ -22,8 +22,8 @@ from Common.DataType import *
 from CommonDataClass.DataClass import *\r
 from Common.Identification import *\r
 from Common.StringUtils import *\r
-from Eot.Parser import *\r
-from Eot import Database\r
+from .Parser import *\r
+from . import Database\r
 \r
 ## EdkInfParser() class\r
 #\r
@@ -153,3 +153,21 @@ class EdkInfParser(object):
         self.ParserSource(CurrentSection, SectionItemList, ArchList, ThirdList)\r
         #End of For\r
 \r
+##\r
+#\r
+# This acts like the main() function for the script, unless it is 'import'ed into another\r
+# script.\r
+#\r
+if __name__ == '__main__':\r
+    EdkLogger.Initialize()\r
+    EdkLogger.SetLevel(EdkLogger.QUIET)\r
+\r
+    Db = Database.Database('Inf.db')\r
+    Db.InitDatabase()\r
+    P = EdkInfParser(os.path.normpath("C:\Framework\Edk\Sample\Platform\Nt32\Dxe\PlatformBds\PlatformBds.inf"), Db, '', '')\r
+    for Inf in P.Sources:\r
+        print(Inf)\r
+    for Item in P.Macros:\r
+        print(Item, P.Macros[Item])\r
+\r
+    Db.Close()\r