From: Aramís Segovia <aramissegovia@gmail.com>
Date: Mon, 12 May 2025 19:59:26 +0000 (-0400)
Subject: Add `__rmatmul__` for symmetry
X-Git-Url: http://git.ipfire.org/?a=commitdiff_plain;h=93aa072dab03c057afe3aa053c6bb215d4f18b31;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

Add `__rmatmul__` for symmetry
---

diff --git a/lib/sqlalchemy/sql/elements.py b/lib/sqlalchemy/sql/elements.py
index 75bf72f033..d4486f5e55 100644
--- a/lib/sqlalchemy/sql/elements.py
+++ b/lib/sqlalchemy/sql/elements.py
@@ -926,6 +926,8 @@ class SQLCoreOperations(Generic[_T_co], ColumnOperators, TypingOnly):
 
         def __matmul__(self, other: Any) -> ColumnElement[Any]: ...
 
+        def __rmatmul__(self, other: Any) -> ColumnElement[Any]: ...
+
         @overload
         def concat(self: _SQO[str], other: Any) -> ColumnElement[str]: ...
 
diff --git a/lib/sqlalchemy/sql/operators.py b/lib/sqlalchemy/sql/operators.py
index ed0582d309..1a3e24711b 100644
--- a/lib/sqlalchemy/sql/operators.py
+++ b/lib/sqlalchemy/sql/operators.py
@@ -681,7 +681,7 @@ class ColumnOperators(Operators):
         return self.operate(rshift, other)
 
     def __matmul__(self, other: Any) -> ColumnOperators:
-        """Implement the @ operator.
+        """Implement the ``@`` operator.
 
         Not used by SQLAlchemy core, this is provided
         for custom operator systems which want to use
@@ -689,6 +689,15 @@ class ColumnOperators(Operators):
         """
         return self.operate(matmul, other)
 
+    def __rmatmul__(self, other: Any) -> ColumnOperators:
+        """Implement the ``@`` operator in reverse.
+
+        Not used by SQLAlchemy core, this is provided
+        for custom operator systems which want to use
+        @ as an extension point.
+        """
+        return self.reverse_operate(matmul, other)
+
     def concat(self, other: Any) -> ColumnOperators:
         """Implement the 'concat' operator.
 
diff --git a/test/sql/test_operators.py b/test/sql/test_operators.py
index 753a2e3c1b..be54cec8e5 100644
--- a/test/sql/test_operators.py
+++ b/test/sql/test_operators.py
@@ -987,6 +987,16 @@ class ExtensionOperatorTest(fixtures.TestBase, testing.AssertsCompiledSQL):
 
         self.assert_compile(Column("x", MyType()) @ 5, "x -> :x_1")
 
+    def test_rmatmul(self):
+        class MyType(UserDefinedType):
+            cache_ok = True
+
+            class comparator_factory(UserDefinedType.Comparator):
+                def __rmatmul__(self, other):
+                    return self.op("->")(other)
+
+        self.assert_compile(5 @ Column("x", MyType()), "x -> :x_1")
+
 
 class JSONIndexOpTest(fixtures.TestBase, testing.AssertsCompiledSQL):
     def setup_test(self):