Annotate harvester_rpc_client.py (#15379)
Annotate harvester_rpc_client.py.
This commit is contained in:
parent
423cbe6248
commit
1904baff04
|
@ -1,6 +1,6 @@
|
||||||
from __future__ import annotations
|
from __future__ import annotations
|
||||||
|
|
||||||
from typing import Any, Dict, List
|
from typing import Any, Dict, List, cast
|
||||||
|
|
||||||
from chia.rpc.rpc_client import RpcClient
|
from chia.rpc.rpc_client import RpcClient
|
||||||
|
|
||||||
|
@ -21,13 +21,25 @@ class HarvesterRpcClient(RpcClient):
|
||||||
await self.fetch("refresh_plots", {})
|
await self.fetch("refresh_plots", {})
|
||||||
|
|
||||||
async def delete_plot(self, filename: str) -> bool:
|
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:
|
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]:
|
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:
|
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
|
||||||
|
|
|
@ -16,7 +16,6 @@ chia.plotting.util
|
||||||
chia.pools.pool_puzzles
|
chia.pools.pool_puzzles
|
||||||
chia.pools.pool_wallet
|
chia.pools.pool_wallet
|
||||||
chia.pools.pool_wallet_info
|
chia.pools.pool_wallet_info
|
||||||
chia.rpc.harvester_rpc_client
|
|
||||||
chia.rpc.rpc_client
|
chia.rpc.rpc_client
|
||||||
chia.rpc.util
|
chia.rpc.util
|
||||||
chia.rpc.wallet_rpc_api
|
chia.rpc.wallet_rpc_api
|
||||||
|
|
Loading…
Reference in New Issue