console: Make baud an optional parameter

Make the baud rate an optional parameter to the console.py tool.  When
not present, it will default to 250000.

Signed-off-by: Kevin O'Connor <kevin@koconnor.net>

1 files changed, 10 insertions, 3 deletions

diff --git a/klippy/console.py b/klippy/console.py
index ab501d9a..bbf5377d 100755
--- a/klippy/console.py
+++ b/klippy/console.py
@@ -202,11 +202,18 @@ class KeyboardReader:
         self.data = kbdlines[-1]
 
 def main():
-    usage = "%prog [options] <serialdevice> <baud>"
+    usage = "%prog [options] <serialdevice>"
     opts = optparse.OptionParser(usage)
+    opts.add_option("-b", "--baud", type="int", dest="baud", help="baud rate")
     options, args = opts.parse_args()
-    serialport, baud = args
-    baud = int(baud)
+    if len(args) != 1:
+        opts.error("Incorrect number of arguments")
+    serialport = args[0]
+
+    baud = options.baud
+    if baud is None and not (serialport.startswith("/dev/rpmsg_")
+                             or serialport.startswith("/tmp/")):
+        baud = 250000
 
     logging.basicConfig(level=logging.DEBUG)
     r = reactor.Reactor()