refactor: update type annotations

src/poetry_plugin_dotenv/__init__.py

@@ -1,5 +1,8 @@
 """poetry-plugin-dotenv - is the plugin that automatically loads environment variables from a dotenv file into the environment before poetry commands are run."""
 
+from __future__ import annotations
+
+
 __title__ = "poetry-plugin-dotenv"
 __summary__ = "poetry-plugin-dotenv - is the plugin that automatically loads environment variables from a dotenv file into the environment before poetry commands are run."
 __uri__ = "/~https://github.com/pivoshenko/poetry-plugin-dotenv"

src/poetry_plugin_dotenv/dotenv/__init__.py

@@ -5,8 +5,14 @@
 References
 ----------
 #. `python-dotenv </~https://github.com/theskumar/python-dotenv>`_
+
 """
 
+from __future__ import annotations
+
 from poetry_plugin_dotenv.dotenv import core
 from poetry_plugin_dotenv.dotenv import parsers
 from poetry_plugin_dotenv.dotenv import variables
+
+
+__all__ = ["core", "parsers", "variables"]

src/poetry_plugin_dotenv/dotenv/core.py

@@ -16,7 +16,6 @@
 
 
 if TYPE_CHECKING:  # pragma: no cover
-    from collections.abc import Generator
     from collections.abc import Iterable
     from collections.abc import Iterator
 
@@ -57,15 +56,15 @@ def dict(self) -> OrderedDict[str, str]:
         else:
             self._dict = OrderedDict(raw_values)
 
-        return self._dict  # type: ignore[return-value]
+        return self._dict
 
-    def parse(self) -> Generator:
+    def parse(self) -> Iterator[tuple[str, str]]:
         """Parse a dotenv file."""
 
         with self._get_stream() as stream:
             for mapping in parsers.parse_stream(stream):
                 if mapping.key is not None:
-                    yield mapping.key, mapping.value
+                    yield mapping.key, mapping.value  # type: ignore[misc]
 
     def set_as_environment_variables(self) -> bool:
         """Load current dotenv as a system environment variable."""
@@ -124,7 +123,7 @@ def resolve(
 
             result = "".join(atom.resolve(env) for atom in atoms)
 
-        new_values[name] = result  # type: ignore[assignment]
+        new_values[name] = result
 
     return new_values
 
@@ -159,7 +158,6 @@ def find(filename: str = ".env", *, usecwd: bool = False) -> str:
         current_file = __file__
 
         while frame.f_code.co_filename == current_file:
-            # S101 - this asserts is part of the original package
             assert frame.f_back is not None  # noqa: S101
             frame = frame.f_back
 

src/poetry_plugin_dotenv/dotenv/variables.py

@@ -32,7 +32,7 @@ class Literal:
 
     value: str
 
-    def resolve(self, *args, **kwargs) -> str:
+    def resolve(self, *args, **kwargs) -> str:  # type: ignore[no-untyped-def]
         """Get a literal value."""
 
         return self.value
@@ -45,7 +45,7 @@ class Variable:
     name: str
     default: str | None = None
 
-    def resolve(self, env: OrderedDict[str, str], *args, **kwargs) -> str:
+    def resolve(self, env: OrderedDict[str, str], *args, **kwargs) -> str:  # type: ignore[no-untyped-def]
         """Get a variable value."""
 
         default = self.default if self.default else ""

src/poetry_plugin_dotenv/plugin.py

@@ -32,7 +32,7 @@ def activate(self, application: Application) -> None:  # pragma: no cover
 
         application.event_dispatcher.add_listener(COMMAND, listener=self.load)  # type: ignore[union-attr, arg-type]
 
-    def load(self, event: ConsoleCommandEvent, *args, **kwargs) -> None:
+    def load(self, event: ConsoleCommandEvent, *args, **kwargs) -> None:  # type: ignore[no-untyped-def]
         """Load a dotenv file."""
 
         logger = Logger(event)

tests/conftest.py

@@ -11,7 +11,7 @@
 
 if TYPE_CHECKING:
     from collections.abc import Callable
-    from collections.abc import Generator
+    from collections.abc import Iterator
     from pathlib import Path
 
 
@@ -46,7 +46,7 @@ def remove(filepath: str) -> None:
 
 
 @pytest.fixture()
-def dotenv_file(tmp_path: Path) -> Generator:
+def dotenv_file(tmp_path: Path) -> Iterator[str]:
     """Get a dotenv file."""
 
     filepath = tmp_path / ".env"

tests/dotenv/test_core.py

@@ -10,7 +10,7 @@
 from typing import TYPE_CHECKING
 from unittest import mock
 
-import sh
+import sh  # type: ignore[import-untyped]
 import pytest
 
 from poetry_plugin_dotenv.dotenv import core as dotenv