adafruit-mirror/circup
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

allow mock version_info in DiskBackend

Browse source
This commit is contained in:
foamyguy 2023-12-11 16:52:13 -06:00
parent 01f0a2d046
commit 56c1ddab31
1 changed files with 26 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
circup/backends.py
View file

@ -454,11 +454,12 @@ class DiskBackend(Backend):
Backend for interacting with a device via USB Workflow
"""
def __init__(self, device_location, logger):
def __init__(self, device_location, logger, version_info=None):
super().__init__(logger)
self.LIB_DIR_PATH = "lib"
self.device_location = device_location
self.library_path = self.device_location + "/" + self.LIB_DIR_PATH
self.version_info = version_info
def get_circuitpython_version(self):
"""
@ -475,28 +476,31 @@ class DiskBackend(Backend):
:return: A tuple with the version string for CircuitPython and the board ID string.
"""
try:
with open(
os.path.join(self.device_location, "boot_out.txt"),
"r",
encoding="utf-8",
) as boot:
version_line = boot.readline()
circuit_python = version_line.split(";")[0].split(" ")[-3]
board_line = boot.readline()
if board_line.startswith("Board ID:"):
board_id = board_line[9:].strip()
else:
board_id = ""
except FileNotFoundError:
click.secho(
"Missing file boot_out.txt on the device: wrong path or drive corrupted.",
fg="red",
)
self.logger.error("boot_out.txt not found.")
sys.exit(1)
if not self.version_info:
try:
with open(
os.path.join(self.device_location, "boot_out.txt"),
"r",
encoding="utf-8",
) as boot:
version_line = boot.readline()
circuit_python = version_line.split(";")[0].split(" ")[-3]
board_line = boot.readline()
if board_line.startswith("Board ID:"):
board_id = board_line[9:].strip()
else:
board_id = ""
except FileNotFoundError:
click.secho(
"Missing file boot_out.txt on the device: wrong path or drive corrupted.",
fg="red",
)
self.logger.error("boot_out.txt not found.")
sys.exit(1)
return circuit_python, board_id
return circuit_python, board_id
else:
return self.version_info
def _get_modules(self, device_lib_path):
"""