urk_trace.py

   1 import sys
   2 import commands
   3 import linecache
   4
   5 import time
   6
   7 last_mem = [0]
   8
   9 def traceit_memory(frame, event, arg):
  10     if event == "line":
  11         mem = int(" " + commands.getoutput(
  12                 "ps -eo cmd,rss | grep urk_trace.py | grep -v grep"
  13                 ).split(" ")[-1])
  14
  15         if mem > last_mem[0]:
  16             last_mem[0] = mem
  17
  18             mem = str(mem)
  19
  20             filename = frame.f_globals["__file__"]
  21
  22             if filename.endswith(".pyc") or filename.endswith(".pyo"):
  23                 filename = filename[:-1]
  24
  25             name = frame.f_globals["__name__"]
  26
  27             lineno = frame.f_lineno
  28             line = linecache.getline(filename,lineno).rstrip()
  29
  30             data = "%s:%i: %s" % (name, lineno, line)
  31
  32             print "%s%s" % (data, mem.rjust(80 - len(data)))
  33
  34     return traceit_memory
  35
  36 lines = {}
  37
  38 def traceit(frame, event, arg):
  39     if event == "line":
  40         try:
  41             filename = frame.f_globals["__file__"]
  42
  43             if filename.endswith(".pyc") or filename.endswith(".pyo"):
  44                 filename = filename[:-1]
  45
  46             name = frame.f_globals["__name__"]
  47
  48             lineno = frame.f_lineno
  49             line = linecache.getline(filename,lineno).rstrip()
  50
  51             data = "%s:%i: %s" % (name, lineno, line)
  52
  53             print time.time(), data
  54
  55             #if data in lines:
  56             #    lines[data] += 1
  57             #else:
  58             #    lines[data] = 1
  59
  60         except Exception, e:
  61             print e
  62
  63     return traceit
  64
  65 def main():
  66     import urk
  67     urk.main()
  68
  69 sys.settrace(traceit)
  70 main()