mirror of
https://github.com/PostHog/posthog.git
synced 2024-11-28 18:26:15 +01:00
27 lines
1.0 KiB
Python
27 lines
1.0 KiB
Python
|
from typing import Any, Dict, Tuple
|
||
|
|
|
||
|
|
from ee.clickhouse.models.action import format_action_filter
|
||
|
|
from ee.clickhouse.models.property import parse_prop_clauses
|
||
|
|
from posthog.constants import TREND_FILTER_TYPE_ACTIONS
|
||
|
|
from posthog.models.entity import Entity
|
||
|
|
|
||
|
|
|
||
|
|
def get_entity_filtering_params(
|
||
|
|
entity: Entity, team_id: int, table_name: str = "", *, with_prop_filters: bool = False
|
||
|
|
) -> Tuple[Dict, Dict]:
|
||
|
|
params: Dict[str, Any] = {}
|
||
|
|
content_sql_params: Dict[str, str]
|
||
|
|
prop_filters = ""
|
||
|
|
if with_prop_filters:
|
||
|
|
prop_filters, params = parse_prop_clauses(entity.properties, team_id)
|
||
|
|
if entity.type == TREND_FILTER_TYPE_ACTIONS:
|
||
|
|
action = entity.get_action()
|
||
|
|
action_query, action_params = format_action_filter(action, table_name=table_name)
|
||
|
|
params.update(action_params)
|
||
|
|
content_sql_params = {"entity_query": f"AND {action_query} {prop_filters}"}
|
||
|
|
else:
|
||
|
|
params["event"] = entity.id
|
||
|
|
content_sql_params = {"entity_query": f"AND event = %(event)s {prop_filters}"}
|
||
|
|
|
||
|
|
return params, content_sql_params
|