From 411f19334de2663d159428538e61a80aa7c98bad Mon Sep 17 00:00:00 2001 From: Todd Leonhardt Date: Fri, 16 Mar 2018 20:40:06 -0400 Subject: Added ability to specify a startup_script in cmd2.Cmd.__init__() --- cmd2.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'cmd2.py') diff --git a/cmd2.py b/cmd2.py index 7ce6e5c0..d357b02a 100755 --- a/cmd2.py +++ b/cmd2.py @@ -1088,7 +1088,7 @@ class Cmd(cmd.Cmd): 'timing': 'Report execution times'} def __init__(self, completekey='tab', stdin=None, stdout=None, persistent_history_file='', - persistent_history_length=1000, use_ipython=False, transcript_files=None): + persistent_history_length=1000, startup_script=None, use_ipython=False, transcript_files=None): """An easy but powerful framework for writing line-oriented command interpreters, extends Python's cmd package. :param completekey: str - (optional) readline name of a completion key, default to Tab @@ -1096,6 +1096,7 @@ class Cmd(cmd.Cmd): :param stdout: (optional) alternate output file object, if not specified, sys.stdout is used :param persistent_history_file: str - (optional) file path to load a persistent readline history from :param persistent_history_length: int - (optional) max number of lines which will be written to the history file + :param startup_script: str - (optional) file path to a a script to load and execute at startup :param use_ipython: (optional) should the "ipy" command be included for an embedded IPython shell :param transcript_files: str - (optional) allows running transcript tests when allow_cli_args is False """ @@ -1180,6 +1181,10 @@ class Cmd(cmd.Cmd): # If this string is non-empty, then this warning message will print if a broken pipe error occurs while printing self.broken_pipe_warning = '' + # If a startup script is provided, then add it in the queue to load + if startup_script is not None: + self.cmdqueue.append('load {}'.format(startup_script)) + # ----- Methods related to presenting output to the user ----- @property -- cgit v1.2.1