From: Andrew M. Kuchling <amk@amk.ca>
Date: Thu, 5 Apr 2001 16:08:41 +0000 (+0000)
Subject: Bug #412086, reported by Peter Wilson: The _curses module doesn't
X-Git-Tag: v2.1c1~186
X-Git-Url: http://git.ipfire.org/gitweb.cgi?a=commitdiff_plain;h=37f026381167db43ab3ccce59bfe26a098eeac56;p=thirdparty%2FPython%2Fcpython.git

Bug #412086, reported by Peter Wilson: The _curses module doesn't
    define COLORS or COLOR_PAIRS until after start_color() is called,
    but they were never added to the curses module.  Fixed by adding
    a wrapper around start_color(), similar to the wrapper around initscr().
---

diff --git a/Lib/curses/__init__.py b/Lib/curses/__init__.py
index 6bcd56602c71..1838de8a694a 100644
--- a/Lib/curses/__init__.py
+++ b/Lib/curses/__init__.py
@@ -32,6 +32,19 @@ def initscr():
     
     return stdscr
 
+# This is a similar wrapper for start_color(), which adds the COLORS and
+# COLOR_PAIRS variables which are only available after start_color() is
+# called.
+ 
+def start_color():
+    import _curses, curses
+    retval = _curses.start_color()
+    if hasattr(_curses, 'COLORS'):
+        curses.COLORS = _curses.COLORS
+    if hasattr(_curses, 'COLOR_PAIRS'):
+        curses.COLOR_PAIRS = _curses.COLOR_PAIRS
+    return retval 
+
 # Import Python has_key() implementation if _curses doesn't contain has_key()
 
 try: