From ada58be9b5c39cfe6f8c4763e2dab9a7f6c2bd14 Mon Sep 17 00:00:00 2001
From: Michal Modzelewski <michal.modzel@gmail.com>
Date: Thu, 7 Dec 2023 14:09:15 +0100
Subject: [PATCH] handle parquet directories

---
 visidata/loaders/parquet.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/visidata/loaders/parquet.py b/visidata/loaders/parquet.py
index d4fb99c69..50ca020ed 100644
--- a/visidata/loaders/parquet.py
+++ b/visidata/loaders/parquet.py
@@ -23,8 +23,11 @@ def iterload(self):
         pq = vd.importExternal("pyarrow.parquet", "pyarrow")
         from visidata.loaders.arrow import arrow_to_vdtype
 
-        with self.source.open('rb') as f:
-            self.tbl = pq.read_table(f)
+        if self.source.is_dir():
+            self.tbl = pq.read_table(str(self.source))
+        else: 
+            with self.source.open('rb') as f:
+                self.tbl = pq.read_table(f)
 
         self.columns = []
         for colname, col in zip(self.tbl.column_names, self.tbl.columns):