1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
import docker
import os
import shutil
from EnvsReader import EnvsReader
from docker.types import Mount
ARCHIVES_PATH = os.getenv("WORKSPACE") + "/archives/"
class PostProcessorDockerContainerUtils:
def __init__(self, mount_path, truststores_path):
self.mount_path = mount_path
self.truststores_path = truststores_path
def run_container(self, docker_image, image_name, path_to_env):
self.remove_mount_dir()
shutil.copytree(self.truststores_path, self.mount_path)
client = docker.from_env()
environment = EnvsReader().read_env_list_from_file(path_to_env)
container = client.containers.run(
image=docker_image,
name=image_name,
environment=environment,
user='root', # Run container as root to avoid permission issues with volume mount access
mounts=[Mount(target='/var/certs', source=self.mount_path, type='bind')],
detach=True
)
exitcode = container.wait()
return exitcode
def create_mount_dir(self):
if not os.path.exists(self.mount_path):
os.makedirs(self.mount_path)
def remove_mount_dir(self):
if os.path.exists(self.mount_path):
shutil.rmtree(self.mount_path)
def remove_container_and_save_logs(self, container_name, log_file_name):
client = docker.from_env()
container = client.containers.get(container_name)
text_file = open(ARCHIVES_PATH + "certservice_postprocessor_container_" + log_file_name + ".log", "wb")
text_file.write(container.logs())
text_file.close()
container.remove()
self.remove_mount_dir()
|
