OmniNX/sphaira
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Clean up python (#212)

Browse Source 
* Python cleanup
This commit is contained in:
AndrewSpangler
2025-08-30 19:48:37 -07:00
committed by GitHub
parent f0bdc01156
commit 0a8bc01870
3 changed files with 162 additions and 102 deletions
Download Patch File Download Diff File Expand all files Collapse all files

76
tools/usb_export.py
View File

@@ -1,34 +1,40 @@
import crc32c
import sys
import os
from pathlib import Path
from usb_common import *
from usb_common import Usb, USB_ENUM as UE, SPLASH
def get_file_name(usb: Usb, name_length: int) -> str:
return bytes(usb.read(name_length)).decode('utf-8')
def create_file_folder(root: Path, file_path: Path) -> Path:
# todo: check if it already exists.
full_path = Path(root + "/" + file_path)
full_path.parent.mkdir(exist_ok=True, parents=True)
print("created folder")
def create_file_folder(root: os.PathLike, target: os.PathLike) -> bool, os.PathLike:
"""
Creates a recursive folder structure at a given location
Returns a boolean indicating if it already exists and the absolute path
"""
path = os.abspath(os.path.join(root, target))
parent = os.path.dirname(path)
if not (os.path.exists(parent)):
os.makedirs(path)
print(f"Created folder {path}")
else:
print(f"Parent folder already exists {path}")
exists_already = os.path.exists(path)
return exists_already, path
return full_path
def wait_for_input(usb: Usb, path: Path) -> None:
def wait_for_input(usb: Usb, path: os.PathLike) -> None:
print("now waiting for intput\n")
with open(path, "wb") as file:
print("opened file {}".format(path))
with open(path, "wb") as f:
print(f"Opened file {path}")
while True:
[off, size, crc32c_want] = usb.get_send_data_header()
# todo: this isn't needed really.
usb.send_result(RESULT_OK)
usb.send_result(UE.RESULT_OK)
# check if we should finish now.
if (off == 0 and size == 0):
if off == 0 and size == 0:
break
# read the buffer and calculate the crc32c.
@@ -36,33 +42,31 @@ def wait_for_input(usb: Usb, path: Path) -> None:
crc32c_got = crc32c.crc32c(buf)
# validate the crc32c matches.
if (crc32c_want != crc32c_got):
usb.send_result(RESULT_ERROR)
if crc32c_want != crc32c_got:
usb.send_result(UE.RESULT_ERROR)
continue
try:
file.seek(off)
file.write(buf)
usb.send_result(RESULT_OK)
f.seek(off)
f.write(buf)
usb.send_result(UE.RESULT_OK)
except BlockingIOError as e:
print("Error: failed to write: {} at: {} size: {} error: {}".format(e.filename, off, size, str(e)))
usb.send_result(RESULT_ERROR)
usb.send_result(UE.RESULT_ERROR)
if __name__ == '__main__':
print("hello world")
print(SPLASH)
# check which mode the user has selected.
args = len(sys.argv)
if (args != 2):
print("pass the folder path")
if not len(args) == 2:
print("Pass root path as argument.")
sys.exit(1)
root_path = sys.argv[1]
if (not os.path.isdir(root_path)):
raise ValueError('must be a dir!')
raise ValueError('')
usb: Usb = Usb()
usb = Usb()
try:
# get usb endpoints.
@@ -72,21 +76,21 @@ if __name__ == '__main__':
while True:
[cmd, arg3, arg4] = usb.get_send_header()
if (cmd == CMD_QUIT):
usb.send_result(RESULT_OK)
if (cmd == UE.CMD_QUIT):
usb.send_result(UE.RESULT_OK)
break
elif (cmd == CMD_EXPORT):
usb.send_result(RESULT_OK)
elif (cmd == UE.CMD_EXPORT):
usb.send_result(UE.RESULT_OK)
# todo: handle and return errors here.
file_name = get_file_name(usb, arg3)
full_path = create_file_folder(root_path, file_name)
usb.send_result(RESULT_OK)
exists_already, full_path = create_file_folder(root_path, file_name)
usb.send_result(UE.RESULT_OK)
wait_for_input(usb, full_path)
else:
usb.send_result(RESULT_ERROR)
usb.send_result(UE.RESULT_ERROR)
break
except Exception as inst:
print("An exception occurred " + str(inst))
except Exception as e:
print(f"An exception occurred - {e} ")