CAT_TOM/pyasic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added tables and basic scanning

Browse Source
This commit is contained in:
UpstreamData
2022-05-04 14:44:19 -06:00
parent b636860ecb
commit 184ada417f
6 changed files with 382 additions and 274 deletions
Download Patch File Download Diff File Expand all files Collapse all files

295
tools/cfg_util/cfg_util_qt/__init__.py
View File

File diff suppressed because one or more lines are too long

3
tools/cfg_util/cfg_util_qt/imgs.py Normal file
View File

File diff suppressed because one or more lines are too long

289
tools/cfg_util/cfg_util_qt/layout.py Normal file
View File

@@ -0,0 +1,289 @@
import PySimpleGUI as sg
from .imgs import WINDOW_ICON, LIGHT
TABLE_HEADERS = {
"SCAN": [
"IP",
"Model",
"Hostname",
"Hashrate",
"Temperature",
"Pool User",
"Wattage",
],
"CMD": ["IP", "Model", "Command Output"],
"POOLS": [
"IP",
"Model",
"Split",
"Pool 1",
"Pool 1 User",
"Pool 2",
"Pool 2 User",
],
"CONFIG": ["IP", "Model", "Pool 1 User"],
}
TABLE_KEYS = {
"table": ["scan_table", "pools_table", "cfg_table"],
"tree": ["cmd_table"],
}
BUTTON_KEYS = [
"btn_scan",
"btn_cmd",
"scan_all",
"scan_refresh",
"scan_web",
"cmd_all",
"cmd_light",
"cmd_reboot",
"cmd_backend",
"pools_all",
"pools_refresh",
"pools_web",
"cfg_import",
"cfg_config",
"cfg_generate",
"cfg_all",
"cfg_web",
]
TABLE_HEIGHT = 27
IMAGE_COL_WIDTH = 6
IP_COL_WIDTH = 17
MODEL_COL_WIDTH = 15
HOST_COL_WIDTH = 15
HASHRATE_COL_WIDTH = 12
TEMP_COL_WIDTH = 12
USER_COL_WIDTH = 31
WATTAGE_COL_WIDTH = 8
SPLIT_COL_WIDTH = 10
SCAN_COL_WIDTHS = [
IP_COL_WIDTH,
MODEL_COL_WIDTH,
HOST_COL_WIDTH,
HASHRATE_COL_WIDTH,
TEMP_COL_WIDTH,
USER_COL_WIDTH,
WATTAGE_COL_WIDTH,
]
TABLE_TOTAL_WIDTH = sum(SCAN_COL_WIDTHS)
async def update_prog_bar(count: int, max: int = None):
bar = window["progress_bar"]
bar.update_bar(count, max=max)
if not hasattr(bar, "maxlen"):
if not max:
max = 100
bar.maxlen = max
percent_done = 100 * (count / bar.maxlen)
window["progress_percent"].Update(f"{round(percent_done, 2)} %")
if percent_done == 100:
window["progress_percent"].Update("")
def get_scan_layout():
scan_layout = [
[
sg.Text("Scan IP"),
sg.InputText(key="scan_ip", size=(31, 1)),
sg.Button("Scan", key="btn_scan"),
sg.Push(),
sg.Button(
"Scan",
disabled=True,
button_color=("black", "white smoke"),
disabled_button_color=("black", "white smoke"),
),
],
[
sg.Button("ALL", key="scan_all"),
sg.Button("REFRESH DATA", key="scan_refresh"),
sg.Button("OPEN IN WEB", key="scan_web"),
],
[
sg.Table(
values=[[]],
headings=[heading for heading in TABLE_HEADERS["SCAN"]],
auto_size_columns=False,
max_col_width=15,
justification="center",
key="scan_table",
col_widths=SCAN_COL_WIDTHS,
background_color="white",
text_color="black",
size=(TABLE_TOTAL_WIDTH, TABLE_HEIGHT),
expand_x=True,
enable_click_events=True,
)
],
]
return scan_layout
def get_command_layout():
data = sg.TreeData()
data.insert("", 0, "", ["192.168.168.168", "S19j Pro (BOS)", ""], icon=LIGHT)
data.insert("", 1, "", ["192.168.168.168", "S19j Pro (BOS)", ""], icon=LIGHT)
col_widths = [
IP_COL_WIDTH,
MODEL_COL_WIDTH,
TABLE_TOTAL_WIDTH - (IP_COL_WIDTH + MODEL_COL_WIDTH + IMAGE_COL_WIDTH + 4),
]
command_layout = [
[
sg.Text("Custom Command"),
sg.InputText(key="cmd_txt", expand_x=True),
sg.Button("Send Command", key="btn_cmd"),
sg.Push(),
sg.Button(
"Command",
disabled=True,
button_color=("black", "white smoke"),
disabled_button_color=("black", "white smoke"),
),
],
[
sg.Button("ALL", key="cmd_all"),
sg.Button("LIGHT", key="cmd_light"),
sg.Button("REBOOT", key="cmd_reboot"),
sg.Button("RESTART BACKEND", key="cmd_backend"),
],
[
sg.Tree(
data,
headings=[heading for heading in TABLE_HEADERS["CMD"]],
auto_size_columns=False,
max_col_width=15,
justification="center",
key="cmd_table",
col_widths=col_widths,
background_color="white",
text_color="black",
expand_x=True,
expand_y=True,
col0_heading="Light",
col0_width=IMAGE_COL_WIDTH,
)
],
]
return command_layout
def get_pools_layout():
pool_col_width = int(
(TABLE_TOTAL_WIDTH - (IP_COL_WIDTH + MODEL_COL_WIDTH + SPLIT_COL_WIDTH)) / 4
)
col_widths = [
IP_COL_WIDTH,
MODEL_COL_WIDTH,
SPLIT_COL_WIDTH,
pool_col_width,
pool_col_width,
pool_col_width,
pool_col_width,
]
pools_layout = [
[
sg.Push(),
sg.Button(
"Pools",
disabled=True,
button_color=("black", "white smoke"),
disabled_button_color=("black", "white smoke"),
),
],
[
sg.Button("ALL", key="pools_all"),
sg.Button("REFRESH DATA", key="pools_refresh"),
sg.Button("OPEN IN WEB", key="pools_web"),
],
[
sg.Table(
values=[],
headings=[heading for heading in TABLE_HEADERS["POOLS"]],
auto_size_columns=False,
max_col_width=15,
justification="center",
key="pools_table",
background_color="white",
text_color="black",
col_widths=col_widths,
size=(0, TABLE_HEIGHT),
expand_x=True,
enable_click_events=True,
)
],
]
return pools_layout
def get_config_layout():
config_layout = [
[
sg.Button("IMPORT", key="cfg_import"),
sg.Button("CONFIG", key="cfg_config"),
sg.Button("GENERATE", key="cfg_generate"),
sg.Push(),
sg.Button(
"Configure",
disabled=True,
button_color=("black", "white smoke"),
disabled_button_color=("black", "white smoke"),
),
],
[
sg.Button("ALL", key="cfg_all"),
sg.Button("OPEN IN WEB", key="cfg_web"),
sg.Push(),
sg.Checkbox("Append IP to Username", key="cfg_append_ip"),
],
[
sg.Table(
values=[],
headings=[heading for heading in TABLE_HEADERS["CONFIG"]],
auto_size_columns=False,
max_col_width=15,
justification="center",
key="cfg_table",
background_color="white",
text_color="black",
col_widths=[
IP_COL_WIDTH,
MODEL_COL_WIDTH,
TABLE_TOTAL_WIDTH - ((2 * 40) - 4),
],
size=(0, TABLE_HEIGHT),
expand_x=True,
enable_click_events=True,
),
sg.Multiline(size=(40, TABLE_HEIGHT + 1)),
],
]
return config_layout
layout = [
[
sg.Text("", size=(10, 1), key="status"),
sg.ProgressBar(
max_value=100, size_px=(0, 20), expand_x=True, key="progress_bar"
),
sg.Text("", size=(10, 1), key="progress_percent"),
],
[
sg.TabGroup(
[
[sg.Tab("Scan", get_scan_layout())],
[sg.Tab("Pools", get_pools_layout())],
[sg.Tab("Configure", get_config_layout())],
[sg.Tab("Command", get_command_layout())],
]
)
],
]
window = sg.Window("Upstream Config Util", layout, icon=WINDOW_ICON)

