import cmd

class PerfShell(cmd.Cmd):
 intro = 'Welcome to the perf shell. Type help or ? to list commands.\n'
 prompt = '(perf) '
 file = None

 def do_static_mem_test(self, arg):
 '静态内存测试'
 print(arg)

 def do_bye(self, arg):
 'Stop recording, close the turtle window, and exit: BYE'
 print('Thank you for using Turtle')
 self.close()
 return True

 # ----- record and playback -----

 def do_record(self, arg):
 'Save future commands to filename: RECORD rose.cmd'
 self.file = open(arg, 'w')

 def do_playback(self, arg):
 'Playback commands from a file: PLAYBACK rose.cmd'
 self.close()
 with open(arg) as f:
 self.cmdqueue.extend(f.read().splitlines())

 def precmd(self, line): #命令行执行前的钩子函数
 line = line.lower()
 if self.file and 'playback' not in line:
 print(line, file=self.file)
 return line

 def close(self):
 if self.file:
 self.file.close()
 self.file = None


if __name__ == '__main__':
 PerfShell().cmdloop()