feat: new primitive for weakref.__call__

python · BobTheBuidler · May 23, 2025 · May 23, 2025 · May 23, 2025 · May 23, 2025
commit 3b71a7f71d1ede3ff858285c68d9cce43fa02ed0
diff --git a/mypyc/primitives/weakref_ops.py b/mypyc/primitives/weakref_ops.py
@@ -1,6 +1,6 @@
 from mypyc.ir.ops import ERR_MAGIC
 from mypyc.ir.rtypes import object_rprimitive
-from mypyc.primitives.registry import function_op
+from mypyc.primitives.registry import ERR_NEG_INT, function_op

 # Weakref operations

@@ -11,3 +11,11 @@
    c_function_name="PyWeakref_NewRef",
    error_kind=ERR_MAGIC,
 )
+
+deref_op = function_op(
+    name="weakref.ReferenceType.__call__",
+    arg_types=[object_rprimitive],
+    return_type=object_rprimitive,
+    c_function_name="PyWeakref_GetRef",
+    error_kind=ERR_NEG_INT,
+)
diff --git a/mypyc/test-data/irbuild-weakref.test b/mypyc/test-data/irbuild-weakref.test
@@ -2,24 +2,26 @@
 import weakref
 from typing import Any, Callable
 def f(x: object, cb: Callable[[object], Any]) -> object:
-    return weakref.ref(x, cb)
+    return weakref.ref(x, cb)()

 [out]
 def f(x, cb):
-    x, cb, r0 :: object
+    x, cb, r0, r1 :: object
 L0:
    r0 = PyWeakref_NewRef(x, cb)
-    return r0
+    r1 = PyWeakref_GetRef(r0)
+    return r1

 [case testFromWeakrefRefCallback]
 from typing import Any, Callable
 from weakref import ref
 def f(x: object, cb: Callable[[object], Any]) -> object:
-    return ref(x, cb)
+    return ref(x, cb)()

 [out]
 def f(x, cb):
-    x, cb, r0 :: object
+    x, cb, r0, r1 :: object
 L0:
    r0 = PyWeakref_NewRef(x, cb)
-    return r0
+    r1 = PyWeakref_GetRef(r0)
+    return r1