71 lines
1.7 KiB
Python
71 lines
1.7 KiB
Python
import sys
|
|
import commands
|
|
import linecache
|
|
|
|
import time
|
|
|
|
last_mem = [0]
|
|
|
|
def traceit_memory(frame, event, arg):
|
|
if event == "line":
|
|
mem = int(" " + commands.getoutput(
|
|
"ps -eo cmd,rss | grep urk_trace.py | grep -v grep"
|
|
).split(" ")[-1])
|
|
|
|
if mem > last_mem[0]:
|
|
last_mem[0] = mem
|
|
|
|
mem = str(mem)
|
|
|
|
filename = frame.f_globals["__file__"]
|
|
|
|
if filename.endswith(".pyc") or filename.endswith(".pyo"):
|
|
filename = filename[:-1]
|
|
|
|
name = frame.f_globals["__name__"]
|
|
|
|
lineno = frame.f_lineno
|
|
line = linecache.getline(filename,lineno).rstrip()
|
|
|
|
data = "%s:%i: %s" % (name, lineno, line)
|
|
|
|
print "%s%s" % (data, mem.rjust(80 - len(data)))
|
|
|
|
return traceit_memory
|
|
|
|
lines = {}
|
|
|
|
def traceit(frame, event, arg):
|
|
if event == "line":
|
|
try:
|
|
filename = frame.f_globals["__file__"]
|
|
|
|
if filename.endswith(".pyc") or filename.endswith(".pyo"):
|
|
filename = filename[:-1]
|
|
|
|
name = frame.f_globals["__name__"]
|
|
|
|
lineno = frame.f_lineno
|
|
line = linecache.getline(filename,lineno).rstrip()
|
|
|
|
data = "%s:%i: %s" % (name, lineno, line)
|
|
|
|
print time.time(), data
|
|
|
|
#if data in lines:
|
|
# lines[data] += 1
|
|
#else:
|
|
# lines[data] = 1
|
|
|
|
except Exception, e:
|
|
print e
|
|
|
|
return traceit
|
|
|
|
def main():
|
|
import urk
|
|
urk.main()
|
|
|
|
sys.settrace(traceit)
|
|
main()
|