From: Neal Norwitz <nnorwitz@gmail.com>
Date: Sun, 11 Jun 2006 07:27:56 +0000 (+0000)
Subject: Fix errors found by pychecker.
X-Git-Tag: v2.5b1~136
X-Git-Url: https://granicus.if.org/sourcecode?a=commitdiff_plain;h=e588c2ba9716a81358438b4e7cd23724dcef2ba9;p=python

Fix errors found by pychecker.
I think these changes are correct, but I'm not sure.  Could someone
who knows how this module works test it?  It can at least start on
the cmd line.
---

diff --git a/Lib/pstats.py b/Lib/pstats.py
index c3a88284ca..4e94b0ce0e 100644
--- a/Lib/pstats.py
+++ b/Lib/pstats.py
@@ -548,8 +548,10 @@ if __name__ == '__main__':
             self.prompt = "% "
             if profile is not None:
                 self.stats = Stats(profile)
+                self.stream = self.stats.stream
             else:
                 self.stats = None
+                self.stream = sys.stdout
 
         def generic(self, fn, line):
             args = line.split()
@@ -667,14 +669,15 @@ if __name__ == '__main__':
             return None
 
     import sys
-    print >> self.stream, "Welcome to the profile statistics browser."
     if len(sys.argv) > 1:
         initprofile = sys.argv[1]
     else:
         initprofile = None
     try:
-        ProfileBrowser(initprofile).cmdloop()
-        print >> self.stream, "Goodbye."
+        browser = ProfileBrowser(initprofile)
+        print >> browser.stream, "Welcome to the profile statistics browser."
+        browser.cmdloop()
+        print >> browser.stream, "Goodbye."
     except KeyboardInterrupt:
         pass