ovs.stream: Simplify logic in Stream.wait().

Suggested-by: Reid Price <reid@nicira.com>

diff --git a/python/ovs/stream.py b/python/ovs/stream.py
index 5954c3148a76768e02483b71d49b553e1dd70db3..82d4557e953035636311ca14c0d777c9ae5684f2 100644 (file)
--- a/python/ovs/stream.py
+++ b/python/ovs/stream.py
@@ -206,10 +206,10 @@ class Stream(object):
 
         if self.state == Stream.__S_CONNECTING:
             wait = Stream.W_CONNECT
-        if wait in (Stream.W_CONNECT, Stream.W_SEND):
-            poller.fd_wait(self.socket, select.POLLOUT)
-        else:
+        if wait == Stream.W_RECV:
             poller.fd_wait(self.socket, select.POLLIN)
+        else:
+            poller.fd_wait(self.socket, select.POLLOUT)
 
     def connect_wait(self, poller):
         self.wait(poller, Stream.W_CONNECT)