From f41d7effdbe51bca0d6b28ecf7ed92fe3617009a Mon Sep 17 00:00:00 2001
From: Casper da Costa-Luis <casper.dcl@physics.org>
Date: Tue, 21 Sep 2021 16:54:55 +0100
Subject: [PATCH] fix dvc[all] (#6657)

* fix dvc[all]

- related https://github.com/iterative/dvc-snap/issues/24

* fix pylint
---
 setup.py | 19 +++++++++++++------
 1 file changed, 13 insertions(+), 6 deletions(-)

diff --git a/setup.py b/setup.py
index 748b19bbc..12a70d4c8 100644
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,7 @@
 import importlib.util
 import os
 from pathlib import Path
+from textwrap import dedent
 
 from setuptools import find_packages, setup
 from setuptools.command.build_py import build_py as _build_py
@@ -35,11 +36,16 @@ version = dvc_version.__version__  # noqa: F821
 #   - https://github.com/python/mypy
 class build_py(_build_py):
     def pin_version(self):
-        path = os.path.join(self.build_lib, "dvc")
-        self.mkpath(path)
-        with open(os.path.join(path, "version.py"), "w") as fobj:
-            fobj.write("# AUTOGENERATED at build time by setup.py\n")
-            fobj.write(f'__version__ = "{version}"\n')
+        ver_file = Path(self.build_lib) / "dvc" / "version.py"
+        self.mkpath(os.fspath(ver_file.parent))
+        ver_file.write_text(
+            dedent(
+                f"""\
+                # AUTOGENERATED at build time by setup.py
+                __version__ = "{version}"
+                """
+            )
+        )
 
     def run(self):
         self.execute(self.pin_version, ())
@@ -59,9 +65,10 @@ requirements = {
 
 install_requires = requirements.pop("default")
 requirements["all"] = [
-    requirements
+    req
     for key, requirements in requirements.items()
     if key not in ("tests", "ssh_gssapi", "terraform")
+    for req in requirements
 ]
 requirements["tests"] += requirements["terraform"]
 requirements["dev"] = requirements["all"] + requirements["tests"]