]> git.proxmox.com Git - mirror_edk2.git/blobdiff - BaseTools/Source/Python/Common/Expression.py
BaseTools: Migrate to the new octal literal
[mirror_edk2.git] / BaseTools / Source / Python / Common / Expression.py
index d7903c98b1d65ff90e9c923534dc66060ff4c899..9ff4f104256e7d84e5ae955a371eb3cd3e62850d 100644 (file)
@@ -12,6 +12,7 @@
 \r
 ## Import Modules\r
 #\r
+from __future__ import print_function\r
 from Common.GlobalData import *\r
 from CommonDataClass.Exceptions import BadExpression\r
 from CommonDataClass.Exceptions import WrnExpression\r
@@ -171,7 +172,7 @@ def ReplaceExprMacro(String, Macros, ExceptionList = None):
                 RetStr += '0'\r
             elif not InQuote:\r
                 Tklst = RetStr.split()\r
-                if Tklst and Tklst[-1] in ['IN', 'in'] and ExceptionList and Macro not in ExceptionList:\r
+                if Tklst and Tklst[-1] in {'IN', 'in'} and ExceptionList and Macro not in ExceptionList:\r
                     raise BadExpression(ERR_IN_OPERAND)\r
                 # Make sure the macro in exception list is encapsulated by double quote\r
                 # For example: DEFINE ARCH = IA32 X64\r
@@ -243,10 +244,10 @@ class ValueExpression(BaseExpression):
     def Eval(Operator, Oprand1, Oprand2 = None):\r
         WrnExp = None\r
 \r
-        if Operator not in ["==", "!=", ">=", "<=", ">", "<", "in", "not in"] and \\r
+        if Operator not in {"==", "!=", ">=", "<=", ">", "<", "in", "not in"} and \\r
             (type(Oprand1) == type('') or type(Oprand2) == type('')):\r
             raise BadExpression(ERR_STRING_EXPR % Operator)\r
-        if Operator in ['in', 'not in']:\r
+        if Operator in {'in', 'not in'}:\r
             if type(Oprand1) != type(''):\r
                 Oprand1 = IntToStr(Oprand1)\r
             if type(Oprand2) != type(''):\r
@@ -259,19 +260,19 @@ class ValueExpression(BaseExpression):
         }\r
 \r
         EvalStr = ''\r
-        if Operator in ["!", "NOT", "not"]:\r
+        if Operator in {"!", "NOT", "not"}:\r
             if type(Oprand1) == type(''):\r
                 raise BadExpression(ERR_STRING_EXPR % Operator)\r
             EvalStr = 'not Oprand1'\r
-        elif Operator in ["~"]:\r
+        elif Operator in {"~"}:\r
             if type(Oprand1) == type(''):\r
                 raise BadExpression(ERR_STRING_EXPR % Operator)\r
             EvalStr = '~ Oprand1'\r
         else:\r
-            if Operator in ["+", "-"] and (type(True) in [type(Oprand1), type(Oprand2)]):\r
+            if Operator in {"+", "-"} and (type(True) in {type(Oprand1), type(Oprand2)}):\r
                 # Boolean in '+'/'-' will be evaluated but raise warning\r
                 WrnExp = WrnExpression(WRN_BOOL_EXPR)\r
-            elif type('') in [type(Oprand1), type(Oprand2)] and type(Oprand1)!= type(Oprand2):\r
+            elif type('') in {type(Oprand1), type(Oprand2)} and type(Oprand1)!= type(Oprand2):\r
                 # == between string and number/boolean will always return False, != return True\r
                 if Operator == "==":\r
                     WrnExp = WrnExpression(WRN_EQCMP_STR_OTHERS)\r
@@ -284,10 +285,10 @@ class ValueExpression(BaseExpression):
                 else:\r
                     raise BadExpression(ERR_RELCMP_STR_OTHERS % Operator)\r
             elif TypeDict[type(Oprand1)] != TypeDict[type(Oprand2)]:\r
-                if Operator in ["==", "!=", ">=", "<=", ">", "<"] and set((TypeDict[type(Oprand1)], TypeDict[type(Oprand2)])) == set((TypeDict[type(True)], TypeDict[type(0)])):\r
+                if Operator in {"==", "!=", ">=", "<=", ">", "<"} and set((TypeDict[type(Oprand1)], TypeDict[type(Oprand2)])) == set((TypeDict[type(True)], TypeDict[type(0)])):\r
                     # comparison between number and boolean is allowed\r
                     pass\r
