i made some changes apparently

thatbirdguythatuknownot · Sep 22, 2023 · b9b6db9 · b9b6db9
1 parent a656827
commit b9b6db9
Show file tree

Hide file tree

Showing 13 changed files with 2,789 additions and 2,459 deletions.
diff --git a/Doc/library/token-list.inc b/Doc/library/token-list.inc
diff --git a/Grammar/Tokens b/Grammar/Tokens
@@ -56,6 +56,7 @@ COLONEQUAL              ':='
 EXCLAMATION             '!'
 PIPEGT                  '|>'
 DOLLAR                  '$'
+PIPEGTEQUAL             '|>='
 
 OP
 TYPE_IGNORE

diff --git a/Grammar/python.gram b/Grammar/python.gram
@@ -155,6 +155,8 @@ assignment[stmt_ty]:
         CHECK_VERSION(stmt_ty, 6, "Variable annotations syntax is", _PyAST_AnnAssign(a, b, c, 0, EXTRA)) }
     | a[asdl_expr_seq*]=(z=star_targets '=' { z })+ b=(yield_expr | star_expressions) !'=' tc=[TYPE_COMMENT] {
          _PyAST_Assign(a, b, NEW_TYPE_COMMENT(p, tc), EXTRA) }
+    | a=single_target t=(t='|>=' { (++p->subn, t) }) ~ c=(d=[yield_expr | star_expressions] { (--p->subn, d) }) {
+         _PyAST_AugAssign(a, Comp, c, EXTRA) }
     | a=single_target b=augassign ~ c=(yield_expr | star_expressions) {
          _PyAST_AugAssign(a, b->kind, c, EXTRA) }
     | invalid_assignment
@@ -175,6 +177,7 @@ augassign[AugOperator*]:
     | '>>=' { _PyPegen_augoperator(p, RShift) }
     | '**=' { _PyPegen_augoperator(p, Pow) }
     | '//=' { _PyPegen_augoperator(p, FloorDiv) }
+    | '|>=' { _PyPegen_augoperator(p, Comp) }
 
 return_stmt[stmt_ty]:
     | 'return' a=[star_expressions] { _PyAST_Return(a, EXTRA) }

diff --git a/Include/internal/pycore_ast.h b/Include/internal/pycore_ast.h
diff --git a/Include/internal/pycore_ast_state.h b/Include/internal/pycore_ast_state.h
diff --git a/Include/internal/pycore_token.h b/Include/internal/pycore_token.h
@@ -70,17 +70,18 @@ extern "C" {
 #define EXCLAMATION     54
 #define PIPEGT          55
 #define DOLLAR          56
-#define OP              57
-#define TYPE_IGNORE     58
-#define TYPE_COMMENT    59
-#define SOFT_KEYWORD    60
-#define FSTRING_START   61
-#define FSTRING_MIDDLE  62
-#define FSTRING_END     63
-#define COMMENT         64
-#define NL              65
-#define ERRORTOKEN      66
-#define N_TOKENS        68
+#define PIPEGTEQUAL     57
+#define OP              58
+#define TYPE_IGNORE     59
+#define TYPE_COMMENT    60
+#define SOFT_KEYWORD    61
+#define FSTRING_START   62
+#define FSTRING_MIDDLE  63
+#define FSTRING_END     64
+#define COMMENT         65
+#define NL              66
+#define ERRORTOKEN      67
+#define N_TOKENS        69
 #define NT_OFFSET       256
 
 /* Special definitions for cooperation with parser */

diff --git a/Lib/token.py b/Lib/token.py
diff --git a/Parser/Python.asdl b/Parser/Python.asdl
@@ -105,7 +105,8 @@ module Python
     boolop = And | Or
 
     operator = Add | Sub | Mult | MatMult | Div | Mod | Pow | LShift
-                 | RShift | BitOr | BitXor | BitAnd | FloorDiv
+                 | RShift | BitOr | BitXor | BitAnd | FloorDiv | Comp
+                 | CompCall
 
     unaryop = Invert | Not | UAdd | USub