Skip to content

Commit e16a586

Browse files
Fix cursor import issue & name duplication.
1 parent eb9c82a commit e16a586

File tree

1 file changed

+10
-9
lines changed

1 file changed

+10
-9
lines changed

db_wrapper/client/sync_client.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,8 @@
1212

1313
from psycopg2.extras import register_uuid
1414
from psycopg2 import sql
15-
from psycopg2._psycopg import cursor
15+
# pylint can't seem to find the items in psycopg2 despite being available
16+
from psycopg2._psycopg import cursor # pylint: disable=no-name-in-module
1617

1718
from db_wrapper.connection import (
1819
sync_connect,
@@ -54,14 +55,14 @@ def disconnect(self) -> None:
5455

5556
@staticmethod
5657
def _execute_query(
57-
cursor: cursor,
58+
db_cursor: cursor,
5859
query: Query,
5960
params: Optional[Dict[Hashable, Any]] = None,
6061
) -> None:
6162
if params:
62-
cursor.execute(query, params) # type: ignore
63+
db_cursor.execute(query, params) # type: ignore
6364
else:
64-
cursor.execute(query)
65+
db_cursor.execute(query)
6566

6667
def execute(
6768
self,
@@ -78,8 +79,8 @@ def execute(
7879
Returns:
7980
None
8081
"""
81-
with self._connection.cursor() as cursor:
82-
self._execute_query(cursor, query, params)
82+
with self._connection.cursor() as db_cursor:
83+
self._execute_query(db_cursor, query, params)
8384

8485
def execute_and_return(
8586
self,
@@ -96,8 +97,8 @@ def execute_and_return(
9697
Returns:
9798
List containing all the rows that matched the query.
9899
"""
99-
with self._connection.cursor() as cursor:
100-
self._execute_query(cursor, query, params)
100+
with self._connection.cursor() as db_cursor:
101+
self._execute_query(db_cursor, query, params)
101102

102-
result: List[T] = cursor.fetchall()
103+
result: List[T] = db_cursor.fetchall()
103104
return result

0 commit comments

Comments
 (0)