feat: lifecycle query runner class (#17440)

posthog/api/query.py

@@ -25,15 +25,16 @@
 from posthog.hogql.errors import HogQLException
 from posthog.hogql.metadata import get_hogql_metadata
 from posthog.hogql.query import execute_hogql_query
-from posthog.hogql_queries.lifecycle_hogql_query import run_lifecycle_query
+
+from posthog.hogql_queries.lifecycle_query_runner import LifecycleQueryRunner
 from posthog.models import Team
 from posthog.models.event.events_query import run_events_query
 from posthog.models.user import User
 from posthog.permissions import ProjectMembershipNecessaryPermissions, TeamMemberAccessPermission
 from posthog.queries.time_to_see_data.serializers import SessionEventsQuerySerializer, SessionsQuerySerializer
 from posthog.queries.time_to_see_data.sessions import get_session_events, get_sessions
 from posthog.rate_limit import AIBurstRateThrottle, AISustainedRateThrottle, TeamRateThrottle
-from posthog.schema import EventsQuery, HogQLQuery, HogQLMetadata, LifecycleQuery
+from posthog.schema import EventsQuery, HogQLQuery, HogQLMetadata
 
 
 class QueryThrottle(TeamRateThrottle):
@@ -221,9 +222,8 @@ def process_query(team: Team, query_json: Dict, default_limit: Optional[int] = N
         metadata_response = get_hogql_metadata(query=metadata_query, team=team)
         return _unwrap_pydantic_dict(metadata_response)
     elif query_kind == "LifecycleQuery":
-        lifecycle_query = LifecycleQuery.parse_obj(query_json)
-        lifecycle_response = run_lifecycle_query(query=lifecycle_query, team=team)
-        return _unwrap_pydantic_dict(lifecycle_response)
+        lifecycle_query_runner = LifecycleQueryRunner(query_json, team)
+        return _unwrap_pydantic_dict(lifecycle_query_runner.run())
     elif query_kind == "DatabaseSchemaQuery":
         database = create_hogql_database(team.pk)
         return serialize_database(database)