hdfs: implement isdir and isfile

Fixes iterative#4332
efiop · Aug 24, 2020 · d10d49c · d10d49c
1 parent f789cdf
commit d10d49c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/dvc/tree/hdfs.py b/dvc/tree/hdfs.py
@@ -130,6 +130,14 @@ def copy(self, from_info, to_info, **_kwargs):
                     self.remove(tmp_info)
                     raise
 
+    def isfile(self, path_info):
+        with self.hdfs(path_info) as hdfs:
+            return hdfs.isfile(path_info.path)
+
+    def isdir(self, path_info):
+        with self.hdfs(path_info) as hdfs:
+            return hdfs.isdir(path_info.path)
+
     def hadoop_fs(self, cmd, user=None):
         cmd = "hadoop fs -" + cmd
         if user:

diff --git a/tests/func/test_import_url.py b/tests/func/test_import_url.py
@@ -146,6 +146,7 @@ def test_import_url(tmp_dir, dvc, workspace):
         pytest.lazy_fixture("local_cloud"),
         pytest.lazy_fixture("s3"),
         pytest.lazy_fixture("gs"),
+        pytest.lazy_fixture("hdfs"),
         pytest.param(
             pytest.lazy_fixture("ssh"),
             marks=pytest.mark.skipif(