Skip to content
This repository was archived by the owner on Aug 19, 2025. It is now read-only.

Commit 2d0177c

Browse files
Support integer index lookup in postgres records
1 parent fd42e3f commit 2d0177c

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

databases/backends/postgres.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -72,14 +72,12 @@ def __init__(
7272
self._row = row
7373
self._result_columns = result_columns
7474
self._dialect = dialect
75-
self._column_map = {
76-
column_name: (idx, datatype)
77-
for idx, (column_name, _, _, datatype) in enumerate(self._result_columns)
78-
}
79-
self._column_map_full = {
80-
str(column[0]): (idx, datatype)
81-
for idx, (_, _, column, datatype) in enumerate(self._result_columns)
82-
}
75+
self._column_map = {}
76+
self._column_map_full = {}
77+
for idx, (column_name, _, _, datatype) in enumerate(self._result_columns):
78+
self._column_map[column_name] = (idx, datatype)
79+
self._column_map[idx] = (idx, datatype)
80+
self._column_map_full[str(column[0])] = (idx, datatype)
8381

8482
def __getitem__(self, key: typing.Any) -> typing.Any:
8583
if len(self._column_map) == 0: # raw query

0 commit comments

Comments
 (0)