From 56f8d1559996db0a4a4baf97f95ab96f3e9d4e47 Mon Sep 17 00:00:00 2001
From: Jean-Christophe Fillion-Robin <jchris.fillionr@kitware.com>
Date: Fri, 22 Mar 2024 19:22:46 -0400
Subject: [PATCH] feat: Add IDC_INDEX_DATA_GENERATE_CSV_ARCHIVE option (enabled
 by default)

---
 CMakeLists.txt        | 7 ++++---
 tests/test_package.py | 6 +++---
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index d2c7863..6db8d1d 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -10,13 +10,14 @@ if(NOT DEFINED ENV{GCP_PROJECT})
   message(FATAL_ERROR "GCP_PROJECT env. variable is not set")
 endif()
 
+option(IDC_INDEX_DATA_GENERATE_CSV_ARCHIVE "Generate idc_index.csv.zip file" ON)
 option(IDC_INDEX_DATA_GENERATE_PARQUET "Generate idc_index.parquet file" OFF)
 
 set(download_dir "${PROJECT_BINARY_DIR}")
 
 add_custom_command(
   OUTPUT
-    ${download_dir}/idc_index.csv.zip
+    $<$<BOOL:IDC_INDEX_DATA_GENERATE_CSV_ARCHIVE>:${download_dir}/idc_index.csv.zip>
     $<$<BOOL:IDC_INDEX_DATA_GENERATE_PARQUET>:${download_dir}/idc_index.parquet>
   COMMAND python ${CMAKE_CURRENT_SOURCE_DIR}/scripts/python/idc_index_data_manager.py
     --generate-csv-archive
@@ -25,12 +26,12 @@ add_custom_command(
 
 add_custom_target(run_idc_index_data_manager ALL
   DEPENDS
-    ${download_dir}/idc_index.csv.zip
+    $<$<BOOL:IDC_INDEX_DATA_GENERATE_CSV_ARCHIVE>:${download_dir}/idc_index.csv.zip>
     $<$<BOOL:IDC_INDEX_DATA_GENERATE_PARQUET>:${download_dir}/idc_index.parquet>
 )
 
 install(
   FILES
-    ${download_dir}/idc_index.csv.zip
+    $<$<BOOL:IDC_INDEX_DATA_GENERATE_CSV_ARCHIVE>:${download_dir}/idc_index.csv.zip>
     $<$<BOOL:IDC_INDEX_DATA_GENERATE_PARQUET>:${download_dir}/idc_index.parquet>
   DESTINATION "idc_index_data")
diff --git a/tests/test_package.py b/tests/test_package.py
index fea7d25..3870f64 100644
--- a/tests/test_package.py
+++ b/tests/test_package.py
@@ -10,9 +10,9 @@ def test_version():
 
 
 def test_filepath():
-    assert m.IDC_INDEX_CSV_ARCHIVE_FILEPATH is not None
-    assert m.IDC_INDEX_CSV_ARCHIVE_FILEPATH.is_file()
-    assert m.IDC_INDEX_CSV_ARCHIVE_FILEPATH.name == "idc_index.csv.zip"
+    if m.IDC_INDEX_CSV_ARCHIVE_FILEPATH is not None:
+        assert m.IDC_INDEX_CSV_ARCHIVE_FILEPATH.is_file()
+        assert m.IDC_INDEX_CSV_ARCHIVE_FILEPATH.name == "idc_index.csv.zip"
 
     if m.IDC_INDEX_PARQUET_FILEPATH is not None:
         assert m.IDC_INDEX_PARQUET_FILEPATH.is_file()