Ticket #25912: add_bitwise_operators.diff

django/db/models/expressions.py

@@ -32 +32 @@
     # usage.
     BITAND = '&'
     BITOR = '|'
+    BITLEFTSHIFT = '<<'
+    BITRIGHTSHIFT = '>>'
 
     def _combine(self, other, connector, reversed, node=None):
         if not hasattr(other, 'resolve_expression'):
@@ -78 +80 @@
     def bitand(self, other):
         return self._combine(other, self.BITAND, False)
 
+    def bitleftshift(self, other):
+        return self._combine(other, self.BITLEFTSHIFT, False)
+
+    def bitrightshift(self, other):
+        return self._combine(other, self.BITRIGHTSHIFT, False)
+
     def __or__(self, other):
         raise NotImplementedError(
             "Use .bitand() and .bitor() for bitwise logical operations."

tests/expressions/tests.py

@@ -604 +604 @@
         self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 40)
         self.assertEqual(Number.objects.get(pk=self.n.pk).float, Approximate(15.500, places=3))
 
+    def test_lefthand_bitwise_binary_left_shift_operator(self):
+        Number.objects.filter(pk=self.n.pk).update(integer=F('integer').bitleftshift(2))
+        self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 168)
+
+    def test_lefthand_bitwise_binary_right_shift_operator(self):
+        Number.objects.filter(pk=self.n.pk).update(integer=F('integer').bitrightshift(2))
+        self.assertEqual(Number.objects.get(pk=self.n.pk).integer, 10)
+
     @skipUnlessDBFeature('supports_bitwise_or')
     def test_lefthand_bitwise_or(self):
         # LH Bitwise or on integers