From 12c0c4bbfaf4eee8328cd2d88a98c93a5b64af36 Mon Sep 17 00:00:00 2001
From: Trinh Anh Ngoc <atw1990@gmail.com>
Date: Mon, 9 May 2022 16:47:44 +0700
Subject: [PATCH] Fix Cursor type

---
 odoo-stubs/modules/registry.pyi | 3 ++-
 odoo-stubs/sql_db.pyi           | 6 ++++--
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/odoo-stubs/modules/registry.pyi b/odoo-stubs/modules/registry.pyi
index 8b3fd2a..ade3a86 100644
--- a/odoo-stubs/modules/registry.pyi
+++ b/odoo-stubs/modules/registry.pyi
@@ -1,9 +1,10 @@
 from collections import Mapping, defaultdict as defaultdict
 
-from odoo.sql_db import Cursor
 from odoo.tools import table_exists as table_exists
 from typing import Any, Optional
 
+from ..sql_db import Cursor
+
 _logger: Any
 
 class Registry(Mapping):
diff --git a/odoo-stubs/sql_db.pyi b/odoo-stubs/sql_db.pyi
index ebaa51f..3a87052 100644
--- a/odoo-stubs/sql_db.pyi
+++ b/odoo-stubs/sql_db.pyi
@@ -1,5 +1,7 @@
 import psycopg2.extensions
-from typing import Any, Optional
+from typing import Any, Optional, TypeVar
+
+_T = TypeVar('_T')
 
 _logger: Any
 
@@ -45,7 +47,7 @@ class Cursor:
     def _pop_event_handlers(self): ...
     def commit(self): ...
     def rollback(self): ...
-    def __enter__(self): ...
+    def __enter__(self: _T) -> _T: ...
     def __exit__(self, exc_type: Any, exc_value: Any, traceback: Any) -> None: ...
     def savepoint(self) -> None: ...
     def __getattr__(self, name: Any): ...