上下文管理器光标的 Mariadb Python 模块类型提示

如何解决上下文管理器光标的 Mariadb Python 模块类型提示

我无法辨别正确的输入。我怎样才能弄清楚这应该是什么？

//in my service layer

public void test() {
    //do stuff here
}

//then in the controller

public ResponseEntity<Void> test() {
    return ok(service.test());
}

当我尝试使用具有此类型的 context_manager 时，它返回：

from contextlib import contextmanager

import mariadb


@contextmanager
def mariadb_con(
        user: str,password: str,host: str,database: str,port: int,commit: bool = False
        ) -> mariadb.connection.cursor:
    con = mariadb.connect(
            user=user,password=password,host=host,database=database,port=port
            )
    cur = con.cursor()
    yield cur
    if commit is True:
        con.commit()
    cur.close()
    con.close()

解决方法

玩完之后我发现了这个：

mariadb._mariadb.connection.cursor
<method 'cursor' of 'mariadb.connection' objects>

这似乎是连接类的游标方法。

打印类型后这看起来是正确的：

print(type(mariadb._mariadb.connection.cursor))
<class 'method_descriptor'>

然后我将其添加到上下文管理器中。或许有经验的人可以告诉我这是不是误会。

from contextlib import contextmanager
from typing import Generator

import mariadb


@contextmanager
def mariadb_con(
        user: str,password: str,host: str,database: str,port: int,commit: bool = False
        ) -> Generator[mariadb._mariadb.connection.cursor,None,None]:
    con = mariadb.connect(
            user=user,password=password,host=host,database=database,port=port
            )
    cur = con.cursor()
    yield cur