Changeset 602:bb62c7392d44 for usb_console

Timestamp:

06/11/08 11:23:37 (7 months ago)

Author:

Maxime Petazzoni <maxime.petazzoni@…>

Branch:

default

Message:

Improve the user interface by providing a readline-powered
prompt, with command history.

Files:

• usb_console/rcmd_frontend.py (modified) (2 diffs)

usb_console/rcmd_frontend.py

@@ -8 +8 @@
 # the terms of the GNU Public License (GPL) version 2.
 
+import atexit
+import code
 import nxt
+import os
+import readline
 import struct
 import sys
+import time
 from nxt.lowlevel import get_device
 
 NXOS_INTERFACE = 0
 
-def prompt():
-    print "rcmd>",
-    return raw_input()
+class RcmdConsole(code.InteractiveConsole):
+
+  def __init__(self, locals=None, filename="<console>",
+               histfile=os.path.expanduser("~/.nxos-rcmd.hist")):
+    code.InteractiveConsole.__init__(self)
+    self.init_history(histfile)
+
+  def init_history(self, histfile):
+    readline.parse_and_bind("tab: complete")
+    if hasattr(readline, "read_history_file"):
+      try:
+        readline.read_history_file(histfile)
+      except IOError:
+        pass
+      atexit.register(self.save_history, histfile)
+
+  def save_history(self, histfile):
+    readline.write_history_file(histfile)
+
+
+  def set_brick(self, brick):
+    self.brick = brick
+
+  def push(self, line):
+    if line == 'quit' or line == 'exit':
+      print "Use Ctrl-D (i.e. EOF) to exit"
+      return False
+
+    self.brick.write(line)
+    if line == 'end':
+      sys.exit(0)
+
+    return False
 
 def main():
@@ -29 +64 @@
     print "ok."
 
-    while True:
-      command = prompt()
-      if not brick.write(command + '\n') or command == 'end':
-        return
+    prompt = RcmdConsole()
+    prompt.set_brick(brick)
+    prompt.interact('Remote robot command console.')
+
+    print "Closing link...",
+    brick.write('end')
+    print "done."
 
 if __name__ == "__main__":