Skip to content

Commit

Permalink
Annotate harvester_rpc_client.py (#15379)
Browse files Browse the repository at this point in the history
Annotate harvester_rpc_client.py.
  • Loading branch information
AmineKhaldi authored Jun 21, 2023
1 parent 423cbe6 commit 1904baf
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 6 deletions.
22 changes: 17 additions & 5 deletions chia/rpc/harvester_rpc_client.py
Original file line number Diff line number Diff line change
@@ -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

Expand All @@ -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
1 change: 0 additions & 1 deletion mypy-exclusions.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 1904baf

Please sign in to comment.