-                elif Operator in ['&', '|', '^', "and", "or"] and set((TypeDict[type(Oprand1)], TypeDict[type(Oprand2)])) == set((TypeDict[type(True)], TypeDict[type(0)])):\r
+                elif Operator in {'&', '|', '^', "and", "or"} and set((TypeDict[type(Oprand1)], TypeDict[type(Oprand2)])) == set((TypeDict[type(True)], TypeDict[type(0)])):\r
                     # bitwise and logical operation between number and boolean is allowed\r
                     pass\r
                 else:\r
@@ -307,10 +308,10 @@ class ValueExpression(BaseExpression):
         }\r
         try:\r
             Val = eval(EvalStr, {}, Dict)\r
-        except Exception, Excpt:\r
+        except Exception as Excpt:\r
             raise BadExpression(str(Excpt))\r
 \r
-        if Operator in ['and', 'or']:\r
+        if Operator in {'and', 'or'}:\r
             if Val:\r
                 Val = True\r
             else:\r
@@ -410,13 +411,13 @@ class ValueExpression(BaseExpression):
 \r
     # Template function to parse binary operators which have same precedence\r
     # Expr [Operator Expr]*\r
-    def _ExprFuncTemplate(self, EvalFunc, OpLst):\r
+    def _ExprFuncTemplate(self, EvalFunc, OpSet):\r
         Val = EvalFunc()\r
-        while self._IsOperator(OpLst):\r
+        while self._IsOperator(OpSet):\r
             Op = self._Token\r
             if Op == '?':\r
                 Val2 = EvalFunc()\r
-                if self._IsOperator(':'):\r
+                if self._IsOperator({':'}):\r
                     Val3 = EvalFunc()\r
                 if Val:\r
                     Val = Val2\r
@@ -425,79 +426,79 @@ class ValueExpression(BaseExpression):
                 continue\r
             try:\r
                 Val = self.Eval(Op, Val, EvalFunc())\r
-            except WrnExpression, Warn:\r
+            except WrnExpression as Warn:\r
                 self._WarnExcept = Warn\r
                 Val = Warn.result\r
         return Val\r
     # A [? B]*\r
     def _ConExpr(self):\r
-        return self._ExprFuncTemplate(self._OrExpr, ['?', ':'])\r
+        return self._ExprFuncTemplate(self._OrExpr, {'?', ':'})\r
 \r
     # A [|| B]*\r
     def _OrExpr(self):\r
-        return self._ExprFuncTemplate(self._AndExpr, ["OR", "or", "||"])\r
+        return self._ExprFuncTemplate(self._AndExpr, {"OR", "or", "||"})\r
 \r
     # A [&& B]*\r
     def _AndExpr(self):\r
-        return self._ExprFuncTemplate(self._BitOr, ["AND", "and", "&&"])\r
+        return self._ExprFuncTemplate(self._BitOr, {"AND", "and", "&&"})\r
 \r
     # A [ | B]*\r
     def _BitOr(self):\r
-        return self._ExprFuncTemplate(self._BitXor, ["|"])\r
+        return self._ExprFuncTemplate(self._BitXor, {"|"})\r
 \r
     # A [ ^ B]*\r
     def _BitXor(self):\r
-        return self._ExprFuncTemplate(self._BitAnd, ["XOR", "xor", "^"])\r
+        return self._ExprFuncTemplate(self._BitAnd, {"XOR", "xor", "^"})\r
 \r
     # A [ & B]*\r
     def _BitAnd(self):\r
-        return self._ExprFuncTemplate(self._EqExpr, ["&"])\r
+        return self._ExprFuncTemplate(self._EqExpr, {"&"})\r
 \r
     # A [ == B]*\r
     def _EqExpr(self):\r
         Val = self._RelExpr()\r
-        while self._IsOperator(["==", "!=", "EQ", "NE", "IN", "in", "!", "NOT", "not"]):\r
+        while self._IsOperator({"==", "!=", "EQ", "NE", "IN", "in", "!", "NOT", "not"}):\r
             Op = self._Token\r
-            if Op in ["!", "NOT", "not"]:\r
-                if not self._IsOperator(["IN", "in"]):\r
+            if Op in {"!", "NOT", "not"}:\r
+                if not self._IsOperator({"IN", "in"}):\r
                     raise BadExpression(ERR_REL_NOT_IN)\r
                 Op += ' ' + self._Token\r
             try:\r
                 Val = self.Eval(Op, Val, self._RelExpr())\r
