mixins.py 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. import sys
  2. from django.db.models.fields import DecimalField, FloatField, IntegerField
  3. from django.db.models.functions import Cast
  4. class FixDecimalInputMixin:
  5. def as_postgresql(self, compiler, connection, **extra_context):
  6. # Cast FloatField to DecimalField as PostgreSQL doesn't support the
  7. # following function signatures:
  8. # - LOG(double, double)
  9. # - MOD(double, double)
  10. output_field = DecimalField(decimal_places=sys.float_info.dig, max_digits=1000)
  11. clone = self.copy()
  12. clone.set_source_expressions(
  13. [
  14. Cast(expression, output_field)
  15. if isinstance(expression.output_field, FloatField)
  16. else expression
  17. for expression in self.get_source_expressions()
  18. ]
  19. )
  20. return clone.as_sql(compiler, connection, **extra_context)
  21. class FixDurationInputMixin:
  22. def as_mysql(self, compiler, connection, **extra_context):
  23. sql, params = super().as_sql(compiler, connection, **extra_context)
  24. if self.output_field.get_internal_type() == "DurationField":
  25. sql = "CAST(%s AS SIGNED)" % sql
  26. return sql, params
  27. def as_oracle(self, compiler, connection, **extra_context):
  28. if self.output_field.get_internal_type() == "DurationField":
  29. expression = self.get_source_expressions()[0]
  30. options = self._get_repr_options()
  31. from django.db.backends.oracle.functions import (
  32. IntervalToSeconds,
  33. SecondsToInterval,
  34. )
  35. return compiler.compile(
  36. SecondsToInterval(
  37. self.__class__(IntervalToSeconds(expression), **options)
  38. )
  39. )
  40. return super().as_sql(compiler, connection, **extra_context)
  41. class NumericOutputFieldMixin:
  42. def _resolve_output_field(self):
  43. source_fields = self.get_source_fields()
  44. if any(isinstance(s, DecimalField) for s in source_fields):
  45. return DecimalField()
  46. if any(isinstance(s, IntegerField) for s in source_fields):
  47. return FloatField()
  48. return super()._resolve_output_field() if source_fields else FloatField()