minor code adjustment

This commit is contained in:
Miroslav Stampar 2010-05-21 13:03:57 +00:00
parent 64f2afe585
commit 5a5b31ad53

View File

@ -36,6 +36,8 @@ import posixpath
import subprocess import subprocess
from StringIO import StringIO from StringIO import StringIO
from subprocess import PIPE
from subprocess import Popen as execute
from tempfile import NamedTemporaryFile from tempfile import NamedTemporaryFile
from tempfile import mkstemp from tempfile import mkstemp
from xml.sax import parse from xml.sax import parse
@ -1111,7 +1113,7 @@ def profile(profileOutputFile=None, imageOutputFile=None):
logger.info(infoMsg) logger.info(infoMsg)
graphScriptPath = os.path.join(paths.SQLMAP_EXTRAS_PATH, 'gprof2dot', 'gprof2dot.py') graphScriptPath = os.path.join(paths.SQLMAP_EXTRAS_PATH, 'gprof2dot', 'gprof2dot.py')
stderr = subprocess.Popen('%s %s -f pstats %s | dot -Tpng -o %s' % (sys.executable, graphScriptPath, profileOutputFile, imageOutputFile), shell=True, stderr=subprocess.PIPE).stderr.read() stderr = execute('%s %s -f pstats %s | dot -Tpng -o %s' % (sys.executable, graphScriptPath, profileOutputFile, imageOutputFile), shell=True, stderr=PIPE).stderr.read()
if stderr or not os.path.exists(imageOutputFile): if stderr or not os.path.exists(imageOutputFile):
errMsg = "there was an error while converting ('%s')" % stderr.strip() errMsg = "there was an error while converting ('%s')" % stderr.strip()