Option to exit once demo loop plays through.

1 files changed, 3 insertions, 1 deletions

diff --git a/system_demo_mode/config.py b/system_demo_mode/config.py
index 6f31fbe8..93379061 100644
--- a/system_demo_mode/config.py
+++ b/system_demo_mode/config.py
@@ -36,7 +36,7 @@ def generate(dirpath, random_order, **kwargs):
     return config, dirpath
 
 
-def as_string(dirpath, random_order, **kwargs):
+def as_string(dirpath, random_order, exit, **kwargs):
     """ Config loader is in demo_mode.py
     """
     cfg, dirpath = generate(dirpath, random_order, **kwargs)
@@ -51,6 +51,8 @@ def as_string(dirpath, random_order, **kwargs):
     cfg_str += ["\n"]
     cfg_str += ["search_path = %r\n" % dirpath]
     cfg_str += ["\n"]
+    cfg_str += ["exit = %r\n" % exit]
+    cfg_str += ["\n"]
 
     # All these work but use nicest formatting!
     if 0:  # works but not nice to edit.