Add compatibility with python 3.13.1 (#2647) (#2649)

jacobtylerwalls · web-flow · commit 0834156734cb · 2024-12-08T07:23:56.000-05:00
Refs python/cpython#125415 (cherry picked from commit fe01bda)
diff --git a/ChangeLog b/ChangeLog
@@ -13,12 +13,15 @@ What's New in astroid 3.3.6?
 ============================
 Release date: TBA
 
+* Fix inability to import `collections.abc` in python 3.13.1.
+
+  Closes pylint-dev/pylint#10112
+
 * Fix crash when typing._alias() call is missing arguments.
 
   Closes #2513
 
 
-
 What's New in astroid 3.3.5?
 ============================
 Release date: 2024-10-04
diff --git a/astroid/brain/brain_collections.py b/astroid/brain/brain_collections.py
@@ -8,7 +8,7 @@
 
 from astroid.brain.helpers import register_module_extender
 from astroid.builder import AstroidBuilder, extract_node, parse
-from astroid.const import PY313_PLUS
+from astroid.const import PY313_0, PY313_PLUS
 from astroid.context import InferenceContext
 from astroid.exceptions import AttributeInferenceError
 from astroid.manager import AstroidManager
@@ -20,7 +20,8 @@
 
 def _collections_transform():
     return parse(
-        """
+        ("    import _collections_abc as abc" if PY313_PLUS and not PY313_0 else "")
+        + """
     class defaultdict(dict):
         default_factory = None
         def __missing__(self, key): pass
@@ -32,7 +33,7 @@ def __getitem__(self, key): return default_factory
     )
 
 
-def _collections_abc_313_transform() -> nodes.Module:
+def _collections_abc_313_0_transform() -> nodes.Module:
     """See https://github.com/python/cpython/pull/124735"""
     return AstroidBuilder(AstroidManager()).string_build(
         "from _collections_abc import *"
@@ -132,7 +133,7 @@ def register(manager: AstroidManager) -> None:
         ClassDef, easy_class_getitem_inference, _looks_like_subscriptable
     )
 
-    if PY313_PLUS:
+    if PY313_0:
         register_module_extender(
-            manager, "collections.abc", _collections_abc_313_transform
+            manager, "collections.abc", _collections_abc_313_0_transform
         )
diff --git a/astroid/const.py b/astroid/const.py
@@ -9,6 +9,7 @@
 PY311_PLUS = sys.version_info >= (3, 11)
 PY312_PLUS = sys.version_info >= (3, 12)
 PY313_PLUS = sys.version_info >= (3, 13)
+PY313_0 = sys.version_info[:3] == (3, 13, 0)
 
 WIN32 = sys.platform == "win32"
 
