87 lines
2.6 KiB
Python
87 lines
2.6 KiB
Python
import ipaddress
|
|
import os
|
|
import re
|
|
import xlsxwriter
|
|
|
|
import aiofiles
|
|
|
|
from tools.bad_board_util.func.ui import update_ui_with_data
|
|
from tools.bad_board_util.layout import window
|
|
from tools.bad_board_util.func.decorators import disable_buttons
|
|
|
|
|
|
@disable_buttons
|
|
async def save_report(file_location):
|
|
data = []
|
|
workbook = xlsxwriter.Workbook(file_location)
|
|
sheet = workbook.add_worksheet()
|
|
for line in window["ip_table"].Values:
|
|
data.append([line[0], line[1], line[2], line[3], line[5], line[7]])
|
|
|
|
data = sorted(data, reverse=True, key=lambda x: x[2])
|
|
|
|
headers = [
|
|
"IP",
|
|
"Miner Model",
|
|
"Total Chip Count",
|
|
"Left Board Chips",
|
|
"Center Board Chips",
|
|
"Right Board Chips",
|
|
]
|
|
print(data)
|
|
row = 0
|
|
col = 0
|
|
for item in headers:
|
|
sheet.write(row, col, item)
|
|
col += 1
|
|
|
|
row = 1
|
|
for line in data:
|
|
col = 0
|
|
for point in line:
|
|
sheet.write(row, col, point)
|
|
col += 1
|
|
row += 1
|
|
|
|
workbook.close()
|
|
|
|
|
|
async def import_iplist(file_location):
|
|
await update_ui_with_data("status", "Importing")
|
|
if not os.path.exists(file_location):
|
|
return
|
|
else:
|
|
ip_list = []
|
|
async with aiofiles.open(file_location, mode="r") as file:
|
|
async for line in file:
|
|
ips = [
|
|
x.group()
|
|
for x in re.finditer(
|
|
"^((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]?)",
|
|
line,
|
|
)
|
|
]
|
|
for ip in ips:
|
|
if ip not in ip_list:
|
|
ip_list.append(ipaddress.ip_address(ip))
|
|
ip_list.sort()
|
|
window["ip_table"].update([[str(ip), "", "", "", ""] for ip in ip_list])
|
|
await update_ui_with_data("ip_count", str(len(ip_list)))
|
|
await update_ui_with_data("status", "")
|
|
|
|
|
|
async def export_iplist(file_location, ip_list_selected):
|
|
await update_ui_with_data("status", "Exporting")
|
|
if not os.path.exists(file_location):
|
|
return
|
|
else:
|
|
if ip_list_selected is not None and not ip_list_selected == []:
|
|
async with aiofiles.open(file_location, mode="w") as file:
|
|
for item in ip_list_selected:
|
|
await file.write(str(item) + "\n")
|
|
else:
|
|
async with aiofiles.open(file_location, mode="w") as file:
|
|
for item in window["ip_table"].Values:
|
|
await file.write(str(item[0]) + "\n")
|
|
await update_ui_with_data("status", "")
|