From 98a7157fa6365fa7d3d3cf747eef478406009c1c Mon Sep 17 00:00:00 2001 From: Trinh Anh Ngoc Date: Mon, 9 May 2022 12:05:24 +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 500bafd..6e85ba7 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, Optional -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 e540f04..989b39a 100644 --- a/odoo-stubs/sql_db.pyi +++ b/odoo-stubs/sql_db.pyi @@ -1,8 +1,10 @@ import psycopg2.extensions -from typing import Any, Optional +from typing import Any, Optional, TypeVar from .tools import Callbacks +_T = TypeVar('_T') + _logger: Any def unbuffer(symb: Any, cr: Any): ... @@ -24,7 +26,7 @@ class BaseCursor: postrollback: Callbacks def __init__(self) -> None: ... def savepoint(self, flush: bool = ...) -> None: ... - def __enter__(self): ... + def __enter__(self: _T) -> _T: ... def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None: ... class Cursor(BaseCursor):