wrapt 2.0.0rc5__cp312-cp312-win_arm64.whl → 2.0.0rc6__cp312-cp312-win_arm64.whl

wrapt/__init__.py

@@ -2,7 +2,7 @@
 Wrapt is a library for decorators, wrappers and monkey patching.
 """
 
-__version_info__ = ("2", "0", "0", "rc5")
+__version_info__ = ("2", "0", "0", "rc6")
 __version__ = ".".join(__version_info__)
 
 from .__wrapt__ import (

wrapt/_wrappers.c

@@ -1658,16 +1658,51 @@ static PyObject *WraptObjectProxy_inplace_matrix_multiply(
   if (PyObject_IsInstance(other, (PyObject *)&WraptObjectProxy_Type))
     other = ((WraptObjectProxyObject *)other)->wrapped;
 
-  object = PyNumber_InPlaceMatrixMultiply(self->wrapped, other);
+  if (PyObject_HasAttrString(self->wrapped, "__imatmul__"))
+  {
+    object = PyNumber_InPlaceMatrixMultiply(self->wrapped, other);
 
-  if (!object)
-    return NULL;
+    if (!object)
+      return NULL;
 
-  Py_DECREF(self->wrapped);
-  self->wrapped = object;
+    Py_DECREF(self->wrapped);
+    self->wrapped = object;
 
-  Py_INCREF(self);
-  return (PyObject *)self;
+    Py_INCREF(self);
+    return (PyObject *)self;
+  }
+  else
+  {
+    PyObject *result = PyNumber_MatrixMultiply(self->wrapped, other);
+
+    if (!result)
+      return NULL;
+
+    PyObject *proxy_type = PyObject_GetAttrString((PyObject *)self, "__object_proxy__");
+
+    if (!proxy_type)
+    {
+      Py_DECREF(proxy_type);
+      return NULL;
+    }
+
+    PyObject *proxy_args = PyTuple_Pack(1, result);
+
+    Py_DECREF(result);
+
+    if (!proxy_args)
+    {
+      Py_DECREF(proxy_type);
+      return NULL;
+    }
+
+    PyObject *proxy_instance = PyObject_Call(proxy_type, proxy_args, NULL);
+
+    Py_DECREF(proxy_type);
+    Py_DECREF(proxy_args);
+
+    return proxy_instance;
+  }
 }
 
 /* ------------------------------------------------------------------------- */

wrapt/wrappers.py

@@ -490,8 +490,11 @@ class ObjectProxy(with_metaclass(_ObjectProxyMetaType)):  # type: ignore[misc]
         return other @ self.__wrapped__
 
     def __imatmul__(self, other):
-        self.__wrapped__ @= other
-        return self
+        if hasattr(self.__wrapped__, "__imatmul__"):
+            self.__wrapped__ @= other
+            return self
+        else:
+            return self.__object_proxy__(self.__wrapped__ @ other)
 
     def __len__(self):
         return len(self.__wrapped__)

{wrapt-2.0.0rc5.dist-info → wrapt-2.0.0rc6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wrapt
-Version: 2.0.0rc5
+Version: 2.0.0rc6
 Summary: Module for decorators, wrappers and monkey patching.
 Home-page: https://github.com/GrahamDumpleton/wrapt
 Author: Graham Dumpleton

{wrapt-2.0.0rc5.dist-info → wrapt-2.0.0rc6.dist-info}/RECORD

@@ -1,8 +1,8 @@
-wrapt/__init__.py,sha256=Q-0pC9jo79xnZvORHx9LgMXEf985YbCk-K24uUkJHls,1591
+wrapt/__init__.py,sha256=OvuYIm6ApyLYhCfKLMMx8lkysIDX4W2SQkxhuq7pPFo,1591
 wrapt/__init__.pyi,sha256=6ZgAOHLYbSsE0hfcayTFUIH0sZkR2Mo13CbeZRstcHc,9631
 wrapt/__wrapt__.py,sha256=hF01bdl0BhpXsSAmBIv3Zux3VNwsGXhoWQhj6kcDt6I,1475
-wrapt/_wrappers.c,sha256=8y2AFO3p-ilIOZtMVQMtcNpyaxT2V92agLu7g0I9gk8,119167
-wrapt/_wrappers.cp312-win_arm64.pyd,sha256=W83uig0E3VnQLvmZ5oBtoEpraspvbLdbyQNM3489hE8,49664
+wrapt/_wrappers.c,sha256=CHiGjkkz7YlPj4aPW8hp0W4rHSPg7ftc5mCnBzHEerU,119897
+wrapt/_wrappers.cp312-win_arm64.pyd,sha256=NpBZicBELftKJsiWdqTXNywxBAYFaJ8ybadaKAP9z1c,50176
 wrapt/arguments.py,sha256=PiJtDy6JMT_0BQei7ZhxJXePIMYyk4GbHFjkTa0n9Q4,2607
 wrapt/decorators.py,sha256=DUSiYLn_xgxLvbEesol1tbmufvmn8Q6l6gcVnrU6RDU,21613
 wrapt/importer.py,sha256=NdOh6aVIIg5Fi1S8eLKL-wmX3qZ6SgOdk3_Xy7RQ5Q4,12467
@@ -10,9 +10,9 @@ wrapt/patches.py,sha256=yYh5_hXX-pIb8HyIs37MwCD6vUTeR96gsxd4W1N-mkw,10146
 wrapt/proxies.py,sha256=jMB_MxrlxF8guNHL6WXKFjuuHZndv4zAXfGaHObZPxk,10300
 wrapt/py.typed,sha256=2Ox0AU4FNnXJ0T5LyWKc_BbJXopWJJUFe8o2V7Zhvjc,9
 wrapt/weakrefs.py,sha256=IlTqGPbJ3mENcvPCboI0tYAzrhZg34cgRqWrP23GkHc,4744
-wrapt/wrappers.py,sha256=h_M2BZyoOm9KJmzTUmhTmLz2-LAXtZ4JtVrZ9KdD0a4,35274
-wrapt-2.0.0rc5.dist-info/licenses/LICENSE,sha256=ROwiyK8_2l1SENok8IcFjc-zHUnlK1JrE_454sDRC-4,1328
-wrapt-2.0.0rc5.dist-info/METADATA,sha256=CqN_5FKP4kbK3AD7MTBmkNTeA7MQjj3GqgtZv8cs7Z4,9029
-wrapt-2.0.0rc5.dist-info/WHEEL,sha256=me1aG6nvouDIdjWXNa5q_zebZZEPzD53N4rwsapSjvI,101
-wrapt-2.0.0rc5.dist-info/top_level.txt,sha256=Jf7kcuXtwjUJMwOL0QzALDg2WiSiXiH9ThKMjN64DW0,6
-wrapt-2.0.0rc5.dist-info/RECORD,,
+wrapt/wrappers.py,sha256=PA3nS-5AW8BS5WYKqN32dsxKgLw0sTMcupLPABaLI-Q,35419
+wrapt-2.0.0rc6.dist-info/licenses/LICENSE,sha256=ROwiyK8_2l1SENok8IcFjc-zHUnlK1JrE_454sDRC-4,1328
+wrapt-2.0.0rc6.dist-info/METADATA,sha256=FUH4kWyhbTu3os8xJZfQbyKBbySt0iW4YjmB29-RDd8,9029
+wrapt-2.0.0rc6.dist-info/WHEEL,sha256=me1aG6nvouDIdjWXNa5q_zebZZEPzD53N4rwsapSjvI,101
+wrapt-2.0.0rc6.dist-info/top_level.txt,sha256=Jf7kcuXtwjUJMwOL0QzALDg2WiSiXiH9ThKMjN64DW0,6
+wrapt-2.0.0rc6.dist-info/RECORD,,