diff options
-rwxr-xr-x | Lib/profile.py | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/Lib/profile.py b/Lib/profile.py index 715aa4b..7f4abed 100755 --- a/Lib/profile.py +++ b/Lib/profile.py @@ -396,8 +396,7 @@ class Profile: def run(self, cmd): import __main__ dict = __main__.__dict__ - self.runctx(cmd, dict, dict) - return self + return self.runctx(cmd, dict, dict) def runctx(self, cmd, globals, locals): self.set_cmd(cmd) @@ -406,16 +405,16 @@ class Profile: exec cmd in globals, locals finally: sys.setprofile(None) + return self # This method is more useful to profile a single function call. def runcall(self, func, *args): self.set_cmd(`func`) sys.setprofile(self.dispatcher) try: - apply(func, args) + return apply(func, args) finally: sys.setprofile(None) - return self #****************************************************************** |