Initial commit

env/lib/python3.10/site-packages/wagtail/test/customuser/fields.py

@@ -0,0 +1,82 @@
+import random
+
+from django.db import models
+
+LOWER_BOUND = -2147483648
+UPPER_BOUND = 2147483647
+SHIFT = 92147483647
+
+
+class ConvertedValue(str):
+    def __new__(cls, value):
+        value = int(value)
+
+        if UPPER_BOUND < value:
+            display_value = value
+            db_value = value - SHIFT
+        else:
+            db_value = value
+            display_value = value + SHIFT
+
+        self = super().__new__(cls, display_value)
+        self.db_value = db_value
+
+        return self
+
+    def __repr__(self):
+        return f"<{self.__class__.__name__}: {self.db_value}>"
+
+    def __eq__(self, other):
+        if isinstance(other, self.__class__):
+            return self.db_value == other.db_value
+        return self.db_value == other
+
+    def __hash__(self):
+        return hash(self.db_value)
+
+
+class ConvertedValueField(models.IntegerField):
+    """
+    Roughly copied from https://github.com/django/django/blob/d6eaf7c0183cd04b78f2a55e1d60bb7e59598310/tests/custom_pk/fields.py
+    """
+
+    def pre_save(self, instance, add):
+        value = getattr(instance, self.attname, None)
+        if not value:
+            value = ConvertedValue(random.randint(LOWER_BOUND, UPPER_BOUND))
+            setattr(instance, self.attname, value)
+        return value
+
+    def to_python(self, value):
+        if not value:
+            return
+        if not isinstance(value, ConvertedValue):
+            value = ConvertedValue(value)
+        return value
+
+    def get_prep_value(self, value):
+        if value is None:
+            return None
+        if not isinstance(value, ConvertedValue):
+            value = ConvertedValue(value)
+        return super().get_prep_value(value.db_value)
+
+    def from_db_value(self, value, expression, connection):
+        if not value:
+            return
+        return ConvertedValue(value)
+
+    def get_db_prep_save(self, value, connection):
+        if not value:
+            return
+        return ConvertedValue(value).db_value
+
+    def get_db_prep_value(self, value, connection, prepared=False):
+        if not value:
+            return
+        return ConvertedValue(value).db_value
+
+    def get_searchable_content(self, value):
+        if not value:
+            return
+        return ConvertedValue(value).db_value