wrapt 2.0.0rc5__cp39-cp39-macosx_11_0_arm64.whl → 2.0.0rc6__cp39-cp39-macosx_11_0_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,18 +1,18 @@
-wrapt-2.0.0rc5.dist-info/RECORD,,
-wrapt-2.0.0rc5.dist-info/WHEEL,sha256=bDWaFWigpG5bEpqw9IoRiyYs8MvmSFh0OhUAOoi_-KA,134
-wrapt-2.0.0rc5.dist-info/top_level.txt,sha256=Jf7kcuXtwjUJMwOL0QzALDg2WiSiXiH9ThKMjN64DW0,6
-wrapt-2.0.0rc5.dist-info/METADATA,sha256=AodtQOhIu5GHSuKT_mm0Auf-MYxFecLHeS3NW24gnSE,8831
-wrapt-2.0.0rc5.dist-info/licenses/LICENSE,sha256=mrxBqmuGkMB-3ptEt8YjPQFCkO0eO1zMN-KSKVtdBY8,1304
 wrapt/importer.py,sha256=E16XxhomZuX5jM_JEH4ZO-MYpNou1t5RZLJHWLCtsgM,12135
 wrapt/__init__.pyi,sha256=sEJ93UmAcYesuYzTdyhmqCX5dQWyfD5IIJKT-wox3R8,9316
 wrapt/arguments.py,sha256=q4bxH7GoCXhTCgxy-AEyvSnOq0ovMSHjN7ru3HWxlhA,2548
-wrapt/__init__.py,sha256=63SajwhyAXfqtZmgp5ok0hTKnMc2coc37Zk0YLzibxE,1528
-wrapt/_wrappers.cpython-39-darwin.so,sha256=2KWmTldgh-0EMNFfH1tVaWDBx4zOeis5KElT1XEIVAE,88304
+wrapt/__init__.py,sha256=4GvDdAUm7sO2zUv-Ai6tRo4TFCbZl_R3KzK59wzjfLI,1528
+wrapt/_wrappers.cpython-39-darwin.so,sha256=eblRxM5Vgm-AdYfNYgMZwSO2pU6FeVPmBwPrIVCJ4gs,88304
 wrapt/__wrapt__.py,sha256=E_Yo7fIk51_OjMzlqXKtIhvplbYbJIPcTGD20b0GWM8,1431
 wrapt/patches.py,sha256=WJAKwOEeozpqgLAq_BlEu2HWbjMg9yaR65szi8J4GRQ,9907
-wrapt/wrappers.py,sha256=xUYyI8ID4gIuUqfu4OP-d8t37SYZVducOfXEVhyhbj8,34297
+wrapt/wrappers.py,sha256=btpuNklFXdpnWlQVrgib1cY08M5tm1vTSd_XEjfEgxs,34439
 wrapt/proxies.py,sha256=x9mergCbZIW4pplUmACkSU6iH0iWy9xNadQUoSp3ksA,10018
 wrapt/weakrefs.py,sha256=5HehYcKOKsRIghxLwnCZ4EJ67rhc0z1GH__pRILRLDc,4630
 wrapt/py.typed,sha256=la67KBlbjXN-_-DfGNcdOcjYumVpKG_Tkw-8n5dnGB4,8
-wrapt/_wrappers.c,sha256=LKRxEkHK-ERyzbHO_cNm6KRRj6FP0C7bUzLtGwxSJqA,115105
+wrapt/_wrappers.c,sha256=ux1b_-Me_8QpJ_QRWvCq9jt_1CZ-9TFA-cVSLqi4_Zg,115800
 wrapt/decorators.py,sha256=ePWsg43Q5uHYSBGvbybwIZpsYdIhLGX9twbGZ7ygous,21091
+wrapt-2.0.0rc6.dist-info/RECORD,,
+wrapt-2.0.0rc6.dist-info/WHEEL,sha256=bDWaFWigpG5bEpqw9IoRiyYs8MvmSFh0OhUAOoi_-KA,134
+wrapt-2.0.0rc6.dist-info/top_level.txt,sha256=Jf7kcuXtwjUJMwOL0QzALDg2WiSiXiH9ThKMjN64DW0,6
+wrapt-2.0.0rc6.dist-info/METADATA,sha256=PJAKVeZj-bxtyCPugdXc44LseD2LTYHJe_zsgpD9tec,8831
+wrapt-2.0.0rc6.dist-info/licenses/LICENSE,sha256=mrxBqmuGkMB-3ptEt8YjPQFCkO0eO1zMN-KSKVtdBY8,1304