kopia lustrzana https://github.com/jupyterhub/repo2docker
63 wiersze
1.9 KiB
Python
63 wiersze
1.9 KiB
Python
"""Generates a variety of Dockerfiles based on an input matrix
|
|
"""
|
|
import os
|
|
|
|
import docker
|
|
|
|
from .base import BuildPack
|
|
|
|
|
|
class DockerBuildPack(BuildPack):
|
|
"""Docker BuildPack"""
|
|
|
|
dockerfile = "Dockerfile"
|
|
|
|
def detect(self):
|
|
"""Check if current repo should be built with the Docker BuildPack"""
|
|
return os.path.exists(self.binder_path("Dockerfile"))
|
|
|
|
def render(self, build_args=None):
|
|
"""Render the Dockerfile using by reading it from the source repo"""
|
|
Dockerfile = self.binder_path("Dockerfile")
|
|
with open(Dockerfile) as f:
|
|
return f.read()
|
|
|
|
def build(
|
|
self,
|
|
client,
|
|
image_spec,
|
|
memory_limit,
|
|
build_args,
|
|
cache_from,
|
|
extra_build_kwargs,
|
|
):
|
|
"""Build a Docker image based on the Dockerfile in the source repo."""
|
|
# If you work on this bit of code check the corresponding code in
|
|
# buildpacks/base.py where it is duplicated
|
|
if not isinstance(memory_limit, int):
|
|
raise ValueError(
|
|
"The memory limit has to be specified as an "
|
|
f"integer but is '{type(memory_limit)}'"
|
|
)
|
|
limits = {}
|
|
if memory_limit:
|
|
# We want to always disable swap. Docker expects `memswap` to
|
|
# be total allowable memory, *including* swap - while `memory`
|
|
# points to non-swap memory. We set both values to the same so
|
|
# we use no swap.
|
|
limits = {"memory": memory_limit, "memswap": memory_limit}
|
|
|
|
build_kwargs = dict(
|
|
path=os.getcwd(),
|
|
dockerfile=self.binder_path(self.dockerfile),
|
|
tag=image_spec,
|
|
buildargs=build_args,
|
|
container_limits=limits,
|
|
cache_from=cache_from,
|
|
labels=self.get_labels(),
|
|
)
|
|
|
|
build_kwargs.update(extra_build_kwargs)
|
|
|
|
yield from client.build(**build_kwargs)
|