add check if data_dir exists

dlt-hub · Jul 5, 2023 · 5ca685e · 5ca685e
1 parent 146d2d6
commit 5ca685e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sources/filesystem/__init__.py b/sources/filesystem/__init__.py
@@ -27,6 +27,12 @@ def local_folder(
     Yields:
         TDataItem: A dictionary representing a file, containing the file path.
     """
+    if not Path(data_dir).is_dir():
+        if not Path(data_dir).exists():
+            raise ValueError(f"Local folder doesn't exist: {data_dir}")
+        else:
+            raise ValueError(f"Local folder is not a directory: {data_dir}")
+
     if extensions:
         files = (
             p.resolve()