Update specialize.py

Author: BobTheBuidler

mypyc/irbuild/specialize.py

@@ -96,6 +96,7 @@
     str_encode_utf8_strict,
 )
 from mypyc.primitives.tuple_ops import new_tuple_set_item_op
+from mypyc.primitives.weakref_ops import new_proxy_op
 
 # Specializers are attempted before compiling the arguments to the
 # function.  Specializers can return None to indicate that they failed
@@ -888,3 +889,31 @@ def translate_ord(builder: IRBuilder, expr: CallExpr, callee: RefExpr) -> Value
     if isinstance(arg, (StrExpr, BytesExpr)) and len(arg.value) == 1:
         return Integer(ord(arg.value))
     return None
+
+
+@specialize_function("weakref.proxy")
+def translate_proxy(builder: IRBuilder, expr: CallExpr, callee: RefExpr) -> Value | None:
+    """
+    Special case for weakref.proxy(...).
+    
+    If there's only one argument (the object), pass None for the callback.
+    If there are two arguments, use them as (object, callback).
+    Otherwise, fallback to a normal Python call by returning None.
+    """
+    line = expr.line
+    num_args = len(expr.args)
+    
+    # If the user supplied something other than 1 or 2 args, just fail
+    # and let MyPyC do a generic Python call.
+    if num_args not in (1, 2):
+        # NOTE: should we raise a compiler error here?
+        return None
+    
+    obj_val = builder.accept(expr.args[0])
+    if num_args == 1:
+        callback_val = builder.none_object()
+    else:
+        callback_val = builder.accept(expr.args[1])
+    
+    # Emit a direct C call to PyWeakref_NewProxy
+    return builder.call_c(new_proxy_op, [obj_val, callback_val], line)