sanchayanmaity/gem5
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Some cosmetic changes to MyPOpen

Browse source 
util/pbs/pbs.py:
    More tweaks that I forgot

--HG--
extra : convert_revision : 7298f91b80bc7d8d946be93fc622e5f9f6e155f9
This commit is contained in:
Nathan Binkert 2005-02-10 00:02:51 -05:00
parent 8efd7d9063
commit 200246d1ca
1 changed files with 12 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
util/pbs/pbs.py
View file

@ -30,7 +30,7 @@ import os, popen2, re, sys
class MyPOpen(object):
def __init__(self, cmd, input = None, output = None, bufsize = -1):
self.sts = -1
self.status = -1
if input is None:
p2c_read, p2c_write = os.pipe()
@ -64,17 +64,9 @@ class MyPOpen(object):
self.pid = os.fork()
if self.pid == 0:
os.dup2(p2c_read, 0)
os.dup2(c2p_write, 1)
os.dup2(c2p_write, 2)
if isinstance(cmd, basestring):
cmd = ['/bin/sh', '-c', cmd]
if False:
for i in range(3, MAXFD):
try:
os.close(i)
except OSError:
pass
os.dup2(p2c_read, sys.stdin.fileno())
os.dup2(c2p_write, sys.stdout.fileno())
os.dup2(c2p_write, sys.stderr.fileno())
try:
os.execvp(cmd[0], cmd)
finally:
@ -84,18 +76,18 @@ class MyPOpen(object):
os.close(c2p_write)
def poll(self):
if self.sts < 0:
pid, sts = os.waitpid(self.pid, os.WNOHANG)
if self.status < 0:
pid, status = os.waitpid(self.pid, os.WNOHANG)
if pid == self.pid:
self.sts = sts
return self.sts
self.status = status
return self.status
def wait(self):
if self.sts < 0:
pid, sts = os.waitpid(self.pid, 0)
if self.status < 0:
pid, status = os.waitpid(self.pid, 0)
if pid == self.pid:
self.sts = sts
return self.sts
self.status = status
return self.status
class qsub:
def __init__(self):