From 9ad75cda1633f6e1935edb27ba370ebb803594a7 Mon Sep 17 00:00:00 2001 From: Trinh Anh Ngoc Date: Mon, 9 May 2022 16:33:53 +0700 Subject: [PATCH] Fix Cursor type --- odoo-stubs/modules/registry.pyi | 2 +- odoo-stubs/sql_db.pyi | 6 ++++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/odoo-stubs/modules/registry.pyi b/odoo-stubs/modules/registry.pyi index 07c17ef..cee8e77 100644 --- a/odoo-stubs/modules/registry.pyi +++ b/odoo-stubs/modules/registry.pyi @@ -1,7 +1,7 @@ from collections.abc import Mapping from typing import Any -from odoo.sql_db import Cursor +from ..sql_db import Cursor _logger: Any _schema: Any diff --git a/odoo-stubs/sql_db.pyi b/odoo-stubs/sql_db.pyi index 1b39249..11565bf 100644 --- a/odoo-stubs/sql_db.pyi +++ b/odoo-stubs/sql_db.pyi @@ -1,10 +1,12 @@ -from typing import Any +from typing import Any, TypeVar import psycopg2.extensions from .api import Transaction from .tools import Callbacks +_T = TypeVar('_T') + _logger: Any def unbuffer(symb, cr): ... @@ -30,7 +32,7 @@ class BaseCursor: def clear(self) -> None: ... def reset(self) -> None: ... def savepoint(self, flush: bool = ...) -> None: ... - def __enter__(self): ... + def __enter__(self: _T) -> _T: ... def __exit__(self, exc_type, exc_value, traceback) -> None: ... class Cursor(BaseCursor):