-            except WrnExpression, Warn:\r
+            except WrnExpression as Warn:\r
                 self._WarnExcept = Warn\r
                 Val = Warn.result\r
         return Val\r
 \r
     # A [ > B]*\r
     def _RelExpr(self):\r
-        return self._ExprFuncTemplate(self._ShiftExpr, ["<=", ">=", "<", ">", "LE", "GE", "LT", "GT"])\r
+        return self._ExprFuncTemplate(self._ShiftExpr, {"<=", ">=", "<", ">", "LE", "GE", "LT", "GT"})\r
 \r
     def _ShiftExpr(self):\r
-        return self._ExprFuncTemplate(self._AddExpr, ["<<", ">>"])\r
+        return self._ExprFuncTemplate(self._AddExpr, {"<<", ">>"})\r
 \r
     # A [ + B]*\r
     def _AddExpr(self):\r
-        return self._ExprFuncTemplate(self._MulExpr, ["+", "-"])\r
+        return self._ExprFuncTemplate(self._MulExpr, {"+", "-"})\r
 \r
     # A [ * B]*\r
     def _MulExpr(self):\r
-        return self._ExprFuncTemplate(self._UnaryExpr, ["*", "/", "%"])\r
+        return self._ExprFuncTemplate(self._UnaryExpr, {"*", "/", "%"})\r
 \r
     # [!]*A\r
     def _UnaryExpr(self):\r
-        if self._IsOperator(["!", "NOT", "not"]):\r
+        if self._IsOperator({"!", "NOT", "not"}):\r
             Val = self._UnaryExpr()\r
             try:\r
                 return self.Eval('not', Val)\r
-            except WrnExpression, Warn:\r
+            except WrnExpression as Warn:\r
                 self._WarnExcept = Warn\r
                 return Warn.result\r
-        if self._IsOperator(["~"]):\r
+        if self._IsOperator({"~"}):\r
             Val = self._UnaryExpr()\r
             try:\r
                 return self.Eval('~', Val)\r
-            except WrnExpression, Warn:\r
+            except WrnExpression as Warn:\r
                 self._WarnExcept = Warn\r
                 return Warn.result\r
         return self._IdenExpr()\r
@@ -531,7 +532,7 @@ class ValueExpression(BaseExpression):
         if self._Token.startswith('"') or self._Token.startswith('L"'):\r
             Flag = 0\r
             for Index in range(len(self._Token)):\r
-                if self._Token[Index] in ['"']:\r
+                if self._Token[Index] in {'"'}:\r
                     if self._Token[Index - 1] == '\\':\r
                         continue\r
                     Flag += 1\r
@@ -540,7 +541,7 @@ class ValueExpression(BaseExpression):
         if self._Token.startswith("'") or self._Token.startswith("L'"):\r
             Flag = 0\r
             for Index in range(len(self._Token)):\r
-                if self._Token[Index] in ["'"]:\r
+                if self._Token[Index] in {"'"}:\r
                     if self._Token[Index - 1] == '\\':\r
                         continue\r
                     Flag += 1\r
@@ -568,7 +569,7 @@ class ValueExpression(BaseExpression):
         IsArray = IsGuid = False\r
         if len(Token.split(',')) == 11 and len(Token.split(',{')) == 2 \\r
             and len(Token.split('},')) == 1:\r
-            HexLen = [11,6,6,5,4,4,4,4,4,4,6]\r
+            HexLen = [11, 6, 6, 5, 4, 4, 4, 4, 4, 4, 6]\r
             HexList= Token.split(',')\r
             if HexList[3].startswith('{') and \\r
                 not [Index for Index, Hex in enumerate(HexList) if len(Hex) > HexLen[Index]]:\r
@@ -645,9 +646,9 @@ class ValueExpression(BaseExpression):
 \r
         if self._Token.startswith('"'):\r
             self._Token = self._Token[1:-1]\r
-        elif self._Token in ["FALSE", "false", "False"]:\r
+        elif self._Token in {"FALSE", "false", "False"}:\r
             self._Token = False\r
-        elif self._Token in ["TRUE", "true", "True"]:\r
+        elif self._Token in {"TRUE", "true", "True"}:\r
             self._Token = True\r
         else:\r
             self.__IsNumberToken()\r
