cinderx 2026.1.16.2__cp314-cp314-manylinux_2_24_x86_64.manylinux_2_28_x86_64.whl

cinderx/compiler/strict/preprocessor.py

@@ -0,0 +1,11 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+
+# pyre-strict
+
+from __future__ import annotations
+
+from ast import AST, Name
+
+
+def is_mutable(node: AST) -> bool:
+    return isinstance(node, Name) and node.id == "mutable"

cinderx/compiler/strict/rewriter/__init__.py

@@ -0,0 +1,5 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# pyre-strict
+
+from .remove_annotations import AnnotationRemover, remove_annotations  # noqa: F401
+from .rewriter import rewrite, StrictModuleRewriter  # noqa: F401

cinderx/compiler/strict/rewriter/remove_annotations.py

@@ -0,0 +1,84 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+
+# pyre-strict
+
+from __future__ import annotations
+
+import ast
+from typing import final, TypeVar
+
+
+FunctionDefNode = TypeVar("FunctionDefNode", ast.FunctionDef, ast.AsyncFunctionDef)
+
+
+def remove_annotations(node: ast.AST) -> ast.Module:
+    return ast.fix_missing_locations(AnnotationRemover().visit(node))
+
+
+def _copy_attrs(src: ast.AST, dest: ast.AST) -> None:
+    """
+    Copies line and column info from one node to another.
+    """
+    # pyre-fixme[16]: `AST` has no attribute `lineno`.
+    dest.lineno = src.lineno
+    # pyre-fixme[16]: `AST` has no attribute `end_lineno`.
+    dest.end_lineno = src.end_lineno
+    # pyre-fixme[16]: `AST` has no attribute `col_offset`.
+    dest.col_offset = src.col_offset
+    # pyre-fixme[16]: `AST` has no attribute `end_col_offset`.
+    dest.end_col_offset = src.end_col_offset
+
+
+@final
+class AnnotationRemover(ast.NodeTransformer):
+    def visit_single_arg(self, arg: ast.arg) -> ast.arg:
+        arg.annotation = None
+        return arg
+
+    def visit_fn_arguments(self, node: ast.arguments) -> ast.arguments:
+        if node.posonlyargs:
+            node.posonlyargs = [self.visit_single_arg(a) for a in node.posonlyargs]
+        if node.args:
+            node.args = [self.visit_single_arg(a) for a in node.args]
+        if node.kwonlyargs:
+            node.kwonlyargs = [self.visit_single_arg(a) for a in node.kwonlyargs]
+        vararg = node.vararg
+        if vararg:
+            node.vararg = self.visit_single_arg(vararg)
+        kwarg = node.kwarg
+        if kwarg:
+            node.kwarg = self.visit_single_arg(kwarg)
+        return node
+
+    def visit_function(self, node: FunctionDefNode) -> FunctionDefNode:
+        node.arguments = self.visit_fn_arguments(node.args)
+        node.returns = None
+        node.decorator_list = [
+            self.visit(decorator) for decorator in node.decorator_list
+        ]
+        return node
+
+    def visit_FunctionDef(self, node: FunctionDefNode) -> FunctionDefNode:
+        return self.visit_function(node)
+
+    def visit_AsyncFunctionDef(self, node: FunctionDefNode) -> FunctionDefNode:
+        return self.visit_function(node)
+
+    def visit_AnnAssign(self, node: ast.AnnAssign) -> ast.Assign:
+        # Here, we replace `x: A = a` with just `x = a`. In case there's no value,
+        # we set the value to an `...`. E.g: `x: A` changes to `x = ...`.
+        #
+        # We need to be a little careful about ensuring that newly created nodes
+        # get the line and column information copied to them. This helps us avoid
+        # an extra pass over the AST with ast.fix_missing_locations()
+        value = node.value
+        if value is None:
+            # pyre-fixme[20]: Argument `value` expected.
+            value = ast.Constant(...)
+            value.kind = None
+            _copy_attrs(node, value)
+
+        assign = ast.Assign(targets=[node.target], value=value, type_comment=None)
+
+        _copy_attrs(node, assign)
+        return assign