From 807df75b70e517a37b9715c8f3792e03adea5883 Mon Sep 17 00:00:00 2001
From: Trinh Anh Ngoc <atw1990@gmail.com>
Date: Mon, 9 May 2022 16:39:15 +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 a5fc583..e58df20 100644
--- a/odoo-stubs/modules/registry.pyi
+++ b/odoo-stubs/modules/registry.pyi
@@ -1,7 +1,7 @@
 from collections 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 7eb9459..179a18e 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
 
@@ -47,7 +49,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, flush: bool = ...) -> None: ...
     def __getattr__(self, name: Any): ...