@@ -764,7 +765,7 @@ class ValueExpression(BaseExpression):
     # Parse operator\r
     def _GetOperator(self):\r
         self.__SkipWS()\r
-        LegalOpLst = ['&&', '||', '!=', '==', '>=', '<='] + self.NonLetterOpLst + ['?',':']\r
+        LegalOpLst = ['&&', '||', '!=', '==', '>=', '<='] + self.NonLetterOpLst + ['?', ':']\r
 \r
         self._Token = ''\r
         Expr = self._Expr[self._Idx:]\r
@@ -816,9 +817,9 @@ class ValueExpressionEx(ValueExpression):
             elif self.PcdType in TAB_PCD_NUMERIC_TYPES and (PcdValue.startswith("'") or \\r
                       PcdValue.startswith('"') or PcdValue.startswith("L'") or PcdValue.startswith('L"') or PcdValue.startswith('{')):\r
                 raise BadExpression\r
-        except WrnExpression, Value:\r
+        except WrnExpression as Value:\r
             PcdValue = Value.result\r
-        except BadExpression, Value:\r
+        except BadExpression as Value:\r
             if self.PcdType in TAB_PCD_NUMERIC_TYPES:\r
                 PcdValue = PcdValue.strip()\r
                 if PcdValue.startswith('{') and PcdValue.endswith('}'):\r
@@ -841,7 +842,7 @@ class ValueExpressionEx(ValueExpression):
                         elif Item.startswith(TAB_UINT64):\r
                             ItemSize = 8\r
                             ValueType = TAB_UINT64\r
-                        elif Item[0] in ['"',"'",'L']:\r
+                        elif Item[0] in {'"', "'", 'L'}:\r
                             ItemSize = 0\r
                             ValueType = TAB_VOID\r
                         else:\r
@@ -854,7 +855,7 @@ class ValueExpressionEx(ValueExpression):
                                 tmpValue = int(Item, 0)\r
                                 if tmpValue > 255:\r
                                     raise BadExpression("Byte  array number %s should less than 0xFF." % Item)\r
-                            except BadExpression, Value:\r
+                            except BadExpression as Value:\r
                                 raise BadExpression(Value)\r
                             except ValueError:\r
                                 pass\r
@@ -870,7 +871,7 @@ class ValueExpressionEx(ValueExpression):
                 else:\r
                     try:\r
                         TmpValue, Size = ParseFieldValue(PcdValue)\r
-                    except BadExpression, Value:\r
+                    except BadExpression as Value:\r
                         raise BadExpression("Type: %s, Value: %s, %s" % (self.PcdType, PcdValue, Value))\r
                 if type(TmpValue) == type(''):\r
                     try:\r
@@ -945,7 +946,7 @@ class ValueExpressionEx(ValueExpression):
                             # replace each offset, except errors\r
                             for Offset in OffsetList:\r
                                 try:\r
-                                    Item = Item.replace('OFFSET_OF({})'.format(Offset),LabelDict[Offset])\r
+                                    Item = Item.replace('OFFSET_OF({})'.format(Offset), LabelDict[Offset])\r
                                 except:\r
                                     raise BadExpression('%s not defined' % Offset)\r
 \r
@@ -998,7 +999,7 @@ class ValueExpressionEx(ValueExpression):
                                 Item = '0x%x' % TmpValue if type(TmpValue) != type('') else TmpValue\r
                                 if ItemSize == 0:\r
                                     ItemValue, ItemSize = ParseFieldValue(Item)\r
-                                    if Item[0] not in ['"','L','{'] and ItemSize > 1:\r
+                                    if Item[0] not in {'"', 'L', '{'} and ItemSize > 1:\r
                                         raise BadExpression("Byte  array number %s should less than 0xFF." % Item)\r
                                 else:\r
                                     ItemValue = ParseFieldValue(Item)[0]\r
@@ -1028,10 +1029,10 @@ if __name__ == '__main__':
         if input in 'qQ':\r
             break\r
         try:\r
-            print ValueExpression(input)(True)\r
-            print ValueExpression(input)(False)\r
-        except WrnExpression, Ex:\r
-            print Ex.result\r
-            print str(Ex)\r
-        except Exception, Ex:\r
-            print str(Ex)\r
+            print(ValueExpression(input)(True))\r
+            print(ValueExpression(input)(False))\r
+        except WrnExpression as Ex:\r
+            print(Ex.result)\r
+            print(str(Ex))\r
+        except Exception as Ex:\r
+            print(str(Ex))\r