aio-libs · sakosha · May 21, 2025 · May 22, 2025 · May 22, 2025 · May 22, 2025
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -48,3 +48,9 @@ repos:
     name: Check with ruff
     entry: ruff
     args: ["check", "--fix", "."]
+
+  - <<: *python-linters
+    id: mypy
+    name: Type check with mypy
+    entry: mypy
+    args: ["--config-file", "pyproject.toml"]
diff --git a/pyproject.toml b/pyproject.toml
@@ -41,6 +41,11 @@ classifiers = [
     "Topic :: Internet :: File Transfer Protocol (FTP)",
 ]
 
+# TODO: remove `typing-extensions` when requires-python = ">= 3.11"
+dependencies = [
+    "typing-extensions>=4.13.2; python_version < '3.11'",  # 3.9-10 typing support
+]
+
 [project.urls]
 Github = "https://github.com/aio-libs/aioftp"
 Documentation = "https://aioftp.rtfd.io"
@@ -62,6 +67,7 @@ dev = [
     # linters
     "pre-commit",
     "ruff",
+    "mypy",
 ]
 
 [build-system]
@@ -70,6 +76,7 @@ build-backend = "setuptools.build_meta"
 
 [tool.setuptools]
 packages.find.where = ["src"]
+package-data.aioftp = ["py.typed"]
 
 # tools
 [tool.ruff]
@@ -80,6 +87,13 @@ src = ["src"]
 exclude = ["ftpbench.py"]
 
 
+[tool.mypy]
+files = "src/aioftp"
+pretty = true
+strict = true
+ignore_missing_imports = true
+warn_unreachable = true
+
 [tool.coverage]
 run.source = ["./src/aioftp"]
 run.omit = ["./src/aioftp/__main__.py"]

diff --git a/src/aioftp/__init__.py b/src/aioftp/__init__.py
@@ -14,5 +14,5 @@
 version = tuple(map(int, __version__.split(".")))
 
 __all__ = (
-    client.__all__ + server.__all__ + errors.__all__ + common.__all__ + pathio.__all__ + ("version", "__version__")
+    client.__all__ + server.__all__ + errors.__all__ + common.__all__ + pathio.__all__ + ("version", "__version__")  # type: ignore[name-defined]
 )
diff --git a/src/aioftp/__main__.py b/src/aioftp/__main__.py
@@ -5,6 +5,7 @@
 import contextlib
 import logging
 import socket
+from typing import Any
 
 import aioftp
 
@@ -67,7 +68,7 @@
     )
 if args.memory:
     user = aioftp.User(args.login, args.password, base_path="/")
-    path_io_factory = aioftp.MemoryPathIO
+    path_io_factory: type[aioftp.AbstractPathIO[Any]] = aioftp.MemoryPathIO
 else:
     if args.home:
         user = aioftp.User(args.login, args.password, base_path=args.home)
@@ -81,7 +82,7 @@
 }[args.family]
 
 
-async def main():
+async def main() -> None:
     server = aioftp.Server([user], path_io_factory=path_io_factory)
     await server.run(args.host, args.port, family=family)