Как я могу ввести подсказку SQLAlchemy Queries? ⇐ Python

[Anonymous]

обновление 2024: sqlalchemy 2.0 имеет гораздо лучшую поддержку типизации
Я использую Pylance (с подсказками типов на базе PyRight) в VSCode и имею следующую функцию:

Код: Выделить всё

def sample(session: sqlalchemy.orm.Session) -> sqlalchemy.orm.query.Query:
return session.query((MyModel)

Если посмотреть на вывод типов Pylance, то эта функция фактически имеет возвращаемый тип Query[Unknown]. Когда я меняю подсказку типа на sqlalchemy.orm.query.Query[MyModel], Pylance может правильно вывести для нее соответствующие типы. отлично! Пока я на самом деле не запустил код и не получил TypeError: объект 'type' не подлежит подписке.
Понятно, почему я получаю эту ошибку, но похоже, что она должна быть какой-нибудь способ заставить это работать, иначе Пайланс просто насмехается надо мной. Я нашел один обходной путь (который действительно работает):

Код: Выделить всё

def sample(session: sqlalchemy.orm.Session) -> (
sqlalchemy.orm.query.Query[MyModel]
if typing.TYPE_CHECKING
else sqlalchemy.orm.query.Query
):
return session.query(MyModel)

но это явно ужасно, и я это ненавижу. Я мог бы создать вспомогательную функцию, которая бы извлекала это и была бы более читабельной, но я действительно чувствую, что это должно быть возможно без ужасного обходного пути!

Подробнее здесь: https://stackoverflow.com/questions/702 ... my-queries