108 lines
3.4 KiB
Python
108 lines
3.4 KiB
Python
import ipaddress
|
|
import re
|
|
|
|
from tools.cfg_util.cfg_util_sg.layout import window
|
|
|
|
import pyperclip
|
|
|
|
|
|
def table_select_all():
|
|
window["ip_table"].update(
|
|
select_rows=(
|
|
[row for row in range(len(window["ip_table"].Values))]
|
|
)
|
|
)
|
|
|
|
|
|
def copy_from_table(table):
|
|
selection = table.selection()
|
|
copy_values = []
|
|
for each in selection:
|
|
try:
|
|
value = table.item(each)["values"][0]
|
|
copy_values.append(str(value))
|
|
except:
|
|
pass
|
|
copy_string = "\n".join(copy_values)
|
|
pyperclip.copy(copy_string)
|
|
|
|
|
|
def copy_from_ssh_table(table):
|
|
selection = table.selection()
|
|
copy_values = []
|
|
for each in selection:
|
|
try:
|
|
value = ", ".join(table.item(each)["values"])
|
|
copy_values.append(str(value))
|
|
except:
|
|
pass
|
|
copy_string = "\n".join(copy_values)
|
|
pyperclip.copy(copy_string)
|
|
|
|
|
|
|
|
async def update_ui_with_data(key, message, append=False):
|
|
if append:
|
|
message = window[key].get_text() + message
|
|
window[key].update(message)
|
|
|
|
|
|
async def update_prog_bar(amount):
|
|
window["progress"].Update(amount)
|
|
percent_done = 100 * (amount / window['progress'].maxlen)
|
|
window["progress_percent"].Update(f"{round(percent_done, 2)} %")
|
|
if percent_done == 100:
|
|
window["progress_percent"].Update("")
|
|
|
|
|
|
async def set_progress_bar_len(amount):
|
|
window["progress"].Update(0, max=amount)
|
|
window["progress"].maxlen = amount
|
|
window["progress_percent"].Update("0.0 %")
|
|
|
|
|
|
async def sort_data(index: int or str):
|
|
if window["scan"].Disabled:
|
|
return
|
|
await update_ui_with_data("status", "Sorting Data")
|
|
data_list = window['ip_table'].Values
|
|
table = window["ip_table"].Widget
|
|
all_data = []
|
|
for idx, item in enumerate(data_list):
|
|
all_data.append({"data": item, "tags": table.item(int(idx) + 1)["tags"]})
|
|
|
|
# wattage
|
|
if re.match("[0-9]* W", str(all_data[0]["data"][index])):
|
|
new_list = sorted(all_data, key=lambda x: int(x["data"][index].replace(" W", "")))
|
|
if all_data == new_list:
|
|
new_list = sorted(all_data, reverse=True, key=lambda x: int(x["data"][index].replace(" W", "")))
|
|
|
|
# hashrate
|
|
elif re.match("[0-9]*\.?[0-9]* TH\/s", str(all_data[0]["data"][index])):
|
|
new_list = sorted(all_data, key=lambda x: float(x["data"][index].replace(" TH/s", "")))
|
|
if all_data == new_list:
|
|
new_list = sorted(all_data, reverse=True, key=lambda x: float(x["data"][index].replace(" TH/s", "")))
|
|
|
|
# ip addresses
|
|
elif re.match("^((25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.){3}(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)",
|
|
str(all_data[0]["data"][index])):
|
|
new_list = sorted(all_data, key=lambda x: ipaddress.ip_address(x["data"][index]))
|
|
if all_data == new_list:
|
|
new_list = sorted(all_data, reverse=True, key=lambda x: ipaddress.ip_address(x["data"][index]))
|
|
|
|
# everything else, hostname, temp, and user
|
|
else:
|
|
new_list = sorted(all_data, key=lambda x: x["data"][index])
|
|
if all_data == new_list:
|
|
new_list = sorted(all_data, reverse=True, key=lambda x: x["data"][index])
|
|
|
|
new_data = []
|
|
for item in new_list:
|
|
new_data.append(item["data"])
|
|
|
|
await update_ui_with_data("ip_table", new_data)
|
|
for idx, item in enumerate(new_list):
|
|
table.item(idx + 1, tags=item["tags"])
|
|
|
|
await update_ui_with_data("status", "")
|