Add support for building with buildkit

Author: felipecrs

docker/api/build.py

@@ -16,7 +16,7 @@ def build(self, path=None, tag=None, quiet=False, fileobj=None,
               decode=False, buildargs=None, gzip=False, shmsize=None,
               labels=None, cache_from=None, target=None, network_mode=None,
               squash=None, extra_hosts=None, platform=None, isolation=None,
-              use_config_proxy=True):
+              version=None, use_config_proxy=True):
         """
         Similar to the ``docker build`` command. Either ``path`` or ``fileobj``
         needs to be set. ``path`` can be a local path (to a directory
@@ -101,6 +101,10 @@ def build(self, path=None, tag=None, quiet=False, fileobj=None,
             platform (str): Platform in the format ``os[/arch[/variant]]``
             isolation (str): Isolation technology used during build.
                 Default: `None`.
+            version (str): Version of the builder backend to use.
+                - `1` is the first generation classic (deprecated) builder in the Docker daemon (default)
+                - `2` is [BuildKit](https://github.com/moby/buildkit)
+                Default: `None`.
             use_config_proxy (bool): If ``True``, and if the docker client
                 configuration file (``~/.docker/config.json`` by default)
                 contains a proxy configuration, the corresponding environment
@@ -253,6 +257,13 @@ def build(self, path=None, tag=None, quiet=False, fileobj=None,
                 )
             params['isolation'] = isolation
 
+        if version is not None:
+            if utils.version_lt(self._version, '1.38'):
+                raise errors.InvalidVersion(
+                    'version was only introduced in API version 1.38'
+                )
+            params['version'] = version
+
         if context is not None:
             headers = {'Content-Type': 'application/tar'}
             if encoding:

tests/integration/api_build_test.py

@@ -201,6 +201,27 @@ def test_build_isolation(self):
         for _chunk in stream:
             pass
 
+    @requires_api_version('1.38')
+    def test_build_with_buildkit(self):
+        script = io.BytesIO('\n'.join([
+            'FROM scratch',
+            'COPY <<EOF greeting.txt',
+            'hello world',
+            'EOF'
+        ]).encode('ascii'))
+
+        self.tmp_imgs.append('buildkit')
+
+        stream = self.client.build(
+            fileobj=script, tag='buildkit',
+            version='2'
+        )
+
+        for _chunk in stream:
+            pass
+
+        assert self.client.inspect_image('buildkit')
+
     @requires_api_version('1.23')
     def test_build_labels(self):
         script = io.BytesIO('\n'.join([