PraBlatt3

2019-10-29 14:28:14 +01:00
parent c0653bb776
commit 59716deba1
4341 changed files with 585665 additions and 0 deletions
@@ -0,0 +1,2 @@
+from .general import *  # NOQA
+from .statistics import *  # NOQA
@@ -0,0 +1,63 @@
+from django.contrib.postgres.fields import ArrayField, JSONField
+from django.db.models.aggregates import Aggregate
+
+from .mixins import OrderableAggMixin
+
+__all__ = [
+    'ArrayAgg', 'BitAnd', 'BitOr', 'BoolAnd', 'BoolOr', 'JSONBAgg', 'StringAgg',
+]
+
+
+class ArrayAgg(OrderableAggMixin, Aggregate):
+    function = 'ARRAY_AGG'
+    template = '%(function)s(%(distinct)s%(expressions)s %(ordering)s)'
+    allow_distinct = True
+
+    @property
+    def output_field(self):
+        return ArrayField(self.source_expressions[0].output_field)
+
+    def convert_value(self, value, expression, connection):
+        if not value:
+            return []
+        return value
+
+
+class BitAnd(Aggregate):
+    function = 'BIT_AND'
+
+
+class BitOr(Aggregate):
+    function = 'BIT_OR'
+
+
+class BoolAnd(Aggregate):
+    function = 'BOOL_AND'
+
+
+class BoolOr(Aggregate):
+    function = 'BOOL_OR'
+
+
+class JSONBAgg(Aggregate):
+    function = 'JSONB_AGG'
+    output_field = JSONField()
+
+    def convert_value(self, value, expression, connection):
+        if not value:
+            return []
+        return value
+
+
+class StringAgg(OrderableAggMixin, Aggregate):
+    function = 'STRING_AGG'
+    template = "%(function)s(%(distinct)s%(expressions)s, '%(delimiter)s'%(ordering)s)"
+    allow_distinct = True
+
+    def __init__(self, expression, delimiter, **extra):
+        super().__init__(expression, delimiter=delimiter, **extra)
+
+    def convert_value(self, value, expression, connection):
+        if not value:
+            return ''
+        return value
@@ -0,0 +1,57 @@
+from django.db.models.expressions import F, OrderBy
+
+
+class OrderableAggMixin:
+
+    def __init__(self, expression, ordering=(), **extra):
+        if not isinstance(ordering, (list, tuple)):
+            ordering = [ordering]
+        ordering = ordering or []
+        # Transform minus sign prefixed strings into an OrderBy() expression.
+        ordering = (
+            (OrderBy(F(o[1:]), descending=True) if isinstance(o, str) and o[0] == '-' else o)
+            for o in ordering
+        )
+        super().__init__(expression, **extra)
+        self.ordering = self._parse_expressions(*ordering)
+
+    def resolve_expression(self, *args, **kwargs):
+        self.ordering = [expr.resolve_expression(*args, **kwargs) for expr in self.ordering]
+        return super().resolve_expression(*args, **kwargs)
+
+    def as_sql(self, compiler, connection):
+        if self.ordering:
+            ordering_params = []
+            ordering_expr_sql = []
+            for expr in self.ordering:
+                expr_sql, expr_params = expr.as_sql(compiler, connection)
+                ordering_expr_sql.append(expr_sql)
+                ordering_params.extend(expr_params)
+            sql, sql_params = super().as_sql(compiler, connection, ordering=(
+                'ORDER BY ' + ', '.join(ordering_expr_sql)
+            ))
+            return sql, sql_params + ordering_params
+        return super().as_sql(compiler, connection, ordering='')
+
+    def set_source_expressions(self, exprs):
+        # Extract the ordering expressions because ORDER BY clause is handled
+        # in a custom way.
+        self.ordering = exprs[self._get_ordering_expressions_index():]
+        return super().set_source_expressions(exprs[:self._get_ordering_expressions_index()])
+
+    def get_source_expressions(self):
+        return self.source_expressions + self.ordering
+
+    def get_source_fields(self):
+        # Filter out fields contributed by the ordering expressions as
+        # these should not be used to determine which the return type of the
+        # expression.
+        return [
+            e._output_field_or_none
+            for e in self.get_source_expressions()[:self._get_ordering_expressions_index()]
+        ]
+
+    def _get_ordering_expressions_index(self):
+        """Return the index at which the ordering expressions start."""
+        source_expressions = self.get_source_expressions()
+        return len(source_expressions) - len(self.ordering)
@@ -0,0 +1,69 @@
+from django.db.models import FloatField, IntegerField
+from django.db.models.aggregates import Aggregate
+
+__all__ = [
+    'CovarPop', 'Corr', 'RegrAvgX', 'RegrAvgY', 'RegrCount', 'RegrIntercept',
+    'RegrR2', 'RegrSlope', 'RegrSXX', 'RegrSXY', 'RegrSYY', 'StatAggregate',
+]
+
+
+class StatAggregate(Aggregate):
+    output_field = FloatField()
+
+    def __init__(self, y, x, output_field=None, filter=None):
+        if not x or not y:
+            raise ValueError('Both y and x must be provided.')
+        super().__init__(y, x, output_field=output_field, filter=filter)
+
+    def resolve_expression(self, query=None, allow_joins=True, reuse=None, summarize=False, for_save=False):
+        return super().resolve_expression(query, allow_joins, reuse, summarize)
+
+
+class Corr(StatAggregate):
+    function = 'CORR'
+
+
+class CovarPop(StatAggregate):
+    def __init__(self, y, x, sample=False, filter=None):
+        self.function = 'COVAR_SAMP' if sample else 'COVAR_POP'
+        super().__init__(y, x, filter=filter)
+
+
+class RegrAvgX(StatAggregate):
+    function = 'REGR_AVGX'
+
+
+class RegrAvgY(StatAggregate):
+    function = 'REGR_AVGY'
+
+
+class RegrCount(StatAggregate):
+    function = 'REGR_COUNT'
+    output_field = IntegerField()
+
+    def convert_value(self, value, expression, connection):
+        return 0 if value is None else value
+
+
+class RegrIntercept(StatAggregate):
+    function = 'REGR_INTERCEPT'
+
+
+class RegrR2(StatAggregate):
+    function = 'REGR_R2'
+
+
+class RegrSlope(StatAggregate):
+    function = 'REGR_SLOPE'
+
+
+class RegrSXX(StatAggregate):
+    function = 'REGR_SXX'
+
+
+class RegrSXY(StatAggregate):
+    function = 'REGR_SXY'
+
+
+class RegrSYY(StatAggregate):
+    function = 'REGR_SYY'