Coverage for /Users/davegaeddert/Development/dropseed/plain/plain-models/plain/models/functions/mixins.py: 39%

23 statements  

« prev     ^ index     » next       coverage.py v7.6.1, created at 2024-10-16 22:03 -0500

1import sys 

2 

3from plain.models.fields import DecimalField, FloatField, IntegerField 

4from plain.models.functions import Cast 

5 

6 

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) 

24 

25 

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 

32 

33 

34class NumericOutputFieldMixin: 

35 def _resolve_output_field(self): 

36 source_fields = self.get_source_fields() 

37 if any(isinstance(s, DecimalField) for s in source_fields): 

38 return DecimalField() 

39 if any(isinstance(s, IntegerField) for s in source_fields): 

40 return FloatField() 

41 return super()._resolve_output_field() if source_fields else FloatField()