lazify, autocurry: leave type statements alone

Technologicat · Technologicat · commit 58388fc0f3d7 · 2024-09-27T14:31:30.000+03:00
diff --git a/unpythonic/syntax/autocurry.py b/unpythonic/syntax/autocurry.py
@@ -10,6 +10,7 @@
 from mcpyrate.quotes import is_captured_value
 from mcpyrate.walkers import ASTTransformer
 
+from .astcompat import TypeAlias
 from .util import (suggest_decorator_index, isx, has_curry, sort_lambda_decorators)
 
 from ..dynassign import dyn
@@ -85,6 +86,10 @@ def transform(self, tree):
             if is_captured_value(tree):
                 return tree
 
+            # Python 3.12+: leave `type` statements alone (autocurrying a type declaration makes no sense)
+            if type(tree) is TypeAlias:
+                return tree
+
             hascurry = self.state.hascurry
             if type(tree) is Call:
                 # Don't auto-curry some calls we know not to need it. This is both a performance optimization
diff --git a/unpythonic/syntax/lazify.py b/unpythonic/syntax/lazify.py
@@ -14,6 +14,7 @@
 from mcpyrate.unparser import unparse
 from mcpyrate.walkers import ASTTransformer
 
+from .astcompat import TypeAlias
 from .util import (suggest_decorator_index, sort_lambda_decorators, detect_lambda,
                    isx, getname, is_decorator)
 from .letdoutil import islet, isdo, ExpandedLetView
@@ -648,6 +649,10 @@ def f(tree):
                 # else forcing_mode == "off"
                 return tree
 
+            # Python 3.12+: leave `type` statements alone (lazifying a type declaration makes no sense)
+            elif type(tree) is TypeAlias:
+                return tree
+
             elif type(tree) in (FunctionDef, AsyncFunctionDef, Lambda):
                 if type(tree) is Lambda and id(tree) not in userlambdas:
                     return self.generic_visit(tree)  # ignore macro-introduced lambdas (but recurse inside them)
