Example #1
0
    def update_progress(self):
        progress = spread_dict(
            self._statement_progress,
            {
                "total": len(self._statement_ranges),
            },
        )

        self._celery_task.update_state(state="PROGRESS", meta=progress)
Example #2
0
    def on_query_start(self):
        with DBSession() as session:
            query_execution = qe_logic.update_query_execution(
                self._query_execution_id,
                status=QueryExecutionStatus.RUNNING,
                session=session,
            ).to_dict()

        query_execution = spread_dict(query_execution, {
            "total": len(self._statement_ranges),
        })
        socketio.emit(
            "query_start",
            query_execution,
            namespace=QUERY_EXECUTION_NAMESPACE,
            room=self._query_execution_id,
        )
        self.update_progress()