Oops. Debian is missing the pstats module too.

1 files changed, 3 insertions, 2 deletions

diff --git a/cherrypy/lib/profiler.py b/cherrypy/lib/profiler.py
index fb06606f..004221c0 100644
--- a/cherrypy/lib/profiler.py
+++ b/cherrypy/lib/profiler.py
@@ -38,13 +38,14 @@ def new_func_strip_path(func_name):
     if filename.endswith("__init__.py"):
         return os.path.basename(filename[:-12]) + filename[-12:], line, name
     return os.path.basename(filename), line, name
-import pstats
-pstats.func_strip_path = new_func_strip_path
 
 try:
     import profile
+    import pstats
+    pstats.func_strip_path = new_func_strip_path
 except ImportError:
     profile = None
+    pstats = None
     import warnings
     msg = ("Your installation of Python doesn't have a profile module. "
            "If you're on Debian, you can apt-get python2.4-profiler from "