[FIX] Socket-mode port has not been added to the argument list. Fix: #34

wexpect/__main__.py

@@ -1,6 +1,7 @@
 import argparse
 import sys
 import logging
+import traceback
 
 import wexpect.console_reader as console_reader
 import wexpect.wexpect_util as wexpect_util
@@ -20,6 +21,7 @@ def str2bool(v):
         raise argparse.ArgumentTypeError('Boolean value expected.')
 
 def main():
+    try:
         parser = argparse.ArgumentParser(description='Wexpect: executable automation for Windows.')
 
         parser.add_argument('--console_reader_class', type=str,
@@ -35,7 +37,17 @@ def main():
                             default=16000)
         parser.add_argument('--local_echo', type=str, help='Echo sent characters', default=True)
         parser.add_argument('--interact', type=str, help='Show console window', default=False)
+        parser.add_argument('--port', type=int, help=
+                            "If the console reader class is SpawnSocket, this option specifies the "
+                            "socket's port.", default=False)
+
+        try:
             args = parser.parse_args()
+        except SystemExit:
+            logger.error('Unexpected exception.')
+            logger.info(traceback.format_exc())
+            raise
+
         logger.info(f'Starter arguments: {args}')
 
         if args.console_reader_class == 'ConsoleReaderSocket':
@@ -46,12 +58,17 @@ def main():
         command = wexpect_util.join_args(args.command)
 
         cons = conole_reader_class(
-        path=command, host_pid=args.host_pid, codepage=args.codepage,
+            path=command, host_pid=args.host_pid, codepage=args.codepage, port=args.port,
             window_size_x=args.window_size_x, window_size_y=args.window_size_y,
             buffer_size_x=args.buffer_size_x, buffer_size_y=args.buffer_size_y,
             local_echo=str2bool(args.local_echo), interact=str2bool(args.interact))
 
         sys.exit(cons.child_exitstatus)
 
+    except Exception as e:
+        logger.error('Unexpected exception.')
+        logger.info(traceback.format_exc())
+        raise
+
 if __name__ == "__main__":
     main()