Adding exitscript option to SelectFromList from pyrevit.forms

Author: iorhanV

pyrevitlib/pyrevit/forms/__init__.py

@@ -804,6 +804,8 @@ class SelectFromList(TemplateUserInputWindow):
             object attribute that should be read as item name.
         multiselect (bool, optional):
             allow multi-selection (uses check boxes). defaults to False
+        exitscript (bool, optional):
+            exit if close or nothing is returned. defaults to False
         info_panel (bool, optional):
             show information panel and fill with .description property of item
         return_all (bool, optional):
@@ -902,6 +904,9 @@ def _setup(self, **kwargs):
             self.list_lb.SelectionMode = Controls.SelectionMode.Single
             self.hide_element(self.checkboxbuttons_g)
 
+        # exitscript
+        self.exitscript = kwargs.get('exitscript', False)
+
         # info panel?
         self.info_panel = kwargs.get("info_panel", False)
 
@@ -1181,6 +1186,24 @@ def clear_search(self, sender, args):  # pylint: disable=W0613
         self.search_tb.Clear()
         self.search_tb.Focus()
 
+    @classmethod
+    def show(cls, context,  #pylint: disable=W0221
+             title='User Input',
+             width=DEFAULT_INPUTWINDOW_WIDTH,
+             height=DEFAULT_INPUTWINDOW_HEIGHT, **kwargs):
+        """Show user input window.
+
+        Args:
+            context (any): window context element(s)
+        """
+        dlg = cls(context, title, width, height, **kwargs)
+        dlg.ShowDialog()
+
+        if dlg.exitscript and not dlg.response:
+            sys.exit()
+
+        return dlg.response
+
 
 class CommandSwitchWindow(TemplateUserInputWindow):
     """Standard form to select from a list of command options.