mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-28 09:16:49 +01:00
784c7d3b08
* Resolve weird SQL formatting issue * Use a discriminated union for ActorType * Use standard response types for related groups * Update typing * Always filter related actors by group type index * Update snapshots & typing
49 lines
2.0 KiB
Python
49 lines
2.0 KiB
Python
from typing import Optional
|
|
|
|
from rest_framework.exceptions import ValidationError
|
|
|
|
from ee.clickhouse.queries.actor_base_query import ActorBaseQuery
|
|
from ee.clickhouse.queries.funnels.funnel_trends import ClickhouseFunnelTrends
|
|
from ee.clickhouse.sql.funnels.funnel import FUNNEL_PERSONS_BY_STEP_SQL
|
|
from posthog.constants import DROP_OFF, ENTRANCE_PERIOD_START
|
|
from posthog.models.filters.filter import Filter
|
|
from posthog.models.filters.mixins.utils import cached_property
|
|
|
|
|
|
class ClickhouseFunnelTrendsActors(ClickhouseFunnelTrends, ActorBaseQuery):
|
|
_filter: Filter
|
|
|
|
@cached_property
|
|
def aggregation_group_type_index(self):
|
|
return self._filter.aggregation_group_type_index
|
|
|
|
def actor_query(self, limit_actors: Optional[bool] = True):
|
|
drop_off = self._filter.drop_off
|
|
if drop_off is None:
|
|
raise ValidationError(f"Filter parameter {DROP_OFF} must be provided and a bool for funnel trends persons!")
|
|
|
|
entrance_period_start = self._filter.entrance_period_start
|
|
if not entrance_period_start:
|
|
raise ValidationError(
|
|
f"Filter parameter {ENTRANCE_PERIOD_START} must be provided and a datetime for funnel trends persons!"
|
|
)
|
|
|
|
step_counts_query = self.get_step_counts_without_aggregation_query(
|
|
specific_entrance_period_start=entrance_period_start
|
|
)
|
|
# Expects multiple rows for same person, first event time, steps taken.
|
|
self.params.update(self.funnel_order.params)
|
|
|
|
_, reached_to_step_count_condition, did_not_reach_to_step_count_condition = self.get_steps_reached_conditions()
|
|
|
|
return (
|
|
FUNNEL_PERSONS_BY_STEP_SQL.format(
|
|
steps_per_person_query=step_counts_query,
|
|
persons_steps=did_not_reach_to_step_count_condition if drop_off else reached_to_step_count_condition,
|
|
extra_fields="",
|
|
limit="LIMIT %(limit)s" if limit_actors else "",
|
|
offset="OFFSET %(offset)s" if limit_actors else "",
|
|
),
|
|
self.params,
|
|
)
|