Macintosh-HW/RASCSI
1
0
Fork 0
mirror of https://github.com/akuker/RASCSI.git synced 2024-12-25 03:32:16 +00:00
Code Issues Projects Releases Wiki Activity

Use wget to download files instead of urllib; reducing code duplication and fixing a bug

Browse Source
This commit is contained in:
Daniel Markstedt 2021-10-26 21:50:00 -07:00
parent e5f58bb967
commit 46698c7b05
2 changed files with 16 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/web/file_cmds.py
View File

@ -188,29 +188,23 @@ def download_file_to_iso(scsi_id, url):
Takes int scsi_id and str url Takes int scsi_id and str url
Returns dict with boolean status and str msg Returns dict with boolean status and str msg
""" """
import urllib.request from time import time
import urllib.error as error
import time
from subprocess import run from subprocess import run
server_info = get_server_info() server_info = get_server_info()
file_name = url.split("/")[-1] file_name = url.split("/")[-1]
tmp_ts = int(time.time()) tmp_ts = int(time())
tmp_dir = "/tmp/" + str(tmp_ts) + "/" tmp_dir = "/tmp/" + str(tmp_ts) + "/"
os.mkdir(tmp_dir) os.mkdir(tmp_dir)
tmp_full_path = tmp_dir + file_name tmp_full_path = tmp_dir + file_name
iso_filename = f"{server_info['image_dir']}/{file_name}.iso" iso_filename = f"{server_info['image_dir']}/{file_name}.iso"
try: wget_proc = download_to_dir(url, tmp_dir)
urllib.request.urlretrieve(url, tmp_full_path)
except (error.URLError, error.HTTPError, error.ContentTooShortError, FileNotFoundError) as e: if wget_proc["status"] == False:
logging.error(str(e)) return {"status": False, "msg": wget_proc["msg"]}
return {"status": False, "msg": str(e)}
except:
return {"status": False, "msg": "Unknown error occurred."}
# iso_filename = make_cd(tmp_full_path, None, None) # not working yet
iso_proc = run( iso_proc = run(
["genisoimage", "-hfs", "-o", iso_filename, tmp_full_path], capture_output=True ["genisoimage", "-hfs", "-o", iso_filename, tmp_full_path], capture_output=True
) )
@ -225,20 +219,17 @@ def download_to_dir(url, save_dir):
Takes str url, str save_dir Takes str url, str save_dir
Returns dict with boolean status and str msg Returns dict with boolean status and str msg
""" """
import urllib.request from subprocess import run
import urllib.error as error
file_name = url.split("/")[-1] wget_proc = run(
full_path = f"{save_dir}/{file_name}" ["wget", url, "-P", save_dir], capture_output=True
)
if wget_proc.returncode != 0:
stderr = wget_proc.stderr.decode("utf-8")
logging.warning(f"Downloading failed: {stderr}")
return {"status": False, "msg": stderr}
try: return {"status": True, "msg": f"{url} downloaded to {save_dir}"}
urllib.request.urlretrieve(url, full_path)
return {"status": True, "msg": "Downloaded the URL"}
except (error.URLError, error.HTTPError, error.ContentTooShortError, FileNotFoundError) as e:
logging.error(str(e))
return {"status": False, "msg": str(e)}
except:
return {"status": False, "msg": "Unknown error occurred."}
def write_config(file_name): def write_config(file_name):

2
src/web/web.py
View File

@ -502,7 +502,7 @@ def download_to_iso():
flash(process["msg"], "error") flash(process["msg"], "error")
return redirect(url_for("index")) return redirect(url_for("index"))
process_attach = attach_image(scsi_id, type="SCCD", image=process["file_name"]) process_attach = attach_image(scsi_id, device_type="SCCD", image=process["file_name"])
if process_attach["status"] == True: if process_attach["status"] == True:
flash(f"Attached to SCSI ID {scsi_id}") flash(f"Attached to SCSI ID {scsi_id}")
return redirect(url_for("index")) return redirect(url_for("index"))