vastly improved scanning in web monitor

This commit is contained in:
UpstreamData
2022-05-18 11:12:38 -06:00
parent c3e285a9ee
commit 8ebcbd3c33
2 changed files with 52 additions and 17 deletions

View File

@@ -4,6 +4,7 @@ from fastapi import WebSocket
from network import MinerNetwork
from tools.web_monitor.func import get_current_miner_list
import ipaddress
from miners.miner_factory import MinerFactory
@@ -16,24 +17,23 @@ async def do_websocket_scan(websocket: WebSocket, network_ip: str):
else:
network = MinerNetwork(network_ip)
miner_generator = network.scan_network_generator()
miners = []
async for miner_ip in miner_generator:
if miner_ip and str(miner_ip) not in cur_miners:
miners.append(miner_ip)
get_miner_generator = MinerFactory().get_miner_generator(miners)
all_miners = []
async for found_miner in get_miner_generator:
all_miners.append(
{"ip": found_miner.ip, "model": await found_miner.get_model()}
)
all_miners.sort(key=lambda x: x["ip"])
send_miners = []
for miner_ip in all_miners:
send_miners.append(
{"ip": str(miner_ip["ip"]), "model": miner_ip["model"]}
async for found_miner in miner_generator:
if found_miner and str(found_miner.ip) not in cur_miners:
# all_miners.append(miner_ip)
# get_miner_generator = MinerFactory().get_miner_generator(miners)
# async for found_miner in get_miner_generator:
all_miners.append(
{"ip": found_miner.ip, "model": await found_miner.get_model()}
)
await websocket.send_json(send_miners)
all_miners.sort(key=lambda x: ipaddress.ip_address(x["ip"]))
send_miners = []
for miner_ip in all_miners:
send_miners.append(
{"ip": str(miner_ip["ip"]), "model": miner_ip["model"]}
)
await websocket.send_json(send_miners)
await websocket.send_text("Done")
except asyncio.CancelledError:
raise