Add check for device connected

[dotfiles.git] / invoke / esptool / tasks.py

diff --git a/invoke/esptool/tasks.py b/invoke/esptool/tasks.py
index 2a807670972e9e83464f64475df59ee058d7ba4c..43df54eaa318ba40446290c601001449da905d0f 100644 (file)
--- a/invoke/esptool/tasks.py
+++ b/invoke/esptool/tasks.py
@@ -9,6 +9,8 @@ TODAY = dt.now().strftime('%Y.%m.%d')
 @task
 def help(c):
     '''
+    On using this tasks file
+
     - Set the Cardputer Adv power to 'off' state
     - Plugging in to USB will cause it to boot
     - Hold 'Go' while booting
@@ -24,12 +26,14 @@ def help(c):
 
 @task
 def groups(c):
+    '''Check that the current user is a member of dialout'''
     result = c.run(f'groups {os.getlogin()}')
     assert 'dialout' in result.stdout
     print('ok')
 
 @task
 def usbdiff(c):
+    '''Use lsusb to try to find the new device.'''
     input('Un-plug in the device...')
     c.run('lsusb > /tmp/before.txt')
     input('Plug in the device...')
@@ -42,6 +46,13 @@ def usbdiff(c):
 
 DEV='/dev/ttyACM0'
 
+@task
+def check_connected(c):
+    dev = 'Espressif'
+    result = c.run(f'lsusb | grep {dev}')
+    assert dev in result.stdout
+    print('[ok]')
+
 @task
 def listusb(c):
     c.run('ls /dev/ttyACM*')