Merge pull request #1733 from volatilityfoundation/add-script-only-volshell-feature

Add a --script-only flag that exits after the given volshell script i…

Author: ikelos

volatility3/cli/volshell/generic.py

@@ -60,7 +60,13 @@ def get_requirements(cls) -> List[interfaces.configuration.RequirementInterface]
                     description="File to load and execute at start",
                     default=None,
                     optional=True,
-                )
+                ),
+                requirements.BooleanRequirement(
+                    name="script-only",
+                    description="Exit volshell after the script specified in --script completes",
+                    default=False,
+                    optional=True,
+                ),
             ]
         return reqs + [
             requirements.TranslationLayerRequirement(
@@ -135,6 +141,9 @@ def in_prompt_tokens(self, cli=None):
         if self.config.get("script", None) is not None:
             self.run_script(location=self.config["script"])
 
+            if self.config.get("script-only"):
+                exit()
+
         if has_ipython:
             self.__console()
         else: