diff --git a/src/connector_postgres_v2/base_command.py b/src/connector_postgres_v2/base_command.py index b4ec248..66ee515 100644 --- a/src/connector_postgres_v2/base_command.py +++ b/src/connector_postgres_v2/base_command.py @@ -1,4 +1,3 @@ -import json from typing import Any import psycopg2 # type: ignore @@ -36,7 +35,7 @@ def _execute(self, sql: str, conn_str: str, handler: Any) -> ConnectorProxyRespo conn.close() command_response: CommandResponseDict = { - "body": json.dumps(command_response_body), + "body": command_response_body, "mimetype": "application/json", } return_response: ConnectorProxyResponseDict = { @@ -66,8 +65,8 @@ def handler(conn: Any, cursor: Any) -> None: return self._execute(sql, conn_str, handler) def fetchall(self, sql: str, conn_str: str, values: list) -> ConnectorProxyResponseDict: - def prep_results(results: dict) -> list: - return list(map(list, results)) + def prep_results(results: list) -> list: + return [r[0][1:-1].replace('"', '').split(",") for r in results] def handler(conn: Any, cursor: Any) -> list: cursor.execute(sql, values) return prep_results(cursor.fetchall())