commit 9d4836e7319c2d32a2cec68a815858db4d5cbd04
author Ned Deily <nad@acm.org> Wed Mar 09 02:31:55 2011 -0800
committer Ned Deily <nad@acm.org> Wed Mar 09 02:31:55 2011 -0800
tree 475f66e81b0031767966453cce72384aed1592bc
parent ee145ecc5fdcbb23152407fc76dac8f6aa886c3f

Issue #5622: Fix curses.wrapper to raise correct exception if curses
initialization fails.

Lib/curses/wrapper.py

diff --git a/Lib/curses/wrapper.py b/Lib/curses/wrapper.py
index 3cdaa82..5183ce7 100644
--- a/Lib/curses/wrapper.py
+++ b/Lib/curses/wrapper.py
@@ -43,7 +43,8 @@
         return func(stdscr, *args, **kwds)
     finally:
         # Set everything back to normal
-        stdscr.keypad(0)
-        curses.echo()
-        curses.nocbreak()
-        curses.endwin()
+        if 'stdscr' in locals():
+            stdscr.keypad(0)
+            curses.echo()
+            curses.nocbreak()
+            curses.endwin()