mirror of
https://github.com/akuker/RASCSI.git
synced 2024-06-10 02:29:33 +00:00
62e287c96d
- Remove the Size option from the Drives page, since we offer custom sizes with Drive profiles now in the Create Image form - Fetch size from drive props data structure rather than the web form - Remove a range of redundant / obvious help text to reduce UI clutter and emphasize the important help text - Remove the Size column from the CD-ROM drives table since it's always N/A - Merge the two Logging related sections, and the two Create Image related sections (semantically associated)
312 lines
11 KiB
Python
312 lines
11 KiB
Python
"""
|
|
Module for RaSCSI Web Interface utility methods
|
|
"""
|
|
|
|
import logging
|
|
from grp import getgrall
|
|
from os import path
|
|
from pathlib import Path
|
|
|
|
from flask import request, make_response
|
|
from flask_babel import _
|
|
from werkzeug.utils import secure_filename
|
|
|
|
from rascsi.sys_cmds import SysCmds
|
|
|
|
def get_valid_scsi_ids(devices, reserved_ids):
|
|
"""
|
|
Takes a list of (dict)s devices, and list of (int)s reserved_ids.
|
|
Returns:
|
|
- (list) of (int)s valid_ids, which are the SCSI ids that are not reserved
|
|
- (list) of (int)s occupied_ids, which are the SCSI ids were a device is attached
|
|
- (int) recommended_id, which is the id that the Web UI should default to recommend
|
|
"""
|
|
occupied_ids = []
|
|
for device in devices:
|
|
occupied_ids.append(device["id"])
|
|
|
|
unoccupied_ids = [i for i in list(range(8)) if i not in reserved_ids + occupied_ids]
|
|
unoccupied_ids.sort()
|
|
valid_ids = [i for i in list(range(8)) if i not in reserved_ids]
|
|
valid_ids.sort(reverse=True)
|
|
|
|
if unoccupied_ids:
|
|
recommended_id = unoccupied_ids[-1]
|
|
else:
|
|
if occupied_ids:
|
|
recommended_id = occupied_ids[0]
|
|
else:
|
|
recommended_id = 0
|
|
|
|
return {
|
|
"valid_ids": valid_ids,
|
|
"occupied_ids": occupied_ids,
|
|
"recommended_id": recommended_id,
|
|
}
|
|
|
|
|
|
def sort_and_format_devices(devices):
|
|
"""
|
|
Takes a (list) of (dict)s devices and returns a (list) of (dict)s.
|
|
Sorts by SCSI ID acending (0 to 7).
|
|
For SCSI IDs where no device is attached, inject a (dict) with placeholder text.
|
|
"""
|
|
occupied_ids = []
|
|
formatted_devices = []
|
|
for device in devices:
|
|
occupied_ids.append(device["id"])
|
|
device["device_name"] = get_device_name(device["device_type"])
|
|
formatted_devices.append(device)
|
|
|
|
# Add placeholder data for non-occupied IDs
|
|
for scsi_id in range(8):
|
|
if scsi_id not in occupied_ids:
|
|
formatted_devices.append(
|
|
{
|
|
"id": scsi_id,
|
|
"unit": "-",
|
|
"device_name": "-",
|
|
"status": "-",
|
|
"file": "-",
|
|
"product": "-",
|
|
}
|
|
)
|
|
|
|
return formatted_devices
|
|
|
|
|
|
def map_device_types_and_names(device_types):
|
|
"""
|
|
Takes a (dict) corresponding to the data structure returned by RaCtlCmds.get_device_types()
|
|
Returns a (dict) of device_type:device_name mappings of localized device names
|
|
"""
|
|
for device in device_types.keys():
|
|
device_types[device]["name"] = get_device_name(device)
|
|
|
|
return device_types
|
|
|
|
|
|
# pylint: disable=too-many-return-statements
|
|
def get_device_name(device_type):
|
|
"""
|
|
Takes a four letter device acronym (str) device_type.
|
|
Returns the human-readable name for the device type.
|
|
"""
|
|
if device_type == "SCHD":
|
|
return _("Hard Disk Drive")
|
|
if device_type == "SCRM":
|
|
return _("Removable Disk Drive")
|
|
if device_type == "SCMO":
|
|
return _("Magneto-Optical Drive")
|
|
if device_type == "SCCD":
|
|
return _("CD/DVD Drive")
|
|
if device_type == "SCBR":
|
|
return _("Host Bridge")
|
|
if device_type == "SCDP":
|
|
return _("Ethernet Adapter")
|
|
if device_type == "SCLP":
|
|
return _("Printer")
|
|
if device_type == "SCHS":
|
|
return _("Host Services")
|
|
return device_type
|
|
|
|
|
|
def map_image_file_descriptions(file_suffixes):
|
|
"""
|
|
Takes a (list) of (str) file suffixes for images file types.
|
|
Returns a (dict) with file suffix and description pairs, both (str)
|
|
"""
|
|
supported_image_types = {}
|
|
for suffix in file_suffixes:
|
|
supported_image_types[suffix] = get_image_description(suffix)
|
|
|
|
return supported_image_types
|
|
|
|
|
|
# pylint: disable=too-many-return-statements
|
|
def get_image_description(file_suffix):
|
|
"""
|
|
Takes a three char file suffix (str) file_suffix.
|
|
Returns the help text description for said file suffix.
|
|
"""
|
|
if file_suffix == "hds":
|
|
return _("Hard Disk Image (Generic)")
|
|
if file_suffix == "hda":
|
|
return _("Hard Disk Image (Apple)")
|
|
if file_suffix == "hdn":
|
|
return _("Hard Disk Image (NEC)")
|
|
if file_suffix == "hd1":
|
|
return _("Hard Disk Image (SCSI-1)")
|
|
if file_suffix == "hdr":
|
|
return _("Removable Disk Image")
|
|
if file_suffix == "mos":
|
|
return _("Magneto-Optical Disk Image")
|
|
return file_suffix
|
|
|
|
|
|
def format_drive_properties(drive_properties):
|
|
"""
|
|
Takes a (dict) with structured drive properties data
|
|
Returns a (dict) with the formatted properties, one (list) per device type
|
|
"""
|
|
hd_conf = []
|
|
cd_conf = []
|
|
rm_conf = []
|
|
mo_conf = []
|
|
FORMAT_FILTER = "{:,.2f}"
|
|
|
|
for device in drive_properties:
|
|
# Add fallback device names, since other code relies on this data for display
|
|
if not device["name"]:
|
|
if device["product"]:
|
|
device["name"] = device["product"]
|
|
else:
|
|
device["name"] = "Unknown Device"
|
|
if device["device_type"] == "SCHD":
|
|
device["secure_name"] = secure_filename(device["name"])
|
|
device["size_mb"] = FORMAT_FILTER.format(device["size"] / 1024 / 1024)
|
|
hd_conf.append(device)
|
|
elif device["device_type"] == "SCCD":
|
|
device["size_mb"] = _("N/A")
|
|
cd_conf.append(device)
|
|
elif device["device_type"] == "SCRM":
|
|
device["secure_name"] = secure_filename(device["name"])
|
|
device["size_mb"] = FORMAT_FILTER.format(device["size"] / 1024 / 1024)
|
|
rm_conf.append(device)
|
|
elif device["device_type"] == "SCMO":
|
|
device["secure_name"] = secure_filename(device["name"])
|
|
device["size_mb"] = FORMAT_FILTER.format(device["size"] / 1024 / 1024)
|
|
mo_conf.append(device)
|
|
|
|
return {
|
|
"hd_conf": hd_conf,
|
|
"cd_conf": cd_conf,
|
|
"rm_conf": rm_conf,
|
|
"mo_conf": mo_conf,
|
|
}
|
|
|
|
def get_properties_by_drive_name(drives, drive_name):
|
|
"""
|
|
Takes (list) of (dict) drives, and (str) drive_name
|
|
Returns (dict) with the collection of drive properties that matches drive_name
|
|
"""
|
|
drives.sort(key=lambda item: item.get("name"))
|
|
|
|
drive_props = None
|
|
prev_drive = {"name": ""}
|
|
for drive in drives:
|
|
# TODO: Make this check into an integration test
|
|
if "name" not in drive:
|
|
logging.warning(
|
|
"Device without a name exists in the drive properties database. This is a bug."
|
|
)
|
|
break
|
|
# TODO: Make this check into an integration test
|
|
if drive["name"] == prev_drive["name"]:
|
|
logging.warning(
|
|
"Device with duplicate name \"%s\" in drive properties database. This is a bug.",
|
|
drive["name"],
|
|
)
|
|
prev_drive = drive
|
|
if drive["name"] == drive_name:
|
|
drive_props = drive
|
|
|
|
return {
|
|
"file_type": drive_props["file_type"],
|
|
"vendor": drive_props["vendor"],
|
|
"product": drive_props["product"],
|
|
"revision": drive_props["revision"],
|
|
"block_size": drive_props["block_size"],
|
|
"size": drive_props["size"],
|
|
}
|
|
|
|
def auth_active(group):
|
|
"""
|
|
Inspects if the group defined in (str) group exists on the system.
|
|
If it exists, tell the webapp to enable authentication.
|
|
Returns a (dict) with (bool) status and (str) msg
|
|
"""
|
|
groups = [g.gr_name for g in getgrall()]
|
|
if group in groups:
|
|
return {
|
|
"status": True,
|
|
"msg": _("You must log in to use this function"),
|
|
}
|
|
return {"status": False, "msg": ""}
|
|
|
|
|
|
def is_bridge_configured(interface):
|
|
"""
|
|
Takes (str) interface of a network device being attached.
|
|
Returns a (dict) with (bool) status and (str) msg
|
|
"""
|
|
status = True
|
|
return_msg = ""
|
|
sys_cmd = SysCmds()
|
|
if interface.startswith("wlan"):
|
|
if not sys_cmd.introspect_file("/etc/sysctl.conf", r"^net\.ipv4\.ip_forward=1$"):
|
|
status = False
|
|
return_msg = _("Configure IPv4 forwarding before using a wireless network device.")
|
|
elif not Path("/etc/iptables/rules.v4").is_file():
|
|
status = False
|
|
return_msg = _("Configure NAT before using a wireless network device.")
|
|
else:
|
|
if not sys_cmd.introspect_file(
|
|
"/etc/dhcpcd.conf",
|
|
r"^denyinterfaces " + interface + r"$",
|
|
):
|
|
status = False
|
|
return_msg = _("Configure the network bridge before using a wired network device.")
|
|
elif not Path("/etc/network/interfaces.d/rascsi_bridge").is_file():
|
|
status = False
|
|
return_msg = _("Configure the network bridge before using a wired network device.")
|
|
|
|
return {"status": status, "msg": return_msg + f" ({interface})"}
|
|
|
|
|
|
def upload_with_dropzonejs(image_dir):
|
|
"""
|
|
Takes (str) image_dir which is the path to the image dir to store files.
|
|
Opens a stream to transfer a file via the embedded dropzonejs library.
|
|
"""
|
|
log = logging.getLogger("pydrop")
|
|
file_object = request.files["file"]
|
|
file_name = secure_filename(file_object.filename)
|
|
|
|
save_path = path.join(image_dir, file_name)
|
|
current_chunk = int(request.form['dzchunkindex'])
|
|
|
|
# Makes sure not to overwrite an existing file,
|
|
# but continues writing to a file transfer in progress
|
|
if path.exists(save_path) and current_chunk == 0:
|
|
return make_response(_("The file already exists!"), 400)
|
|
|
|
try:
|
|
with open(save_path, "ab") as save:
|
|
save.seek(int(request.form["dzchunkbyteoffset"]))
|
|
save.write(file_object.stream.read())
|
|
except OSError:
|
|
log.exception("Could not write to file")
|
|
return make_response(_("Unable to write the file to disk!"), 500)
|
|
|
|
total_chunks = int(request.form["dztotalchunkcount"])
|
|
|
|
if current_chunk + 1 == total_chunks:
|
|
# Validate the resulting file size after writing the last chunk
|
|
if path.getsize(save_path) != int(request.form["dztotalfilesize"]):
|
|
log.error(
|
|
"Finished transferring %s, "
|
|
"but it has a size mismatch with the original file. "
|
|
"Got %s but we expected %s.",
|
|
file_object.filename,
|
|
path.getsize(save_path),
|
|
request.form['dztotalfilesize'],
|
|
)
|
|
return make_response(_("Transferred file corrupted!"), 500)
|
|
|
|
log.info("File %s has been uploaded successfully", file_object.filename)
|
|
log.debug("Chunk %s of %s for file %s completed.",
|
|
current_chunk + 1, total_chunks, file_object.filename)
|
|
|
|
return make_response(_("File upload successful!"), 200)
|