diff --git a/odoo-stubs/modules/registry.pyi b/odoo-stubs/modules/registry.pyi index 3965981..07c17ef 100644 --- a/odoo-stubs/modules/registry.pyi +++ b/odoo-stubs/modules/registry.pyi @@ -1,6 +1,8 @@ from collections.abc import Mapping from typing import Any +from odoo.sql_db import Cursor + _logger: Any _schema: Any @@ -83,7 +85,7 @@ class Registry(Mapping): def in_test_mode(self): ... def enter_test_mode(self, cr) -> None: ... def leave_test_mode(self) -> None: ... - def cursor(self): ... + def cursor(self) -> Cursor: ... class DummyRLock: def acquire(self) -> None: ... diff --git a/odoo-stubs/sql_db.pyi b/odoo-stubs/sql_db.pyi index e6ab43c..79c3f46 100644 --- a/odoo-stubs/sql_db.pyi +++ b/odoo-stubs/sql_db.pyi @@ -1,6 +1,8 @@ import psycopg2.extensions from typing import Any +from odoo.tools import Callbacks + _logger: Any def unbuffer(symb, cr): ... @@ -16,10 +18,10 @@ sql_counter: int def check(f, self, *args, **kwargs): ... class BaseCursor: - precommit: Any - postcommit: Any - prerollback: Any - postrollback: Any + precommit: Callbacks + postcommit: Callbacks + prerollback: Callbacks + postrollback: Callbacks transaction: Any def __init__(self) -> None: ... def flush(self) -> None: ...