89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
import docker
|
|
from loguru import logger
|
|
|
|
|
|
def get_docker_client():
|
|
return docker.from_env()
|
|
|
|
|
|
def build_image(path, tag):
|
|
logger.info("Building image named %s with file %s..." % (tag, path))
|
|
|
|
docker_client = get_docker_client()
|
|
return docker_client.images.build(
|
|
path=path,
|
|
tag=tag,
|
|
)
|
|
|
|
|
|
def run_container(
|
|
image_name,
|
|
container_name,
|
|
command=False,
|
|
ports=False,
|
|
volumes=False,
|
|
links=False,
|
|
detach=False,
|
|
auto_remove=False,
|
|
):
|
|
client = get_docker_client()
|
|
|
|
logger.info("Launching Docker container named %s ..." % (image_name))
|
|
debug_docker_command = "docker run --name %s\\\n" % (container_name)
|
|
if ports:
|
|
for internal_port, host_port in ports.items():
|
|
debug_docker_command += (
|
|
" --publish {host_port}:{internal_port}\\\n".format(
|
|
internal_port=internal_port, host_port=host_port
|
|
)
|
|
)
|
|
if volumes:
|
|
for volume in volumes:
|
|
external_path, internal_path = volume.split(":")
|
|
debug_docker_command += (
|
|
" --volume {external_path}:{internal_path}\\\n".format(
|
|
external_path=external_path, internal_path=internal_path
|
|
)
|
|
)
|
|
if links:
|
|
for k, v in links.items():
|
|
debug_docker_command += " --link {k}:{v}\\\n".format(k=k, v=v)
|
|
if auto_remove:
|
|
debug_docker_command += " --rm"
|
|
if detach:
|
|
debug_docker_command += " --detach"
|
|
debug_docker_command += " %s\\\n" % (image_name)
|
|
debug_docker_command += " %s" % (command)
|
|
logger.debug(debug_docker_command)
|
|
|
|
container = client.containers.run(
|
|
image_name,
|
|
name=container_name,
|
|
command=command,
|
|
ports=ports,
|
|
volumes=volumes,
|
|
links=links,
|
|
detach=detach,
|
|
auto_remove=auto_remove,
|
|
)
|
|
if detach:
|
|
logger.info("Container launched.")
|
|
elif auto_remove:
|
|
logger.info("Container closed.")
|
|
|
|
return container
|
|
|
|
|
|
def kill_container(container_name):
|
|
client = get_docker_client()
|
|
containers = client.containers.list(
|
|
all=True,
|
|
filters={"name": container_name},
|
|
)
|
|
for container in containers:
|
|
logger.info(
|
|
"Stop container %s, based on image '%s'."
|
|
% (container.name, ",".join(container.image.tags))
|
|
)
|
|
container.stop()
|