168 lines
7.0 KiB
Python
168 lines
7.0 KiB
Python
import asyncio
|
|
import datetime
|
|
|
|
import websockets.exceptions
|
|
from fastapi import WebSocket, WebSocketDisconnect, APIRouter
|
|
|
|
from miners.miner_factory import MinerFactory
|
|
from tools.web_monitor._settings.func import get_current_settings
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.websocket("/{miner_ip}/ws")
|
|
async def miner_websocket(websocket: WebSocket, miner_ip):
|
|
await websocket.accept()
|
|
settings = get_current_settings()
|
|
miner_identify_timeout = settings["miner_identify_timeout"]
|
|
miner_data_timeout = settings["miner_data_timeout"]
|
|
|
|
try:
|
|
while True:
|
|
try:
|
|
cur_miner = await asyncio.wait_for(
|
|
MinerFactory().get_miner(str(miner_ip)), miner_identify_timeout
|
|
)
|
|
|
|
data = await asyncio.wait_for(
|
|
cur_miner.api.multicommand(
|
|
"summary", "fans", "stats", "devs", "temps"
|
|
),
|
|
miner_data_timeout,
|
|
)
|
|
miner_model = await cur_miner.get_model()
|
|
|
|
miner_summary = None
|
|
miner_fans = None
|
|
if "summary" in data.keys():
|
|
miner_summary = data["summary"][0]
|
|
elif "SUMMARY" in data.keys():
|
|
miner_summary = data
|
|
miner_fans = {"FANS": []}
|
|
for item in ["Fan Speed In", "Fan Speed Out"]:
|
|
if item in miner_summary["SUMMARY"][0].keys():
|
|
miner_fans["FANS"].append(
|
|
{"RPM": miner_summary["SUMMARY"][0][item]}
|
|
)
|
|
|
|
if "fans" in data.keys():
|
|
miner_fans = data["fans"][0]
|
|
|
|
miner_temp_list = []
|
|
if "temps" in data.keys():
|
|
miner_temps = data["temps"][0]
|
|
for board in miner_temps["TEMPS"]:
|
|
if board["Chip"] is not None and not board["Chip"] == 0.0:
|
|
miner_temp_list.append(board["Chip"])
|
|
|
|
if "devs" in data.keys() and not len(miner_temp_list) > 0:
|
|
if not data["devs"][0].get("DEVS") == []:
|
|
if "Chip Temp Avg" in data["devs"][0]["DEVS"][0].keys():
|
|
for board in data["devs"][0]["DEVS"]:
|
|
if (
|
|
board["Chip Temp Avg"] is not None
|
|
and not board["Chip Temp Avg"] == 0.0
|
|
):
|
|
miner_temp_list.append(board["Chip Temp Avg"])
|
|
|
|
if "stats" in data.keys() and not len(miner_temp_list) > 0:
|
|
if not data["stats"][0]["STATS"] == []:
|
|
for temp in ["temp2", "temp1", "temp3"]:
|
|
if temp in data["stats"][0]["STATS"][1].keys():
|
|
if (
|
|
data["stats"][0]["STATS"][1][temp] is not None
|
|
and not data["stats"][0]["STATS"][1][temp] == 0.0
|
|
):
|
|
miner_temp_list.append(
|
|
data["stats"][0]["STATS"][1][temp]
|
|
)
|
|
data["stats"][0]["STATS"][0].keys()
|
|
if any(
|
|
"MM ID" in string
|
|
for string in data["stats"][0]["STATS"][0].keys()
|
|
):
|
|
temp_all = []
|
|
for key in [
|
|
string
|
|
for string in data["stats"][0]["STATS"][0].keys()
|
|
if "MM ID" in string
|
|
]:
|
|
for value in [
|
|
string
|
|
for string in data["stats"][0]["STATS"][0][key].split(
|
|
" "
|
|
)
|
|
if "TMax" in string
|
|
]:
|
|
temp_all.append(
|
|
int(value.split("[")[1].replace("]", ""))
|
|
)
|
|
miner_temp_list.append(round(sum(temp_all) / len(temp_all)))
|
|
|
|
if "stats" in data.keys() and not miner_fans:
|
|
miner_stats = data["stats"][0]
|
|
miner_fans = {"FANS": []}
|
|
for item in ["fan1", "fan2", "fan3", "fan4"]:
|
|
if item in miner_stats["STATS"][1].keys():
|
|
miner_fans["FANS"].append(
|
|
{"RPM": miner_stats["STATS"][1][item]}
|
|
)
|
|
|
|
if miner_summary:
|
|
if "MHS av" in miner_summary["SUMMARY"][0].keys():
|
|
hashrate = float(
|
|
format(
|
|
round(
|
|
miner_summary["SUMMARY"][0]["MHS av"] / 1000000, 2
|
|
),
|
|
".2f",
|
|
)
|
|
)
|
|
elif "GHS av" in miner_summary["SUMMARY"][0].keys():
|
|
hashrate = float(
|
|
format(
|
|
round(miner_summary["SUMMARY"][0]["GHS av"] / 1000, 2),
|
|
".2f",
|
|
)
|
|
)
|
|
else:
|
|
hashrate = 0
|
|
else:
|
|
hashrate = 0
|
|
|
|
fan_speeds = []
|
|
if miner_fans:
|
|
for fan in miner_fans["FANS"]:
|
|
fan_speeds.append(fan["RPM"])
|
|
|
|
while len(fan_speeds) < 4:
|
|
fan_speeds.append(0)
|
|
|
|
if len(miner_temp_list) == 0:
|
|
miner_temp_list = [0]
|
|
|
|
data = {
|
|
"hashrate": hashrate,
|
|
"fans": fan_speeds,
|
|
"temp": round(sum(miner_temp_list) / len(miner_temp_list), 2),
|
|
"datetime": datetime.datetime.now().isoformat(),
|
|
"model": miner_model,
|
|
}
|
|
print(data)
|
|
await websocket.send_json(data)
|
|
await asyncio.sleep(settings["graph_data_sleep_time"])
|
|
except asyncio.exceptions.TimeoutError:
|
|
data = {"error": "The miner is not responding."}
|
|
await websocket.send_json(data)
|
|
await asyncio.sleep(0.5)
|
|
except KeyError as e:
|
|
print(e)
|
|
data = {"error": "The miner returned unusable/unsupported data."}
|
|
await websocket.send_json(data)
|
|
await asyncio.sleep(0.5)
|
|
except WebSocketDisconnect:
|
|
print("Websocket disconnected.")
|
|
except websockets.exceptions.ConnectionClosedOK:
|
|
pass
|