[tor-commits] [stem/master] Support SETEVENTS with tor-prompt's run argument

[atagar at torproject.org]

commit 9302fd23ab19cf2c294a376392525c88c3ef0436
Author: Damian Johnson <atagar at torproject.org>
Date:   Wed Apr 19 10:43:28 2017 -0700

    Support SETEVENTS with tor-prompt's run argument
    
    Adding special handling for SETEVENTS so running...
    
      tor-prompt --run "SETEVENTS BW DEBUG"
    
    ... will listen for events until the user presses a key. These events can be
    piped to a file. Something like this was requested by anrelanos on...
    
      https://trac.torproject.org/projects/tor/ticket/21590
---
 stem/interpreter/__init__.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/stem/interpreter/__init__.py b/stem/interpreter/__init__.py
index 8cdbb3c..033d819 100644
--- a/stem/interpreter/__init__.py
+++ b/stem/interpreter/__init__.py
@@ -126,7 +126,28 @@ def main():
     interpreter = stem.interpreter.commands.ControlInterpreter(controller)
 
     if args.run_cmd:
-      interpreter.run_command(args.run_cmd, print_response = True)
+      if args.run_cmd.upper().startswith('SETEVENTS '):
+        # TODO: we can use a lambda here when dropping python 2.x support, but
+        # until then print's status as a keyword prevents it from being used in
+        # lambdas
+
+        def handle_event(event_message):
+          print(format(str(event_message), *STANDARD_OUTPUT))
+
+        controller._handle_event = handle_event
+
+        if sys.stdout.isatty():
+          events = args.run_cmd.upper().split(' ', 1)[1]
+          print(format('Listening to %s events. Press any key to quit.\n' % events, *HEADER_BOLD_OUTPUT))
+
+        controller.msg(args.run_cmd)
+
+        try:
+          raw_input()
+        except (KeyboardInterrupt, stem.SocketClosed) as exc:
+          pass
+      else:
+        interpreter.run_command(args.run_cmd, print_response = True)
     elif args.run_path:
       try:
         for line in open(args.run_path).readlines():