1 """curses.wrapper 2 3 Contains one function, wrapper(), which runs another function which 4 should be the rest of your curses-based application. If the 5 application raises an exception, wrapper() will restore the terminal 6 to a sane state so you can read the resulting traceback. 7 8 """ 9 10 import curses 11 12 def wrapper(func, *args, **kwds): 13 """Wrapper function that initializes curses and calls another function, 14 restoring normal keyboard/screen behavior on error. 15 The callable object 'func' is then passed the main window 'stdscr' 16 as its first argument, followed by any other arguments passed to 17 wrapper(). 18 """ 19 20 try: 21 # Initialize curses 22 stdscr = curses.initscr() 23 24 # Turn off echoing of keys, and enter cbreak mode, 25 # where no buffering is performed on keyboard input 26 curses.noecho() 27 curses.cbreak() 28 29 # In keypad mode, escape sequences for special keys 30 # (like the cursor keys) will be interpreted and 31 # a special value like curses.KEY_LEFT will be returned 32 stdscr.keypad(1) 33 34 # Start color, too. Harmless if the terminal doesn't have 35 # color; user can test with has_color() later on. The try/catch 36 # works around a minor bit of over-conscientiousness in the curses 37 # module -- the error return from C start_color() is ignorable. 38 try: 39 curses.start_color() 40 except: 41 pass 42 43 return func(stdscr, *args, **kwds) 44 finally: 45 # Set everything back to normal 46 if 'stdscr' in locals(): 47 stdscr.keypad(0) 48 curses.echo() 49 curses.nocbreak() 50 curses.endwin() 51
