diff --git a/bin/ad2-firmwareupload b/bin/ad2-firmwareupload index 614fa53..bf5f96c 100755 --- a/bin/ad2-firmwareupload +++ b/bin/ad2-firmwareupload @@ -1,25 +1,25 @@ #!/usr/bin/env python import sys, time -import pyad2.ad2 +import pyad2 def handle_firmware(stage): - if stage == pyad2.ad2.util.Firmware.STAGE_START: + if stage == pyad2.util.Firmware.STAGE_START: handle_firmware.wait_tick = 0 handle_firmware.upload_tick = 0 - elif stage == pyad2.ad2.util.Firmware.STAGE_WAITING: + elif stage == pyad2.util.Firmware.STAGE_WAITING: if handle_firmware.wait_tick == 0: sys.stdout.write('Waiting for device.') handle_firmware.wait_tick += 1 sys.stdout.write('.') sys.stdout.flush() - elif stage == pyad2.ad2.util.Firmware.STAGE_BOOT: + elif stage == pyad2.util.Firmware.STAGE_BOOT: if handle_firmware.wait_tick > 0: print "" print "Rebooting device.." - elif stage == pyad2.ad2.util.Firmware.STAGE_LOAD: + elif stage == pyad2.util.Firmware.STAGE_LOAD: print 'Waiting for boot loader..' - elif stage == pyad2.ad2.util.Firmware.STAGE_UPLOADING: + elif stage == pyad2.util.Firmware.STAGE_UPLOADING: if handle_firmware.upload_tick == 0: sys.stdout.write('Uploading firmware.') @@ -28,7 +28,7 @@ def handle_firmware(stage): if handle_firmware.upload_tick % 30 == 0: sys.stdout.write('.') sys.stdout.flush() - elif stage == pyad2.ad2.util.Firmware.STAGE_DONE: + elif stage == pyad2.util.Firmware.STAGE_DONE: print "\r\nDone!" def main(): @@ -45,11 +45,11 @@ def main(): print "Flashing device: {0}\r\nFirmware: {1}".format(device, firmware) - dev = pyad2.ad2.devices.SerialDevice(interface=device) + dev = pyad2.devices.SerialDevice(interface=device) dev.open(baudrate=19200) time.sleep(3) - pyad2.ad2.util.Firmware.upload(dev, firmware, handle_firmware) + pyad2.util.Firmware.upload(dev, firmware, handle_firmware) dev.close() diff --git a/bin/ad2-sslterm b/bin/ad2-sslterm index 5f64f00..ed66971 100755 --- a/bin/ad2-sslterm +++ b/bin/ad2-sslterm @@ -1,6 +1,6 @@ #!/usr/bin/env python -import pyad2.ad2 +import pyad2 import sys, select import termios, tty import time @@ -23,7 +23,7 @@ def main(): try: print "Opening connection to {0}:{1}\r".format(host, port) - dev = pyad2.ad2.devices.SocketDevice(interface=(host, int(port))) + dev = pyad2.devices.SocketDevice(interface=(host, int(port))) dev.ssl = True dev.ssl_certificate = client_cert dev.ssl_key = client_key