Replace mypy with zuban: add pre-commit hook, convert [tool.mypy] to [tool.zuban] with default mode, and fix test type narrowing

Author: adamlogan73

.pre-commit-config.yaml

@@ -1,4 +1,12 @@
 repos:
+  - repo: local
+    hooks:
+      - id: zuban
+        name: zuban
+        entry: uv run zuban check homeassistant_api
+        language: system
+        types: [python]
+        pass_filenames: false
   - repo: https://github.com/pre-commit/pre-commit-hooks
     hooks:
       - id: trailing-whitespace

pyproject.toml

@@ -30,7 +30,6 @@ docs = [
     "autodoc-pydantic>=2,<3",
 ]
 dev = [
-    "pre-commit>=4,<5",
     "types-docutils>=0.22",
     "types-simplejson>=3.20",
     "types-toml>=0.10",
@@ -40,6 +39,7 @@ dev = [
     "pytest-cov>=7",
     "pytest>=8",
     "aiosqlite>=0.22",
+    "prek>=0.3.8",
 ]
 
 [tool.pytest.ini_options]
@@ -99,14 +99,9 @@ ignore = [
 [tool.ruff.lint.isort]
 force-single-line = true
 
-[tool.mypy]
-disable_error_code = [
-    "no-untyped-def",
-    "name-defined",
-]
-exclude = [
-    "^docs/"
-]
+[tool.zuban]
+exclude = ["^docs/"]
+untyped_function_return_mode = "inferred"
 
 [tool.hatch.build.targets.wheel]
 packages = ["homeassistant_api"]

tests/test_client.py

@@ -72,12 +72,12 @@ def test_prepare_entity_histories_naive_timestamps() -> None:
         end_timestamp=naive_end,
     )
     # Naive timestamps should get a timezone attached
-    assert "+" in url or "-" in url.split("T")[-1], (
-        "start_timestamp should have timezone offset"
-    )
-    assert "+" in params["end_time"] or "-" in params["end_time"].split("T")[-1], (
-        "end_time should have timezone offset"
-    )
+    start_time = datetime.fromisoformat(url.split("/")[-1])
+    assert start_time.tzinfo is not None, "start_timestamp should have timezone offset"
+    end_time_str = params["end_time"]
+    assert end_time_str is not None
+    end_time = datetime.fromisoformat(end_time_str)
+    assert end_time.tzinfo is not None, "end_time should have timezone offset"
 
 
 # --- BaseClient: prepare_get_logbook_entry_params ---