34
tools/cfg_util/cfg_util_qt/scan/__init__.py Normal file
View File

@@ -0,0 +1,34 @@
from tools.cfg_util.cfg_util_qt.tables import clear_tables, update_tables
from tools.cfg_util.cfg_util_qt.layout import window, update_prog_bar
from network import MinerNetwork
from miners.miner_factory import MinerFactory
async def scan_miners(network: MinerNetwork):
clear_tables()
scan_generator = network.scan_network_generator()
MinerFactory().clear_cached_miners()
progress_bar_len = 0
network_size = len(network)
await update_prog_bar(progress_bar_len, max=(2 * network_size))
scanned_miners = []
async for miner in scan_generator:
if miner:
scanned_miners.append(miner)
progress_bar_len += 1
await update_prog_bar(progress_bar_len)
progress_bar_len += network_size - len(scanned_miners)
await update_prog_bar(progress_bar_len)
get_miner_genenerator = MinerFactory().get_miner_generator(scanned_miners)
resolved_miners = []
async for found_miner in get_miner_genenerator:
resolved_miners.append(found_miner)
resolved_miners.sort(key=lambda x: x.ip)
update_tables([{"IP": str(miner.ip)} for miner in resolved_miners])
progress_bar_len += 1
await update_prog_bar(progress_bar_len)

35
tools/cfg_util/cfg_util_qt/tables.py Normal file
View File

@@ -0,0 +1,35 @@
from tools.cfg_util.cfg_util_qt.layout import TABLE_KEYS, TABLE_HEADERS, window
from tools.cfg_util.cfg_util_qt.imgs import LIGHT
from PySimpleGUI import TreeData
def clear_tables():
for table in TABLE_KEYS["table"]:
window[table].update([])
for tree in TABLE_KEYS["tree"]:
window[tree].update(TreeData())
def update_tables(data: dict):
tables = {
"SCAN": [["" for _ in TABLE_HEADERS["SCAN"]] for _ in data],
"CMD": [["" for _ in TABLE_HEADERS["CMD"]] for _ in data],
"POOLS": [["" for _ in TABLE_HEADERS["POOLS"]] for _ in data],
"CONFIG": [["" for _ in TABLE_HEADERS["CONFIG"]] for _ in data],
}
for data_idx, item in enumerate(data):
for key in item.keys():
for table in TABLE_HEADERS.keys():
for idx, header in enumerate(TABLE_HEADERS[table]):
if key == header:
tables[table][data_idx][idx] = item[key]
window["scan_table"].update(tables["SCAN"])
window["pools_table"].update(tables["POOLS"])
window["cfg_table"].update(tables["CONFIG"])
tree_data = TreeData()
for item in tables["CMD"]:
tree_data.insert("", "", "", item, icon=LIGHT)
window["cmd_table"].update(tree_data)

0
tools/cfg_util/cfg_util_qt/ui.py Normal file
View File