Annotate harvester_rpc_client.py

chia/rpc/harvester_rpc_client.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Any, Dict, List
+from typing import Any, Dict, List, cast
 
 from chia.rpc.rpc_client import RpcClient
 
@@ -21,13 +21,25 @@ async def refresh_plots(self) -> None:
         await self.fetch("refresh_plots", {})
 
     async def delete_plot(self, filename: str) -> bool:
-        return (await self.fetch("delete_plot", {"filename": filename}))["success"]
+        response = await self.fetch("delete_plot", {"filename": filename})
+        # TODO: casting due to lack of type checked deserialization
+        result = cast(bool, response["success"])
+        return result
 
     async def add_plot_directory(self, dirname: str) -> bool:
-        return (await self.fetch("add_plot_directory", {"dirname": dirname}))["success"]
+        response = await self.fetch("add_plot_directory", {"dirname": dirname})
+        # TODO: casting due to lack of type checked deserialization
+        result = cast(bool, response["success"])
+        return result
 
     async def get_plot_directories(self) -> List[str]:
-        return (await self.fetch("get_plot_directories", {}))["directories"]
+        response = await self.fetch("get_plot_directories", {})
+        # TODO: casting due to lack of type checked deserialization
+        result = cast(List[str], response["directories"])
+        return result
 
     async def remove_plot_directory(self, dirname: str) -> bool:
-        return (await self.fetch("remove_plot_directory", {"dirname": dirname}))["success"]
+        response = await self.fetch("remove_plot_directory", {"dirname": dirname})
+        # TODO: casting due to lack of type checked deserialization
+        result = cast(bool, response["success"])
+        return result

mypy-exclusions.txt

@@ -16,7 +16,6 @@ chia.plotting.util
 chia.pools.pool_puzzles
 chia.pools.pool_wallet
 chia.pools.pool_wallet_info
-chia.rpc.harvester_rpc_client
 chia.rpc.rpc_client
 chia.rpc.util
 chia.rpc.wallet_rpc_api