Python:封装允许执行命令有超时的类

menny

　　封装允许执行命令有超时的类


#!/usr/bin/env python
import os;
import sys;
import time;
import fcntl;
import select;
import signal;
import commands;
import subprocess;
class CRunCmd:
def __init__(self):
pass;
def __AsyncRecv(self, fdSock, nMaxRead = 1024 * 8):
if not fdSock or fdSock.closed:
return (False, '')
#set fd non-block
nFlags = fcntl.fcntl(fdSock, fcntl.F_GETFL);
fcntl.fcntl(fdSock, fcntl.F_SETFL, nFlags | os.O_NONBLOCK)
bRet = False
strRead = ''
try:
#check can be read
if not select.select([fdSock], [], [], 0)[0]:
return (True, '')
strRead = fdSock.read(nMaxRead);
#if read empty, then close it
if len(strRead) <= 0:
fdSock.close();
bRet = True;
except:
bRet = False;
#reset fd
if not fdSock.closed:
fcntl.fcntl(fdSock, fcntl.F_SETFL, nFlags)
return (bRet, strRead)
def Run(self, lsCmd, nTimeOut = 0, nIntervalTime = 1):
oProc = subprocess.Popen(lsCmd, stdout = subprocess.PIPE, stderr = subprocess.PIPE)
strOut = ''
strErr = '';
nStartTime = time.time()
while True:
print(oProc.poll());
if None != oProc.poll():
break;
#sleep nIntervalTime
time.sleep(nIntervalTime)
bRet, strBuf1 = self.__AsyncRecv(oProc.stdout);
bRet, strBuf2 = self.__AsyncRecv(oProc.stderr);
if len(strBuf1) > 0:
strOut += strBuf1;
if len(strBuf2) > 0:
strErr += strBuf2;
if (nTimeOut > 0) and (time.time() - nStartTime) > nTimeOut:
print("time out...");
break;
#get last buff
bRet, strBuf1 = self.__AsyncRecv(oProc.stdout);
bRet, strBuf2 = self.__AsyncRecv(oProc.stderr);
if len(strBuf1) > 0:
strOut += strBuf1;
if len(strBuf2) > 0:
strErr += strBuf2;
#if not finish, so timeout
if None == oProc.poll():
self.KillAll(oProc.pid)
return (oProc.returncode, strOut, strErr)
def KillAll(self, nKillPid, nKillSignal = signal.SIGKILL):
print("kill pid:%s" %nKillPid)
nRet, strOutput = commands.getstatusoutput('ps -A -o pid,ppid');
if 0 != nRet:
return (False, strOutput);
mapPid = {};
#make all ppid & pid map
for strLine in strOutput.split('\n'):
lsPid = strLine.strip().split();
if 2 != len(lsPid):
continue;
strPid = lsPid[0];
strPPid = lsPid[1];
if strPPid in mapPid.keys():
mapPid[strPPid].append(strPid);
else:
mapPid[strPPid] = [strPid];
#get all kill pid list
lsAllKillPid = [str(nKillPid)];
lsToKillPid = [str(nKillPid)];
while True:
if len(lsToKillPid) <= 0:
break;
lsChild = []
for strPid in lsToKillPid:
if strPid in mapPid.keys():
lsAllKillPid.extend(mapPid[strPid]);
lsChild.extend(mapPid[strPid]);
print("[%s]append:%s" %(strPid, mapPid[strPid]));
lsToKillPid = lsChild;
print("kill pid list\n%s" %lsAllKillPid)
#kill all process
for strPid in reversed(lsAllKillPid):
try:
print("kill %s" %(strPid))
#os.kill(int(strPid), nKillSignal)
except:
pass
return (True, '')


　　 

　　/tmp/a.py


#!/usr/bin/env python
import time;
import os;
import sys;
def Fork():
os.fork();
print("pid:%s" %os.getpid());

if '__main__' == __name__:
for i in xrange(0, int(sys.argv[1])):
print("[%s] befor fork" %os.getpid());
Fork();
print("[%s] after fork" %os.getpid());
time.sleep(1);
print("[%s] start sleep" %os.getpid());
time.sleep(10);
print("[%s] end of process..." %os.getpid());


　　 

　　


　　#test code



if "__main__" == __name__:
oCmd = CRunCmd();
nRet, strOut, strErr = oCmd.Run(['python', '/tmp/a.py', '10'], 5)
print("ret:%s" %nRet);
print("stdout:%s" %strOut);
print("stderr:%s" %strErr);


　　 

　　


　　


　　执行结果：

　　$python w_p.py

None

None

None

None

None

time out...

kill pid:29443

[29443]append:['29444', '29446', '29450', '29457', '29474']

[29444]append:['29445', '29448', '29454', '29463']

[29446]append:['29449', '29456', '29465']

[29450]append:['29458', '29471']

[29457]append:['29473']

[29445]append:['29447', '29452', '29462']

[29448]append:['29453', '29461']

[29454]append:['29470']

[29449]append:['29455', '29468']

[29456]append:['29467']

[29458]append:['29472']

[29447]append:['29451', '29466']

[29452]append:['29459']

[29453]append:['29460']

[29455]append:['29469']

[29451]append:['29464']

kill pid list

['29443', '29444', '29446', '29450', '29457', '29474', '29445', '29448', '29454', '29463', '29449', '29456', '29465', '29458', '29471', '29473', '29447', '29452', '29462', '29453', '29461', '29470', '29455', '29468', '29467', '29472', '29451', '29466', '29459', '29460', '29469', '29464']

kill 29464

kill 29469

kill 29460

kill 29459

kill 29466

kill 29451

kill 29472

kill 29467

kill 29468

kill 29455

kill 29470

kill 29461

kill 29453

kill 29462

kill 29452

kill 29447

kill 29473

kill 29471

kill 29458

kill 29465

kill 29456

kill 29449

kill 29463

kill 29454

kill 29448

kill 29445

kill 29474

kill 29457

kill 29450

kill 29446

kill 29444

kill 29443

ret:None

stdout:

stderr: