support '_' separator in dec/oct/float numbers

Allows 100_000 in place of 100000
vyperlang · Nov 8, 2023 · dc3ac07 · dc3ac07
1 parent 5482bbc
commit dc3ac07
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/vyper/ast/grammar.lark b/vyper/ast/grammar.lark
@@ -298,11 +298,11 @@ _NEWLINE: ( /\r?\n[\t ]*/ | COMMENT )+
 STRING: /b?("(?!"").*?(?<!\\)(\\\\)*?"|'(?!'').*?(?<!\\)(\\\\)*?')/i
 DOCSTRING: /(""".*?(?<!\\)(\\\\)*?"""|'''.*?(?<!\\)(\\\\)*?''')/is
 
-DEC_NUMBER: /0|[1-9]\d*/i
+DEC_NUMBER: /0|[1-9]([0-9]*_)?([0-9]+_)*[0-9]+/
 HEX_NUMBER.2: /0x[\da-f]*/i
-OCT_NUMBER.2: /0o[0-7]*/i
+OCT_NUMBER.2: /0o([0-7]+_)*[0-7]+/
 BIN_NUMBER.2 : /0b[0-1]*/i
-FLOAT_NUMBER.2: /((\d+\.\d*|\.\d+)(e[-+]?\d+)?|\d+(e[-+]?\d+))/i
+FLOAT_NUMBER.2: /((\d+(_\d+)*\.\d*|\.\d+)(e[-+]?\d+)?|\d+(_\d+)*(e[-+]?\d+)?)/i
 
 _number: DEC_NUMBER
        | HEX_NUMBER