Skip to content

Commit

Permalink
Revert the string parsing, issue a commit
Browse files Browse the repository at this point in the history
  • Loading branch information
jbirddog committed Jan 25, 2024
1 parent 851001e commit 99e4957
Showing 1 changed file with 2 additions and 7 deletions.
9 changes: 2 additions & 7 deletions src/connector_postgres_v2/base_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,15 +66,10 @@ def handler(conn: Any, cursor: Any) -> None:

def fetchall(self, sql: str, conn_str: str, values: list) -> ConnectorProxyResponseDict:
def prep_results(results: list) -> list:
# takes the raw results which is a list of a single item list of strings that
# look like tuples with embedded quotes:
# - [["(1,\"some vendor\")"], ["(2,\"another vendor\")"]]
# and turns it into a list of lists of strings that represent the data for each
# column. this way the individual values can be accessed directly from task data.
# - [["1", "some vender"], ["2", "another_vendor"]]
return [r[0][1:-1].replace('"', '').split(",") for r in results]
return [list(result) for result in results]
def handler(conn: Any, cursor: Any) -> list:
cursor.execute(sql, values)
conn.commit()
return prep_results(cursor.fetchall())

return self._execute(sql, conn_str, handler)
Expand Down

0 comments on commit 99e4957

Please sign in to comment.