Skip to content

Commit

Permalink
process_new_vendor fix, part duex (#2)
Browse files Browse the repository at this point in the history
  • Loading branch information
jbirddog authored Jan 25, 2024
1 parent 851001e commit af95cbc
Showing 1 changed file with 2 additions and 9 deletions.
11 changes: 2 additions & 9 deletions src/connector_postgres_v2/base_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -65,17 +65,10 @@ 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: 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]
def handler(conn: Any, cursor: Any) -> list:
cursor.execute(sql, values)
return prep_results(cursor.fetchall())
conn.commit()
return cursor.fetchall()

return self._execute(sql, conn_str, handler)

Expand Down

0 comments on commit af95cbc

Please sign in to comment.