Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-02 07:18:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

use higher level API selectors instead of select (#3899)

Browse Source
This commit is contained in:
Saugat Pachhai
2020-05-29 09:25:07 +05:45
committed by GitHub
parent 777a9f80ae
commit bb50fb8664
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
dvc/remote/ssh/connection.py
+7 -3
View File
@@ -237,11 +237,15 @@ class SSHConnection:
or channel.recv_ready()
or channel.recv_stderr_ready()
):
import select
import selectors
selector = selectors.DefaultSelector()
selector.register(stdout.channel, selectors.EVENT_READ)
got_chunk = False
readq, _, _ = select.select([stdout.channel], [], [], self.timeout)
for c in readq:
events = selector.select(self.timeout)
for key, _ in events:
c = key.fileobj
if c.recv_ready():
stdout_chunks.append(stdout.channel.recv(len(c.in_buffer)))
got_chunk = True