281 lines
9.3 KiB
Python
281 lines
9.3 KiB
Python
# ------------------------------------------------------------------------------
|
|
# Copyright 2022 Upstream Data Inc -
|
|
# -
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); -
|
|
# you may not use this file except in compliance with the License. -
|
|
# You may obtain a copy of the License at -
|
|
# -
|
|
# http://www.apache.org/licenses/LICENSE-2.0 -
|
|
# -
|
|
# Unless required by applicable law or agreed to in writing, software -
|
|
# distributed under the License is distributed on an "AS IS" BASIS, -
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -
|
|
# See the License for the specific language governing permissions and -
|
|
# limitations under the License. -
|
|
# ------------------------------------------------------------------------------
|
|
from typing import Optional
|
|
|
|
from pyasic import APIError
|
|
from pyasic.config import MinerConfig, MiningModeConfig
|
|
from pyasic.miners.backends import BMMiner
|
|
from pyasic.miners.data import (
|
|
DataFunction,
|
|
DataLocations,
|
|
DataOptions,
|
|
RPCAPICommand,
|
|
WebAPICommand,
|
|
)
|
|
from pyasic.miners.device.firmware import HiveonFirmware
|
|
from pyasic.web.hiveon import HiveonWebAPI
|
|
|
|
HIVEON_MODERN_DATA_LOC = DataLocations(
|
|
**{
|
|
str(DataOptions.MAC): DataFunction(
|
|
"_get_mac",
|
|
[WebAPICommand("web_get_system_info", "get_system_info")],
|
|
),
|
|
str(DataOptions.HOSTNAME): DataFunction(
|
|
"_get_hostname",
|
|
[WebAPICommand("web_get_system_info", "get_system_info")],
|
|
),
|
|
str(DataOptions.API_VERSION): DataFunction(
|
|
"_get_api_ver",
|
|
[RPCAPICommand("rpc_version", "version")],
|
|
),
|
|
str(DataOptions.FW_VERSION): DataFunction(
|
|
"_get_fw_ver",
|
|
[RPCAPICommand("rpc_version", "version")],
|
|
),
|
|
str(DataOptions.HASHRATE): DataFunction(
|
|
"_get_hashrate",
|
|
[RPCAPICommand("rpc_summary", "summary")],
|
|
),
|
|
str(DataOptions.EXPECTED_HASHRATE): DataFunction(
|
|
"_get_expected_hashrate",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.HASHBOARDS): DataFunction(
|
|
"_get_hashboards",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.WATTAGE): DataFunction(
|
|
"_get_wattage",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.FANS): DataFunction(
|
|
"_get_fans",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.UPTIME): DataFunction(
|
|
"_get_uptime",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.POOLS): DataFunction(
|
|
"_get_pools",
|
|
[RPCAPICommand("rpc_pools", "pools")],
|
|
),
|
|
str(DataOptions.FAULT_LIGHT): DataFunction(
|
|
"_get_fault_light",
|
|
[WebAPICommand("web_get_blink_status", "get_blink_status")],
|
|
),
|
|
str(DataOptions.IS_MINING): DataFunction(
|
|
"_is_mining",
|
|
[WebAPICommand("web_get_conf", "get_miner_conf")],
|
|
),
|
|
}
|
|
)
|
|
|
|
|
|
class HiveonModern(HiveonFirmware, BMMiner):
|
|
data_locations = HIVEON_MODERN_DATA_LOC
|
|
|
|
web: HiveonWebAPI
|
|
_web_cls = HiveonWebAPI
|
|
|
|
async def get_config(self) -> MinerConfig:
|
|
data = await self.web.get_miner_conf()
|
|
if data:
|
|
self.config = MinerConfig.from_hiveon_modern(data)
|
|
return self.config
|
|
|
|
async def fault_light_on(self) -> bool:
|
|
data = await self.web.blink(blink=True)
|
|
if data:
|
|
if data.get("code") == "B000":
|
|
self.light = True
|
|
return self.light
|
|
|
|
async def fault_light_off(self) -> bool:
|
|
data = await self.web.blink(blink=False)
|
|
if data:
|
|
if data.get("code") == "B100":
|
|
self.light = False
|
|
return self.light
|
|
|
|
async def reboot(self) -> bool:
|
|
data = await self.web.reboot()
|
|
if data:
|
|
return True
|
|
return False
|
|
|
|
async def stop_mining(self) -> bool:
|
|
cfg = await self.get_config()
|
|
cfg.mining_mode = MiningModeConfig.sleep()
|
|
await self.send_config(cfg)
|
|
return True
|
|
|
|
async def resume_mining(self) -> bool:
|
|
cfg = await self.get_config()
|
|
cfg.mining_mode = MiningModeConfig.normal()
|
|
await self.send_config(cfg)
|
|
return True
|
|
|
|
async def _get_wattage(self, rpc_stats: dict = None) -> Optional[int]:
|
|
if not rpc_stats:
|
|
try:
|
|
rpc_stats = await self.rpc.stats()
|
|
except APIError:
|
|
pass
|
|
|
|
if rpc_stats:
|
|
boards = rpc_stats.get("STATS")
|
|
try:
|
|
wattage_raw = boards[1]["chain_power"]
|
|
except (KeyError, IndexError):
|
|
pass
|
|
else:
|
|
# parse wattage position out of raw data
|
|
return round(float(wattage_raw.split(" ")[0]))
|
|
|
|
async def _get_hostname(self, web_get_system_info: dict = None) -> Optional[str]:
|
|
if web_get_system_info is None:
|
|
try:
|
|
web_get_system_info = await self.web.get_system_info()
|
|
except APIError:
|
|
pass
|
|
|
|
if web_get_system_info is not None:
|
|
try:
|
|
return web_get_system_info["hostname"]
|
|
except KeyError:
|
|
pass
|
|
|
|
async def _get_mac(self, web_get_system_info: dict = None) -> Optional[str]:
|
|
if web_get_system_info is None:
|
|
try:
|
|
web_get_system_info = await self.web.get_system_info()
|
|
except APIError:
|
|
pass
|
|
|
|
if web_get_system_info is not None:
|
|
try:
|
|
return web_get_system_info["macaddr"]
|
|
except KeyError:
|
|
pass
|
|
|
|
try:
|
|
data = await self.web.get_network_info()
|
|
if data:
|
|
return data["macaddr"]
|
|
except KeyError:
|
|
pass
|
|
|
|
async def _get_fault_light(
|
|
self, web_get_blink_status: dict = None
|
|
) -> Optional[bool]:
|
|
if self.light:
|
|
return self.light
|
|
|
|
if web_get_blink_status is None:
|
|
try:
|
|
web_get_blink_status = await self.web.get_blink_status()
|
|
except APIError:
|
|
pass
|
|
|
|
if web_get_blink_status is not None:
|
|
try:
|
|
self.light = web_get_blink_status["blink"]
|
|
except KeyError:
|
|
pass
|
|
return self.light
|
|
|
|
async def _is_mining(self, web_get_conf: dict = None) -> Optional[bool]:
|
|
if web_get_conf is None:
|
|
try:
|
|
web_get_conf = await self.web.get_miner_conf()
|
|
except APIError:
|
|
pass
|
|
|
|
if web_get_conf is not None:
|
|
try:
|
|
if str(web_get_conf["bitmain-work-mode"]).isdigit():
|
|
return (
|
|
False if int(web_get_conf["bitmain-work-mode"]) == 1 else True
|
|
)
|
|
return False
|
|
except LookupError:
|
|
pass
|
|
|
|
|
|
HIVEON_OLD_DATA_LOC = DataLocations(
|
|
**{
|
|
str(DataOptions.API_VERSION): DataFunction(
|
|
"_get_api_ver",
|
|
[RPCAPICommand("rpc_version", "version")],
|
|
),
|
|
str(DataOptions.FW_VERSION): DataFunction(
|
|
"_get_fw_ver",
|
|
[RPCAPICommand("rpc_version", "version")],
|
|
),
|
|
str(DataOptions.HASHRATE): DataFunction(
|
|
"_get_hashrate",
|
|
[RPCAPICommand("rpc_summary", "summary")],
|
|
),
|
|
str(DataOptions.EXPECTED_HASHRATE): DataFunction(
|
|
"_get_expected_hashrate",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.HASHBOARDS): DataFunction(
|
|
"_get_hashboards",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.FANS): DataFunction(
|
|
"_get_fans",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.UPTIME): DataFunction(
|
|
"_get_uptime",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
str(DataOptions.POOLS): DataFunction(
|
|
"_get_pools",
|
|
[RPCAPICommand("rpc_pools", "pools")],
|
|
),
|
|
str(DataOptions.WATTAGE): DataFunction(
|
|
"_get_wattage",
|
|
[RPCAPICommand("rpc_stats", "stats")],
|
|
),
|
|
}
|
|
)
|
|
|
|
|
|
class HiveonOld(HiveonFirmware, BMMiner):
|
|
data_locations = HIVEON_OLD_DATA_LOC
|
|
|
|
async def _get_wattage(self, rpc_stats: dict = None) -> Optional[int]:
|
|
if not rpc_stats:
|
|
try:
|
|
rpc_stats = await self.rpc.stats()
|
|
except APIError:
|
|
pass
|
|
|
|
if rpc_stats:
|
|
boards = rpc_stats.get("STATS")
|
|
try:
|
|
wattage_raw = boards[1]["chain_power"]
|
|
except (KeyError, IndexError):
|
|
pass
|
|
else:
|
|
# parse wattage position out of raw data
|
|
return round(float(wattage_raw.split(" ")[0]))
|