aboutsummaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rwxr-xr-xsupport/scripts/reset.py28
1 files changed, 20 insertions, 8 deletions
diff --git a/support/scripts/reset.py b/support/scripts/reset.py
index bfc91e1..67a72c1 100755
--- a/support/scripts/reset.py
+++ b/support/scripts/reset.py
@@ -9,16 +9,28 @@ import sys
import time
from struct import pack
-def unix_get_maple_path(file_prefix):
- """Try to find the device file for the Maple on *nix; assuming
- that it looks like /dev/<file_prefix>* and has the correct ID in
- the /sys tree. If there are multiple possibilities, ask the user
- what to do. If the user chooses not to say, returns None."""
+def unix_get_maple_path(file_prefix, dev_is_maple=lambda dev: True):
+ """Try to find the device file for the Maple on *nix.
+
+ This function works assuming that the device file globs like
+ '/dev/<file_prefix>*'. The caller may pass an additional
+ dev_is_maple predicate if the platform supports additional tests
+ to determine if a device is a Maple.
+
+ If there are multiple possibilities, ask the user what to do. If
+ the user chooses not to say, returns None."""
possible_paths = [os.path.join('/dev', x) for x in os.listdir('/dev') \
- if (x.startswith(file_prefix) and tty_is_maple(x))]
+ if x.startswith(file_prefix) and dev_is_maple(x)]
return choose_path(possible_paths)
-def tty_is_maple(device):
+def linux_get_maple_path(file_prefix='ttyACM'):
+ """Specialized unix_get_maple_path() for Linux.
+
+ Attempts to check that a candidate device has the correct ID in
+ the /sys tree when deciding if it's a Maple."""
+ return unix_get_maple_path(file_prefix, linux_tty_is_maple)
+
+def linux_tty_is_maple(device):
try:
sysfile = open("/sys/class/tty/%s/device/uevent" % device, "r")
text = "".join(sysfile.readlines())
@@ -72,7 +84,7 @@ def choose_among_options(options):
plat_sys = platform.system()
plat_bits = platform.architecture()[0]
if plat_sys == 'Linux':
- maple_path = unix_get_maple_path('ttyACM')
+ maple_path = linux_get_maple_path()
# fall back on /dev/maple if that doesn't work
if maple_path is None:
maple_path = '/dev/maple'