fix: parquet column parsing (#127)

* fix: parquet column parsing * avoid use of unwrap() --------- Co-authored-by: sslivkoff <[email protected]>
paradigmxyz · Dec 27, 2023 · 17d25d7 · 17d25d7
1 parent ae8d152
commit 17d25d7
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/crates/cli/src/parse/parse_utils.rs b/crates/cli/src/parse/parse_utils.rs
@@ -51,8 +51,13 @@ pub(crate) fn parse_binary_arg(
     let mut parsed = HashMap::new();
 
     // separate into files vs explicit
-    let (files, hex_strings): (Vec<&String>, Vec<&String>) =
-        inputs.iter().partition(|tx| std::path::Path::new(tx).exists());
+    let (files, hex_strings): (Vec<&String>, Vec<&String>) = inputs.iter().partition(|tx| {
+        // strip off column name if present
+        match parse_file_column_reference(tx, default_column) {
+            Ok(reference) => std::path::Path::new(&reference.path).exists(),
+            _ => false,
+        }
+    });
 
     // files columns
     for path in files {