@willbooster/tree-sitter-python 2.1.2 → 2.3.0

package/grammar.js

@@ -43,6 +43,12 @@ module.exports = grammar({
   extras: ($) => [$.comment, /[\s\f\uFEFF\u2060\u200B]|\r?\n/, $.line_continuation],
 
   conflicts: ($) => [
+    [$.default_type_parameter, $.primary_expression],
+    [$.primary_expression, $._default_type_primary],
+    [$.primary_expression, $._default_generic_type],
+    [$.primary_expression, $._default_member_type],
+    [$.list, $._default_type_list],
+    [$._type_parameter_expression, $.expression],
     [$.primary_expression, $.pattern],
     [$.primary_expression, $.list_splat_pattern],
     [$.tuple, $.tuple_pattern],
@@ -94,6 +100,8 @@ module.exports = grammar({
     $._expressions,
     $._left_hand_side,
     $.keyword_identifier,
+    $._type_parameter_name,
+    $._default_member_base,
   ],
 
   reserved: {
@@ -152,6 +160,7 @@ module.exports = grammar({
       choice(
         $.future_import_statement,
         $.import_statement,
+        $.lazy_import_statement,
         $.import_from_statement,
         $.print_statement,
         $.assert_statement,
@@ -170,6 +179,8 @@ module.exports = grammar({
 
     import_statement: ($) => seq('import', $._import_list),
 
+    lazy_import_statement: ($) => seq('lazy', choice($.import_statement, $.import_from_statement)),
+
     import_prefix: () => repeat1('.'),
 
     relative_import: ($) => seq($.import_prefix, optional($.dotted_name)),
@@ -390,7 +401,100 @@ module.exports = grammar({
         ':',
         field('body', $._suite)
       ),
-    type_parameter: ($) => seq('[', commaSep1(choice($.type, $.splat_type)), optional(','), ']'),
+    type_parameter: ($) =>
+      seq('[', commaSep1(choice($.type, $.splat_type, $.default_type_parameter)), optional(','), ']'),
+
+    default_type_parameter: ($) =>
+      choice(
+        seq(
+          choice(
+            seq(
+              field('name', $._type_parameter_name),
+              optional(seq(':', field('bound', alias($._type_parameter_expression, $.type))))
+            ),
+            seq('**', field('name', $._type_parameter_name))
+          ),
+          '=',
+          field('value', alias($._type_parameter_expression, $.type))
+        ),
+        seq(
+          '*',
+          field('name', $._type_parameter_name),
+          '=',
+          field(
+            'value',
+            choice(alias($._type_parameter_expression, $.type), alias($._default_splat_type, $.list_splat))
+          )
+        )
+      ),
+
+    _type_parameter_name: ($) =>
+      choice($.identifier, alias(choice('type', 'match', 'print', 'exec', 'lazy'), $.identifier)),
+
+    _default_member_base: ($) => choice($._type_parameter_name, alias($._default_member_type, $.member_type)),
+
+    _type_parameter_expression: ($) => choice($._expression_without_named_expression, prec.dynamic(1, $._default_type)),
+
+    _default_type: ($) => choice($._default_type_primary, alias($._default_union_type, $.union_type)),
+
+    _default_type_primary: ($) =>
+      choice(
+        $.identifier,
+        $.keyword_identifier,
+        $.none,
+        $.string,
+        alias($._default_member_type, $.member_type),
+        alias($._default_generic_type, $.generic_type),
+        alias($._default_type_list, $.list),
+        alias($._default_type_tuple, $.tuple),
+        alias($._default_parenthesized_type, $.parenthesized_expression)
+      ),
+
+    _default_member_type: ($) => seq($._default_member_base, '.', $.identifier),
+
+    _default_generic_type: ($) => seq($._default_member_base, alias($._default_type_arguments, $.type_parameter)),
+
+    _default_union_type: ($) =>
+      prec.left(PREC.bitwise_or, seq(alias($._default_type, $.type), '|', alias($._default_type_primary, $.type))),
+
+    _default_parenthesized_type: ($) => seq('(', alias($._default_type, $.type), ')'),
+
+    _default_type_arguments: ($) =>
+      seq(
+        '[',
+        commaSep1(
+          choice(
+            alias($._type_parameter_expression, $.type),
+            $.named_expression,
+            alias($._default_splat_argument, $.splat_type),
+            $.slice
+          )
+        ),
+        optional(','),
+        ']'
+      ),
+
+    _default_splat_argument: ($) => seq('*', alias($._type_parameter_expression, $.type)),
+
+    _default_type_list: ($) =>
+      prec.dynamic(
+        1,
+        seq('[', optional(seq(commaSep1(alias($._type_parameter_expression, $.type)), optional(','))), ']')
+      ),
+
+    _default_type_tuple: ($) =>
+      prec.dynamic(
+        1,
+        seq(
+          '(',
+          alias($._type_parameter_expression, $.type),
+          ',',
+          optional(seq(commaSep1(alias($._type_parameter_expression, $.type)), optional(','))),
+          ')'
+        )
+      ),
+
+    _default_splat_type: ($) => seq('*', choice($.primary_expression, prec.dynamic(1, alias($._default_type, $.type)))),
 
     _generic_type_arguments: ($) =>
       seq('[', commaSep1(choice($.type, alias($._list_splat_argument, $.splat_type))), optional(','), ']'),
@@ -913,7 +1017,7 @@ module.exports = grammar({
     keyword_identifier: ($) =>
       choice(
         prec(-3, alias(choice('print', 'exec', 'async', 'await'), $.identifier)),
-        alias(choice('type', 'match'), $.identifier)
+        alias(choice('type', 'match', 'lazy'), $.identifier)
       ),
 
     true: () => 'True',

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@willbooster/tree-sitter-python",
-  "version": "2.1.2",
+  "version": "2.3.0",
   "description": "Python grammar for tree-sitter",
   "keywords": [
     "incremental",

package/queries/highlights.scm

@@ -63,6 +63,9 @@
 
 (attribute attribute: (identifier) @property)
 (type (identifier) @type)
+(member_type (identifier) @type)
+(generic_type (identifier) @type)
+(default_type_parameter name: (identifier) @type)
 
 ; Literals
 
@@ -164,3 +167,5 @@
   "match"
   "case"
 ] @keyword
+
+(lazy_import_statement "lazy" @keyword)

package/src/node-types.json

@@ -720,6 +720,10 @@
           "type": "import_statement",
           "named": true
         },
+        {
+          "type": "lazy_import_statement",
+          "named": true
+        },
         {
           "type": "match_statement",
           "named": true
@@ -1303,6 +1307,46 @@
       }
     }
   },
+  {
+    "type": "default_type_parameter",
+    "named": true,
+    "fields": {
+      "bound": {
+        "multiple": false,
+        "required": false,
+        "types": [
+          {
+            "type": "type",
+            "named": true
+          }
+        ]
+      },
+      "name": {
+        "multiple": false,
+        "required": true,
+        "types": [
+          {
+            "type": "identifier",
+            "named": true
+          }
+        ]
+      },
+      "value": {
+        "multiple": false,
+        "required": true,
+        "types": [
+          {
+            "type": "list_splat",
+            "named": true
+          },
+          {
+            "type": "type",
+            "named": true
+          }
+        ]
+      }
+    }
+  },
   {
     "type": "delete_statement",
     "named": true,
@@ -1990,6 +2034,10 @@
           "type": "identifier",
           "named": true
         },
+        {
+          "type": "member_type",
+          "named": true
+        },
         {
           "type": "type_parameter",
           "named": true
@@ -2332,6 +2380,25 @@
       ]
     }
   },
+  {
+    "type": "lazy_import_statement",
+    "named": true,
+    "fields": {},
+    "children": {
+      "multiple": false,
+      "required": true,
+      "types": [
+        {
+          "type": "import_from_statement",
+          "named": true
+        },
+        {
+          "type": "import_statement",
+          "named": true
+        }
+      ]
+    }
+  },
   {
     "type": "list",
     "named": true,
@@ -2347,6 +2414,10 @@
         {
           "type": "list_splat",
           "named": true
+        },
+        {
+          "type": "type",
+          "named": true
         }
       ]
     }
@@ -2435,6 +2506,10 @@
         {
           "type": "primary_expression",
           "named": true
+        },
+        {
+          "type": "type",
+          "named": true
         }
       ]
     }
@@ -2504,6 +2579,10 @@
           "type": "identifier",
           "named": true
         },
+        {
+          "type": "member_type",
+          "named": true
+        },
         {
           "type": "type",
           "named": true
@@ -2580,6 +2659,10 @@
           "type": "import_statement",
           "named": true
         },
+        {
+          "type": "lazy_import_statement",
+          "named": true
+        },
         {
           "type": "match_statement",
           "named": true
@@ -2758,6 +2841,10 @@
           "type": "expression",
           "named": true
         },
+        {
+          "type": "type",
+          "named": true
+        },
         {
           "type": "yield",
           "named": true
@@ -3002,6 +3089,10 @@
         {
           "type": "primary_expression",
           "named": true
+        },
+        {
+          "type": "type",
+          "named": true
         }
       ]
     }
@@ -3135,6 +3226,10 @@
         {
           "type": "list_splat",
           "named": true
+        },
+        {
+          "type": "type",
+          "named": true
         }
       ]
     }
@@ -3178,10 +3273,6 @@
           "type": "generic_type",
           "named": true
         },
-        {
-          "type": "identifier",
-          "named": true
-        },
         {
           "type": "member_type",
           "named": true
@@ -3227,6 +3318,18 @@
       "multiple": true,
       "required": true,
       "types": [
+        {
+          "type": "default_type_parameter",
+          "named": true
+        },
+        {
+          "type": "named_expression",
+          "named": true
+        },
+        {
+          "type": "slice",
+          "named": true
+        },
         {
           "type": "splat_type",
           "named": true
@@ -3858,6 +3961,10 @@
     "type": "lambda",
     "named": false
   },
+  {
+    "type": "lazy",
+    "named": false
+  },
   {
     "type": "line_continuation",
     "named": true,

package/tree-sitter.json

@@ -19,7 +19,7 @@
     }
   ],
   "metadata": {
-    "version": "2.1.2",
+    "version": "2.3.0",
     "license": "MIT",
     "description": "Python grammar for tree-sitter",
     "authors": [