-
Notifications
You must be signed in to change notification settings - Fork 1
/
execr.py
54 lines (49 loc) · 1.54 KB
/
execr.py
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
import sys
import subprocess as sub
from datetime import datetime
import select
logfile = open('main.log', 'w')
def write_log(*arg):
s = ''.join(str(i) for i in arg).strip()
date = str(datetime.now())
for i in s.split('\n'):
logfile.write(date + ': ' + i + '\n')
sys.stdout.write(date + ': ' + i + '\n')
logfile.flush()
def runcmd(cmd, outpip=sub.PIPE, errpip=sub.STDOUT):
"""
runcmd(cmd)
Executes 'cmd' and yield the output
by default it yield strerr also
"""
write_log('cmd: ', cmd)
process = sub.Popen(cmd, stdout=outpip, stderr=errpip)
# for c in iter(lambda: process.stdout.readline(), b''): ## For line by line
for c in iter(lambda: process.stdout.read(4), b''):
write_log(c.decode())
# yield c #.decode() ## uncomment to send string instead.
yield { 'data': c.decode(), 'stream': 'both' }
def stream_cmd(cmd):
outpip = sub.PIPE
errpip = sub.PIPE
write_log('cmd: ', cmd)
process = sub.Popen(cmd, shell=True, stdout=outpip, stderr=errpip)
while True:
reads = [process.stdout.fileno(), process.stderr.fileno()]
ret = select.select(reads, [], [])
for fd in ret[0]:
if fd == process.stdout.fileno():
data = process.stdout.readline()
write_log(data.decode())
yield { 'data': data.decode(), 'stream': 'stdout' }
if fd == process.stderr.fileno():
data = process.stderr.readline()
write_log(data.decode())
yield { 'data': data.decode(), 'stream': 'stderr' }
if process.poll() != None:
print("Poll", process.poll())
return
if __name__ == '__main__':
a = runcmd("./temp")
for i in a:
print(i, end='')