Patch #445538: add completion for pstats.py sort cmd.

1 files changed, 2 insertions, 0 deletions

diff --git a/Lib/pstats.py b/Lib/pstats.py
index 20a6cd1..4fd55c8 100644
--- a/Lib/pstats.py
+++ b/Lib/pstats.py
@@ -640,6 +640,8 @@ if __name__ == '__main__':
         def help_sort(self):
             print "Sort profile data according to specified keys."
             print "(Typing `sort' without arguments lists valid keys.)"
+        def complete_sort(self, text, *args):
+            return [a for a in Stats.sort_arg_dict_default.keys() if a.startswith(text)]
 
         def do_stats(self, line):
             return self.generic('print_stats', line)