openrewrite-migrate-python 0.8.0.dev20260602181751__tar.gz → 0.9.0__tar.gz

{openrewrite_migrate_python-0.8.0.dev20260602181751 → openrewrite_migrate_python-0.9.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openrewrite-migrate-python
-Version: 0.8.0.dev20260602181751
+Version: 0.9.0
 Summary: OpenRewrite recipes for migrating Python codebases to newer Python versions.
 Author-email: "Moderne Inc." <support@moderne.io>
 License: Moderne Proprietary
@@ -21,7 +21,8 @@ Classifier: Typing :: Typed
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 Provides-Extra: dev
-Requires-Dist: openrewrite>=8.74.3; extra == "dev"
+Requires-Dist: openrewrite>=8.84.3; extra == "dev"
+Requires-Dist: pydantic>=2.11; extra == "dev"
 Requires-Dist: pytest>=8.0.0; extra == "dev"
 Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
 Requires-Dist: black>=24.0.0; extra == "dev"

{openrewrite_migrate_python-0.8.0.dev20260602181751 → openrewrite_migrate_python-0.9.0}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 [project]
 name = "openrewrite-migrate-python"
 description = "OpenRewrite recipes for migrating Python codebases to newer Python versions."
-version = "0.8.0.dev20260602181751"  # Updated dynamically during release
+version = "0.9.0"  # Updated dynamically during release
 authors = [{ name = "Moderne Inc.", email = "support@moderne.io" }]
 license = { text = "Moderne Proprietary" }
 readme = "README.md"
@@ -30,7 +30,14 @@ dependencies = []
 
 [project.optional-dependencies]
 dev = [
-    "openrewrite>=8.74.3",
+    # 8.84.3 brings ty external-type resolution (cross-package supertype chains),
+    # J.MethodInvocation.type, sorted insertion in maybe_add_import, and the
+    # type_utils comparison helpers — all of which the recipes rely on for
+    # type-aware matching. openrewrite is host-provided at runtime, so this floor
+    # only affects dev/CI.
+    "openrewrite>=8.84.3",
+    # Needed so type-attribution tests can resolve `pydantic` types.
+    "pydantic>=2.11",
     "pytest>=8.0.0",
     "pytest-cov>=4.0.0",
     "black>=24.0.0",

{openrewrite_migrate_python-0.8.0.dev20260602181751 → openrewrite_migrate_python-0.9.0}/src/openrewrite_migrate_python/migrate/__init__.py

@@ -63,6 +63,17 @@ from .pep594_system_migrations import (
 )
 from .pipes_migrations import FindPipesModule
 from .pkgutil_deprecations import ReplacePkgutilFindLoader, ReplacePkgutilGetLoader
+from .pydantic_deprecations import (
+    FindDeprecatedJsonEncoders,
+    FindDeprecatedSchemaGenerator,
+    FindEvalTypeBackportUsage,
+    FindModelValidatorAfterClassmethod,
+    FindSerializeAsAnyUsage,
+    ReplaceFinalFieldWithClassVar,
+    ReplacePopulateByNameWithValidateByName,
+)
+from .pydantic_fields import RemoveEllipsisFromField
+from .pydantic_model_fields import ReplaceModelFieldsInstanceAccess
 from .re_deprecations import ReplaceReTemplate
 from .removed_modules_312 import FindRemovedModules312
 from .shutil_deprecations import FindShutilRmtreeOnerror
@@ -105,6 +116,13 @@ from .upgrade_to_python311 import UpgradeToPython311
 from .upgrade_to_python312 import UpgradeToPython312
 from .upgrade_to_python313 import UpgradeToPython313
 from .upgrade_to_python314 import UpgradeToPython314
+from .upgrade_to_pydantic import (
+    UpgradeToPydantic210,
+    UpgradeToPydantic211,
+    UpgradeToPydantic212,
+    UpgradeToPydantic213,
+    UpgradeToPydantic214,
+)
 from .urllib_deprecations import FindUrllibParseSplitFunctions, FindUrllibParseToBytes
 from .uu_migrations import FindUuModule
 from .xdrlib_migrations import FindXdrlibModule
@@ -230,4 +248,21 @@ __all__ = [
     "UpgradeToPython312",
     "UpgradeToPython313",
     "UpgradeToPython314",
+    # Auto-fix: Pydantic
+    "ReplaceModelFieldsInstanceAccess",
+    "ReplaceFinalFieldWithClassVar",
+    "ReplacePopulateByNameWithValidateByName",
+    "RemoveEllipsisFromField",
+    # Detection: Pydantic
+    "FindDeprecatedSchemaGenerator",
+    "FindDeprecatedJsonEncoders",
+    "FindSerializeAsAnyUsage",
+    "FindModelValidatorAfterClassmethod",
+    "FindEvalTypeBackportUsage",
+    # Composite upgrade recipes: Pydantic
+    "UpgradeToPydantic210",
+    "UpgradeToPydantic211",
+    "UpgradeToPydantic212",
+    "UpgradeToPydantic213",
+    "UpgradeToPydantic214",
 ]

openrewrite_migrate_python-0.9.0/src/openrewrite_migrate_python/migrate/_markers.py

@@ -0,0 +1,22 @@
+"""Shared marker utilities for deprecation recipes."""
+
+from typing import Any, Optional
+
+from rewrite.markers import Markers, MarkupWarn
+from rewrite.utils import random_id
+
+
+def mark_deprecated(tree: Any, message: str, detail: Optional[str] = None) -> Any:
+    """Attach a ``MarkupWarn`` warning marker flagging a deprecation.
+
+    ``MarkupWarn`` is OpenRewrite's warning-markup marker (the intended marker
+    for deprecations that need manual attention), as opposed to ``SearchResult``
+    which denotes a plain search hit. It renders the same comment in printed
+    output but surfaces as a warning diagnostic in the tooling.
+    """
+    full_message = f"{message}: {detail}" if detail else message
+    warn_marker = MarkupWarn(random_id(), full_message)
+    current_markers = tree.markers
+    new_markers_list = list(current_markers.markers) + [warn_marker]
+    new_markers = Markers(current_markers.id, new_markers_list)
+    return tree.replace(_markers=new_markers)

{openrewrite_migrate_python-0.8.0.dev20260602181751 → openrewrite_migrate_python-0.9.0}/src/openrewrite_migrate_python/migrate/array_deprecations.py

@@ -18,6 +18,8 @@ from rewrite.python.preconditions import uses_method
 from rewrite.python.visitor import PythonVisitor
 from rewrite.java.tree import Identifier, MethodInvocation
 
+from rewrite.python.type_utils import is_assignable_to
+
 # Define category path: Python > Migrate > Python 3.14
 _Python314 = [
     *Python,
@@ -72,6 +74,8 @@ class ReplaceArrayTostring(Recipe):
                     return method
                 if method.name.simple_name != "tostring":
                     return method
+                if method.select is None or not is_assignable_to("array.array", method.select.type):
+                    return method
 
                 # Replace tostring with tobytes
                 new_name = method.name.replace(_simple_name="tobytes")
@@ -126,6 +130,8 @@ class ReplaceArrayFromstring(Recipe):
                     return method
                 if method.name.simple_name != "fromstring":
                     return method
+                if method.select is None or not is_assignable_to("array.array", method.select.type):
+                    return method
 
                 # Replace fromstring with frombytes
                 new_name = method.name.replace(_simple_name="frombytes")

{openrewrite_migrate_python-0.8.0.dev20260602181751 → openrewrite_migrate_python-0.9.0}/src/openrewrite_migrate_python/migrate/ast_deprecations.py

@@ -23,6 +23,8 @@ from rewrite.marketplace import Python
 from rewrite.python.visitor import PythonVisitor
 from rewrite.java.tree import Identifier, FieldAccess
 
+from rewrite.python.type_utils import is_of_class_type
+
 # Define category path: Python > Migrate > Python 3.8
 _Python38 = [
     *Python,
@@ -44,12 +46,9 @@ def _is_ast_attribute(field_access: FieldAccess, attr_name: str) -> bool:
     if field_access.name.simple_name != attr_name:
         return False
 
-    # Check if the target is 'ast'
-    target = field_access.target
-    if isinstance(target, Identifier) and target.simple_name == "ast":
-        return True
-
-    return False
+    # Confirm the target resolves to the `ast` module (by type, so aliased
+    # imports work and a same-named variable is not rewritten).
+    return is_of_class_type(field_access.target.type, "ast")
 
 
 def _replace_ast_attribute(field_access: FieldAccess, new_name: str) -> FieldAccess:

{openrewrite_migrate_python-0.8.0.dev20260602181751 → openrewrite_migrate_python-0.9.0}/src/openrewrite_migrate_python/migrate/calendar_deprecations.py

@@ -20,6 +20,8 @@ from rewrite.marketplace import Python
 from rewrite.python.visitor import PythonVisitor
 from rewrite.java.tree import Identifier, FieldAccess
 
+from rewrite.python.type_utils import is_of_class_type
+
 # Define category path: Python > Migrate > Python 3.12
 _Python312 = [
     *Python,
@@ -53,16 +55,6 @@ CALENDAR_CONSTANTS: Dict[str, str] = {
 }
 
 
-def _is_calendar_access(identifier: Identifier) -> bool:
-    """
-    Check if this identifier looks like a calendar module constant access.
-
-    Without full type attribution, we check for common patterns.
-    """
-    # The identifier itself should be a known calendar constant name
-    return identifier.simple_name in CALENDAR_CONSTANTS
-
-
 @categorize(_Python312)
 class ReplaceCalendarConstants(Recipe):
     """
@@ -118,9 +110,10 @@ class ReplaceCalendarConstants(Recipe):
                 if const_name not in CALENDAR_CONSTANTS:
                     return field_access
 
-                # Check if the target is the calendar module
+                # Confirm the target resolves to the `calendar` module (by type,
+                # so aliases work and same-named variables are excluded).
                 target = field_access.target
-                if isinstance(target, Identifier) and target.simple_name == "calendar":
+                if is_of_class_type(target.type, "calendar"):
                     # Replace with uppercase version
                     new_name_str = CALENDAR_CONSTANTS[const_name]
                     new_identifier = name.replace(_simple_name=new_name_str)

{openrewrite_migrate_python-0.8.0.dev20260602181751 → openrewrite_migrate_python-0.9.0}/src/openrewrite_migrate_python/migrate/platform_deprecations.py

@@ -16,6 +16,8 @@ from rewrite.marketplace import Python
 from rewrite.markers import Markers
 from rewrite.python.preconditions import uses_method
 from rewrite.python.visitor import PythonVisitor
+
+from rewrite.python.type_utils import is_of_class_type
 from rewrite.java.support_types import JLeftPadded, JRightPadded, Space
 from rewrite.java.tree import Identifier, Literal, MethodInvocation
 from rewrite.python.tree import NamedArgument
@@ -81,7 +83,10 @@ class ReplacePlatformPopen(Recipe):
                 select = method.select
                 if not isinstance(select, Identifier):
                     return method
-                if select.simple_name != "platform":
+                # Confirm the receiver resolves to the `platform` module (by
+                # type, so an aliased import works and a same-named variable is
+                # not rewritten into a destructive `subprocess.check_output`).
+                if not is_of_class_type(select.type, "platform"):
                     return method
 
                 # Replace platform with subprocess