Make ast_node_interactivity trait an Enum instead of Unicode

timo · May 28, 2012 · f277c8b · f277c8b
1 parent e369577
commit f277c8b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/IPython/core/interactiveshell.py b/IPython/core/interactiveshell.py
@@ -368,10 +368,11 @@ def _prompt_trait_changed(self, name, old, new):
             '"\C-u": unix-line-discard',
         ], allow_none=False, config=True)
 
-    ast_node_interactivity = Unicode('last_expr', config=True, help="""
+    ast_node_interactivity = Enum(['all', 'last', 'last_expr', 'none'],
+                                  default_value='last_expr', config=True, 
+                                  help="""
         'all', 'last', 'last_expr' or 'none'," specifying which nodes should be
-        run interactively (displaying output from expressions).
-        """)
+        run interactively (displaying output from expressions).""")
 
     # TODO: this part of prompt management should be moved to the frontends.
     # Use custom TraitTypes that convert '0'->'' and '\\n'->'\n'