SQLAlchemyのORMでquery()に続くfilter()やall()などLSPで補完したい

Question

ENGINE = create_engine('mysql://....', encoding='utf-8')
session = scoped_session(session_maker(....))
Base = declarative_base()
Base.query = session.query_property()

class User(Base):
    __tablename__ = 'users'
   
    id = Column(INT, primary_key=True, autoincrement=True)
    name = Column(VARCHAR(255))

このようなModelがあったとしまして、User.query.filter(...).first()のfilter()やfirst()をVSCodeのLSPで補完する方法は無いのかなと思いました。
queryも補完できればしたいです・・・。

なんとか型を見つけ出して宣言しようとしてみたのですがSQLAlchemy内部はメタプログラミングだらけで型らしい型はみつかりませんでした。

もし補完する方法ありましたら教えていただけないでしょうか。
よろしくお願いします。

Answer 1

from sqlalchemy.orm import Query

ENGINE = create_engine('mysql://....', encoding='utf-8')
session = scoped_session(session_maker(....))
Base = declarative_base()
Base.query = session.query_property()

class User(Base):
    __tablename__ = 'users'
    query: Query

    id = Column(INT, primary_key=True, autoincrement=True)
    name = Column(VARCHAR(255))

完璧な回答ではないですが、Modelの属性にquery: Queryと宣言しておくとUser.query.<-ここでall()やfilter()が補完きました。

ただしfilterの後、User.query.filter().は、再び型がわからなくなってしまうので補完できません。
Pythonの型は難しいです。