Add --compression to export-jobs subcommand

crate · Jul 17, 2023 · 8041764 · 8041764
1 parent ba46c0e
commit 8041764
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/croud/__main__.py b/croud/__main__.py
@@ -818,6 +818,13 @@
                                 choices=["csv", "json", "parquet"],
                                 help="The format of the data in the file.",
                             ),
+                            Argument(
+                                "--compression",
+                                type=str,
+                                required=False,
+                                choices=["gzip"],
+                                help="The compression method of the exported file.",
+                            ),
                             Argument(
                                 "--save-as",
                                 type=str,

diff --git a/croud/clusters/commands.py b/croud/clusters/commands.py
@@ -657,6 +657,9 @@ def export_jobs_create(args: Namespace) -> None:
         "destination": {"format": args.file_format},
     }
 
+    if args.compression:
+        body["compression"] = args.compression
+
     client = Client.from_args(args)
     data, errors = client.post(
         f"/api/v2/clusters/{args.cluster_id}/export-jobs/", body=body

diff --git a/tests/commands/test_clusters.py b/tests/commands/test_clusters.py
@@ -1749,6 +1749,8 @@ def test_export_job_create(mock_client_request, mock_copy, save_file):
                 "csv",
                 "--table",
                 "my-table",
+                "--compression",
+                "gzip",
             )
             if save_file:
                 cmd = cmd + (
@@ -1762,6 +1764,7 @@ def test_export_job_create(mock_client_request, mock_copy, save_file):
             "table": "my-table",
         },
         "destination": {"format": "csv"},
+        "compression": "gzip",
     }
     assert_rest(
         mock_client_request,