Coverage for /var/srv/projects/api.amasfac.comuna18.com/tmp/venv/lib/python3.9/site-packages/django/db/models/functions/mixins.py: 33%
30 statements
« prev ^ index » next coverage.py v6.4.4, created at 2023-07-17 14:22 -0600
« prev ^ index » next coverage.py v6.4.4, created at 2023-07-17 14:22 -0600
1import sys
3from django.db.models.fields import DecimalField, FloatField, IntegerField
4from django.db.models.functions import Cast
7class FixDecimalInputMixin:
8 def as_postgresql(self, compiler, connection, **extra_context):
9 # Cast FloatField to DecimalField as PostgreSQL doesn't support the
10 # following function signatures:
11 # - LOG(double, double)
12 # - MOD(double, double)
13 output_field = DecimalField(decimal_places=sys.float_info.dig, max_digits=1000)
14 clone = self.copy()
15 clone.set_source_expressions(
16 [
17 Cast(expression, output_field)
18 if isinstance(expression.output_field, FloatField)
19 else expression
20 for expression in self.get_source_expressions()
21 ]
22 )
23 return clone.as_sql(compiler, connection, **extra_context)
26class FixDurationInputMixin:
27 def as_mysql(self, compiler, connection, **extra_context):
28 sql, params = super().as_sql(compiler, connection, **extra_context)
29 if self.output_field.get_internal_type() == "DurationField":
30 sql = "CAST(%s AS SIGNED)" % sql
31 return sql, params
33 def as_oracle(self, compiler, connection, **extra_context):
34 if self.output_field.get_internal_type() == "DurationField":
35 expression = self.get_source_expressions()[0]
36 options = self._get_repr_options()
37 from django.db.backends.oracle.functions import (
38 IntervalToSeconds,
39 SecondsToInterval,
40 )
42 return compiler.compile(
43 SecondsToInterval(
44 self.__class__(IntervalToSeconds(expression), **options)
45 )
46 )
47 return super().as_sql(compiler, connection, **extra_context)
50class NumericOutputFieldMixin:
51 def _resolve_output_field(self):
52 source_fields = self.get_source_fields()
53 if any(isinstance(s, DecimalField) for s in source_fields):
54 return DecimalField()
55 if any(isinstance(s, IntegerField) for s in source_fields):
56 return FloatField()
57 return super()._resolve_output_field() if source_fields else FloatField()