[Python-checkins] r83389 - python/branches/py3k/Lib/pstats.py

[georg.brandl]

Author: georg.brandl
Date: Sun Aug  1 09:57:47 2010
New Revision: 83389

Log:
Small improvements to pstats browser: do not crash on reading invalid file, and actually do a reload when executing "read" as intended.

Modified:
   python/branches/py3k/Lib/pstats.py

Modified: python/branches/py3k/Lib/pstats.py
==============================================================================
--- python/branches/py3k/Lib/pstats.py	(original)
+++ python/branches/py3k/Lib/pstats.py	Sun Aug  1 09:57:47 2010
@@ -614,14 +614,19 @@
                 except IOError as err:
                     print(err.args[1], file=self.stream)
                     return
+                except Exception as err:
+                    print(err.__class__.__name__ + ':', err, file=self.stream)
+                    return
                 self.prompt = line + "% "
             elif len(self.prompt) > 2:
-                line = self.prompt[-2:]
+                line = self.prompt[:-2]
+                self.do_read(line)
             else:
                 print("No statistics object is current -- cannot reload.", file=self.stream)
             return 0
         def help_read(self):
             print("Read in profile data from a specified file.", file=self.stream)
+            print("Without argument, reload the current file.", file=self.stream)
 
         def do_reverse(self, line):
             if self.stats:
@@ -664,6 +669,9 @@
         def help_strip(self):
             print("Strip leading path information from filenames in the report.", file=self.stream)
 
+        def help_help(self):
+            print("Show help for a given command.", file=self.stream)
+
         def postcmd(self, stop, line):
             if stop:
                 return stop