psycopg: How can it be made Pyright-compatible?

[rafrafek]

Hi! I'm using the latest Pyright and the latest psycopg and I'm getting errors for my code. How can I make my code Pyright-compatible, or how can be psycopg made Pyright-compatible in this case? Basically I need my rows to be dicts rather than the default tuples.

import psycopg
from psycopg.rows import dict_row, DictRow

def create_psycopg_connection() -> psycopg.Connection[DictRow]:
    return psycopg.connect(
        host="",
        port="",
        user="",
        password="",
        database="",
        row_factory=dict_row,
    )

Errors:

/Users/foo/scripts/demo.py
  /Users/foo/scripts/demo.py:5:12 - error: Expression of type "Connection[TupleRow]" is incompatible with return type "Connection[DictRow]"
    "Connection[TupleRow]" is incompatible with "Connection[DictRow]"
      Type parameter "Row@Connection" is covariant, but "TupleRow" is not a subtype of "DictRow"
        "tuple[Any, ...]" is incompatible with "dict[str, Any]" (reportReturnType)
  /Users/foo/scripts/demo.py:11:21 - error: Argument of type "(cursor: BaseCursor[Any, Any]) -> RowMaker[DictRow]" cannot be assigned to parameter "row_factory" of type "RowFactory[TupleRow] | None" in function "connect"
    Type "(cursor: BaseCursor[Any, Any]) -> RowMaker[DictRow]" is incompatible with type "RowFactory[TupleRow] | None"
      Type "(cursor: BaseCursor[Any, Any]) -> RowMaker[DictRow]" is incompatible with type "(__cursor: Cursor[Any], /) -> RowMaker[TupleRow]"
        Function return type "RowMaker[DictRow]" is incompatible with type "RowMaker[TupleRow]"
          "RowMaker[DictRow]" is incompatible with "RowMaker[TupleRow]"
            Type parameter "Row@RowMaker" is covariant, but "DictRow" is not a subtype of "TupleRow"
      "function" is incompatible with "None" (reportArgumentType)

Versions:
pyright 1.1.372

psycopg==3.2.1
psycopg-binary==3.2.1

(All the latest stable)

Similar problem:
psycopg/psycopg#865

[rafrafek]

I've found a workaround!

Instead of passing row_factory=dict_row to psycopg.connect, pass it when creating a cursor, like this:

import psycopg
from psycopg.rows import dict_row, DictRow, TupleRow

def create_psycopg_connection() -> psycopg.Connection[TupleRow]:
    return psycopg.connect(
        host="",
        port="",
        user="",
        password="",
        database="",
    )

connection = create_psycopg_connection()
cursor: psycopg.Cursor[DictRow] = connection.cursor(row_factory=dict_row)