Check if startup commands are a list.

dbcli · May 3, 2024 · c03632a · c03632a
1 parent 9d5bcf0
commit c03632a
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/litecli/liteclirc b/litecli/liteclirc
@@ -122,7 +122,7 @@ output.even-row = ""
 
 # Startup commands
 # litecli commands or sqlite commands to be executed on startup.
-# some of them will require you to have a database attached. 
+# some of them will require you to have a database attached.
 # they will be executed in the same order as they appear in the list.
 [startup_commands]
-#commands = ".tables", "pragma foreign_keys = ON;"
+#commands = ".tables", "pragma foreign_keys = ON;"
diff --git a/litecli/main.py b/litecli/main.py
@@ -588,7 +588,11 @@ def one_iteration(text=None):
         def startup_commands():
             if self.startup_commands:
                 if "commands" in self.startup_commands:
-                    for command in self.startup_commands["commands"]:
+                    if isinstance(self.startup_commands["commands"], str):
+                        commands = [self.startup_commands["commands"]]
+                    else:
+                        commands = self.startup_commands["commands"]
+                    for command in commands:
                         try:
                             res = sqlexecute.run(command)
                         except Exception as e: