tmux/crispyconductor.tmux-copy-toolkit
1
0
Fork 0
mirror of https://github.com/CrispyConductor/tmux-copy-toolkit.git synced 2026-09-10 04:06:14 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

fixed the issue that Popen.communicate hangs waiting stdout of a command

Browse source
This commit is contained in:
Jaehyung Choi 2021-11-12 03:19:32 -08:00
parent 7072e6c58e
commit 626c408c7d
1 changed files with 3 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
copytk.py
View file

@ -106,28 +106,20 @@ def runtmux(args, one=False, lines=False, noblanklines=False, sendstdin=None):
return dlines[0] if len(dlines) > 0 else ''
return dlines if lines else data
def runshellcommand(command, one=False, lines=False, noblanklines=False, sendstdin=None, raisenonzero=True):
def runshellcommand(command, sendstdin=None, raisenonzero=True):
log('run shell command: ' + command, time=True)
with subprocess.Popen(
command,
shell=True,
executable='/bin/bash',
stdin=subprocess.PIPE if sendstdin != None else subprocess.DEVNULL,
stdout=subprocess.PIPE
stdout=subprocess.DEVNULL
) as proc:
if sendstdin != None and isinstance(sendstdin, str):
sendstdin = bytearray(sendstdin, 'utf8')
recvstdout, _ = proc.communicate(input=sendstdin)
proc.communicate(input=sendstdin)
if proc.returncode != 0 and raisenonzero:
raise Exception(f'Command {command} returned exit code {proc.returncode}')
data = recvstdout.decode('utf8')
if one or lines: # return list of lines
dlines = data.split('\n')
if not one and noblanklines:
dlines = [ l for l in dlines if len(l) > 0 ]
if one: # single-line
return dlines[0] if len(dlines) > 0 else ''
return dlines if lines else data
def runtmuxmulti(argsets):