File size: 1,037 Bytes
7feac49 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
# Copyright (c) Alibaba, Inc. and its affiliates.
import asyncio
import sys
from asyncio.subprocess import PIPE, STDOUT
async def run_and_get_log(*args, timeout=None):
process = await asyncio.create_subprocess_exec(*args, stdout=PIPE, stderr=STDOUT)
lines = []
while True:
try:
line = await asyncio.wait_for(process.stdout.readline(), timeout)
except asyncio.TimeoutError:
break
else:
if not line:
break
else:
lines.append(str(line))
return process, lines
def run_command_in_subprocess(*args, timeout):
if sys.platform == 'win32':
loop = asyncio.ProactorEventLoop()
asyncio.set_event_loop(loop)
else:
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
process, lines = loop.run_until_complete(run_and_get_log(*args, timeout=timeout))
return (loop, process), lines
def close_loop(handler):
loop, process = handler
process.kill()
loop.close()
|