pyinstaller: build: check dvc doctor output (iterative#6425)

2796gaurav · Aug 15, 2021 · 27b5419 · 27b5419
1 parent a7f01b6
commit 27b5419
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 1 deletion.
diff --git a/scripts/build-requirements.txt b/scripts/build-requirements.txt
@@ -1,2 +1,2 @@
-PyInstaller==4.3
+PyInstaller==4.5.1
 numpy==1.19.3
diff --git a/scripts/pyinstaller/build.py b/scripts/pyinstaller/build.py
@@ -20,3 +20,11 @@
     cwd=path,
     stderr=STDOUT,
 )
+
+check_call(
+    [
+        path / "dist" / "dvc" / "dvc",
+        "doctor",
+    ],
+    stderr=STDOUT,
+)
diff --git a/scripts/pyinstaller/hooks/hook-dvc.py b/scripts/pyinstaller/hooks/hook-dvc.py
@@ -1,2 +1,19 @@
+from PyInstaller.utils.hooks import (  # pylint:disable=import-error
+    copy_metadata,
+)
+
+# needed for `dvc doctor` to show dep versions
+datas = copy_metadata("adlfs", recursive=True)
+datas += copy_metadata("knack")
+datas += copy_metadata("gcsfs")
+datas += copy_metadata("pyarrow")
+datas += copy_metadata("hdfs")
+datas += copy_metadata("pydrive2")
+datas += copy_metadata("s3fs", recursive=True)
+datas += copy_metadata("boto3")
+datas += copy_metadata("ossfs")
+datas += copy_metadata("sshfs")
+datas += copy_metadata("webdav4")
+
 # https://github.com/pypa/setuptools/issues/1963
 hiddenimports = ["pkg_resources.py2_warn"]