adding support for multiples -c / --command parameter to pgcli w/tests

Author: diego-feito-stori

changelog.rst

@@ -9,6 +9,9 @@ Features:
     * Support dsn specific init-command in the config file
 * Add suggestion when setting the search_path
 * Allow per dsn_alias ssh tunnel selection
+* Add support for `single-command` to run a SQL command and exit.
+    * Command line option `-c` or `--command`.
+    * You can specify multiple times.
 
 Internal:
 ---------

pgcli/main.py

@@ -911,12 +911,14 @@ def _check_ongoing_transaction_and_allow_quitting(self):
     def run_cli(self):
         logger = self.logger
 
-        # Handle single command mode (-c flag) - similar to psql behavior
-        if hasattr(self, 'single_command') and self.single_command:
+        # Handle command mode (-c flag) - similar to psql behavior
+        # Multiple -c options are executed sequentially
+        if hasattr(self, 'commands') and self.commands:
             try:
-                logger.debug("Running single command: %s", self.single_command)
-                # Execute the command using the same logic as interactive mode
-                self.handle_watch_command(self.single_command)
+                for command in self.commands:
+                    logger.debug("Running command: %s", command)
+                    # Execute the command using the same logic as interactive mode
+                    self.handle_watch_command(command)
             except PgCliQuitError:
                 # Normal exit from quit command
                 sys.exit(0)
@@ -925,7 +927,7 @@ def run_cli(self):
                 logger.error("traceback: %r", traceback.format_exc())
                 click.secho(str(e), err=True, fg="red")
                 sys.exit(1)
-            # Exit successfully after executing the command
+            # Exit successfully after executing all commands
             sys.exit(0)
 
         history_file = self.config["main"]["history_file"]
@@ -1295,7 +1297,8 @@ def is_too_tall(self, lines):
         return len(lines) >= (self.prompt_app.output.get_size().rows - 4)
 
     def echo_via_pager(self, text, color=None):
-        if self.pgspecial.pager_config == PAGER_OFF or self.watch_command:
+        # Disable pager for -c/--command mode and \watch command
+        if self.pgspecial.pager_config == PAGER_OFF or self.watch_command or (hasattr(self, 'commands') and self.commands):
             click.echo(text, color=color)
         elif self.pgspecial.pager_config == PAGER_LONG_OUTPUT and self.table_format != "csv":
             lines = text.split("\n")
@@ -1446,8 +1449,9 @@ def echo_via_pager(self, text, color=None):
 @click.option(
     "-c",
     "--command",
-    default="",
-    help="run only single command (SQL or internal) and exit.",
+    "commands",
+    multiple=True,
+    help="run command (SQL or internal) and exit. Multiple -c options are allowed.",
 )
 @click.argument("dbname", default=lambda: None, envvar="PGDATABASE", nargs=1)
 @click.argument("username", default=lambda: None, envvar="PGUSER", nargs=1)
@@ -1477,7 +1481,7 @@ def cli(
     ssh_tunnel: str,
     init_command: str,
     log_file: str,
-    command: str,
+    commands: tuple,
 ):
     if version:
         print("Version:", __version__)
@@ -1538,8 +1542,8 @@ def cli(
         log_file=log_file,
     )
 
-    # Store single command for -c option
-    pgcli.single_command = command if command else None
+    # Store commands for -c option (can be multiple)
+    pgcli.commands = commands if commands else None
 
     # Choose which ever one has a valid value.
     if dbname_opt and dbname: