feat: Create an empty py.typed file to add mypy type hint support (#61)

* Create py.typed * add __all__ * Inherit ObjectIdField from ObjectId * Update __init__.py
jefersondaniel · Jan 4, 2024 · 10e58de · 10e58de
1 parent a850ea7
commit 10e58de
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/pydantic_mongo/__init__.py b/pydantic_mongo/__init__.py
@@ -1,3 +1,5 @@
-from .abstract_repository import AbstractRepository  # noqa
-from .fields import ObjectIdField  # noqa
-from .version import __version__  # noqa
+from .abstract_repository import AbstractRepository as AbstractRepository 
+from .fields import ObjectIdField as ObjectIdField  
+from .version import __version__ as __version__  
+
+__all__ = [ "fields", "version", "abstract_repository" ]
diff --git a/pydantic_mongo/fields.py b/pydantic_mongo/fields.py
@@ -4,7 +4,7 @@
 from pydantic_core import core_schema
 
 
-class ObjectIdField(str):
+class ObjectIdField(ObjectId):
     @classmethod
     def __get_pydantic_core_schema__(
         cls, _source_type: Any, _handler: Any

diff --git a/pydantic_mongo/py.typed b/pydantic_mongo/py.typed
@@ -0,0 +1 @@
+