I'm working on a feature where the query tool will show separate data output for all the select statements run in a single batch. psycopg does provide the result sets (as libpq provides) but there is a catch. Let me explain how pgAdmin currently works:
1. psycopg provides a cursor object on query execution.
2. The cursor object has a function called nextset which can be used to move to the next result set of queries executed.
3. Once you move to the nextset, you cannot get data for the previous set. It will only point to the current set.
4. Right now, we keep on looping through nextset until it reaches the last set and then fetch the data from the last set (using pagination).
5. The fetched result is stored in client memory (python process)
So if we need to show the output of all the queries, we'll have to fetch the result for each query and store it in python memory before moving to the next set.
psycopg already stores the data on the client side, the only difference will be that we'll store all sets and not just the last one.
