93 lines
2.2 KiB
Python
93 lines
2.2 KiB
Python
"""WizardKit: Launch Snappy Driver Installer Origin"""
|
|
# vim: sts=2 sw=2 ts=2
|
|
|
|
import os
|
|
import sys
|
|
|
|
os.chdir(os.path.dirname(os.path.realpath(__file__)))
|
|
sys.path.append(os.getcwd())
|
|
import wk # noqa: E402
|
|
from wk.cfg.net import SDIO_SERVER # noqa: E402
|
|
|
|
# STATIC VARIABLES
|
|
MOUNT_EXCEPTIONS = (
|
|
RuntimeError,
|
|
wk.exe.subprocess.CalledProcessError,
|
|
)
|
|
SDIO_LOCAL_PATH = wk.kit.tools.get_tool_path("SDIO", "SDIO")
|
|
SDIO_REMOTE_PATH = wk.io.get_path_obj(
|
|
(
|
|
fr'\\{SDIO_SERVER["Address"]}\{SDIO_SERVER["Share"]}\{SDIO_SERVER["Path"]}'
|
|
fr'\SDIO{"64" if wk.os.win.ARCH == "64" else ""}.exe'
|
|
),
|
|
resolve=False,
|
|
)
|
|
|
|
# Functions
|
|
def try_again():
|
|
"""Ask to try again or quit."""
|
|
if wk.std.ask(' Try again?'):
|
|
return True
|
|
if not wk.std.ask(' Use local version?'):
|
|
wk.std.abort()
|
|
return False
|
|
|
|
|
|
def use_network_sdio():
|
|
"""Try to mount SDIO server."""
|
|
use_network = False
|
|
def _mount_server():
|
|
print('Connecting to server... (Press CTRL+c to use local copy)')
|
|
return wk.net.mount_network_share(SDIO_SERVER, read_write=False)
|
|
|
|
# Bail early
|
|
if not SDIO_SERVER['Address']:
|
|
return use_network
|
|
|
|
# Main loop
|
|
while True:
|
|
try:
|
|
proc = _mount_server()
|
|
except KeyboardInterrupt:
|
|
break
|
|
except MOUNT_EXCEPTIONS as err:
|
|
wk.std.print_error(f' {err}')
|
|
if not try_again():
|
|
break
|
|
else:
|
|
if proc.returncode == 0:
|
|
# Network copy available
|
|
use_network = True
|
|
break
|
|
|
|
# Failed to mount
|
|
wk.std.print_error(' Failed to mount server')
|
|
if not try_again():
|
|
break
|
|
|
|
# Done
|
|
return use_network
|
|
|
|
|
|
if __name__ == '__main__':
|
|
wk.std.set_title(
|
|
f'{wk.cfg.main.KIT_NAME_FULL}: Snappy Driver Installer Origin Launcher',
|
|
)
|
|
log_dir = wk.log.format_log_path(tool=True).parent
|
|
USE_NETWORK = False
|
|
|
|
# Try to mount server
|
|
try:
|
|
USE_NETWORK = use_network_sdio()
|
|
except KeyboardInterrupt:
|
|
wk.std.abort()
|
|
|
|
# Run SDIO
|
|
EXE_PATH = SDIO_LOCAL_PATH
|
|
if USE_NETWORK:
|
|
EXE_PATH = SDIO_REMOTE_PATH
|
|
print('Using network copy!')
|
|
else:
|
|
print('Using local copy!')
|
|
cmd = [EXE_PATH, '-log_dir', log_dir]
|
|
wk.exe.run_program(cmd, check=False, cwd=EXE_PATH.parent)
|