def set_expose_ports(expose: PortMapping, pod: Manifest,
container_name: str) -> None:
"""Merge container ports into the expose list."""
pod_spec = pod["spec"] # type: Manifest
container = find_container(pod_spec, container_name)
expose.merge_automatic_ports([
port["containerPort"] for port in container.get("ports", [])
if port["protocol"] == "TCP"
])
def new_swapped_deployment(
old_deployment: Dict,
container_to_update: str,
run_id: str,
expose: PortMapping,
add_custom_nameserver: bool,
) -> Dict:
"""
Create a new Deployment that uses telepresence-k8s image.
Makes the following changes:
1. Changes to single replica.
2. Disables command, args, livenessProbe, readinessProbe, workingDir.
3. Adds labels.
4. Adds TELEPRESENCE_NAMESERVER env variable, if requested.
5. Runs as root, if requested.
6. Sets terminationMessagePolicy.
7. Adds TELEPRESENCE_CONTAINER_NAMESPACE env variable so the forwarder does
not have to access the k8s API from within the pod.
Returns dictionary that can be encoded to JSON and used with kubectl apply.
Mutates the passed-in PortMapping to include container ports.
"""
new_deployment_json = deepcopy(old_deployment)
new_deployment_json["spec"]["replicas"] = 1
new_deployment_json["metadata"].setdefault("labels",
{})["telepresence"] = run_id
new_deployment_json["spec"]["template"]["metadata"].setdefault(
"labels", {})["telepresence"] = run_id
for container, old_container in zip(
new_deployment_json["spec"]["template"]["spec"]["containers"],
old_deployment["spec"]["template"]["spec"]["containers"],
):
if container["name"] == container_to_update:
# Merge container ports into the expose list
expose.merge_automatic_ports([
port["containerPort"] for port in container.get("ports", [])
if port["protocol"] == "TCP"
])
container["image"] = get_image_name(expose)
# Not strictly necessary for real use, but tests break without this
# since we don't upload test images to Docker Hub:
container["imagePullPolicy"] = "IfNotPresent"
# Drop unneeded fields:
for unneeded in [
"command", "args", "livenessProbe", "readinessProbe",
"workingDir", "lifecycle"
]:
try:
container.pop(unneeded)
except KeyError:
pass
# We don't write out termination file:
container["terminationMessagePolicy"] = "FallbackToLogsOnError"
# Use custom name server if necessary:
if add_custom_nameserver:
container.setdefault("env", []).append({
"name":
"TELEPRESENCE_NAMESERVER",
"value":
get_alternate_nameserver()
})
# Add namespace environment variable to support deployments using
# automountServiceAccountToken: false. To be used by forwarder.py
# in the k8s-proxy.
container.setdefault("env", []).append({
"name": "TELEPRESENCE_CONTAINER_NAMESPACE",
"valueFrom": {
"fieldRef": {
"fieldPath": "metadata.namespace"
}
}
})
return new_deployment_json
raise RuntimeError("Couldn't find container {} in the Deployment.".format(
container_to_update))
