src/dockerfiles/satlab_remote_access/tools/host_command.py

# Copyright 2021 The Chromium OS Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import docker
import logging
import os
import sys

from moblab_common import host_connector

DOCKER_REGISTRY = "gcr.io/chromeos-partner-moblab/"

def reboot():
    """Reboot the docker host box."""
    host_connector.HostServicesConnector().reboot()


def factory_reset():
    """Run a factory reset on the docker host box."""
    host_connector.HostServicesConnector.factory_reset()


def get_host_identifier():
    """Get the serial number for the satlab as a whole"""
    print(host_connector.HostServicesConnector.get_host_identifier())


def update_satlab():
    """
        Sets off an update of all docker containers, using watchtower

    Returns: String result message.
    """
    client = docker.from_env(timeout=600)

    try:
        container = client.containers.get("update")
        if container.status != "running":
            container.remove(force=True)
        else:
            print("Already updating please wait.")
            return
    except docker.errors.NotFound:
        pass

    label = os.environ.get("WATCHTOWER_TAG", "release")
    container = client.containers.run(
        DOCKER_REGISTRY + "watchtower:%s" % label,
        volumes=["/var/run/docker.sock:/var/run/docker.sock"],
        name="update",
        detach=True,
        remove=True,
        command="--run-once --include-restarting --include-stopped --revive-stopped -d",
    )

    try:
        container.wait(timeout=60)
    except requests.exceptions.ReadTimeout:
        logging.exception("Watchtower update raised an exception.")
        container.stop()
    finally:
        container.remove()
    print("Update finished.")
    return

def update_system():
    """
        Initiates system update if it is available.
    Returns:
        An error message.
    """
    try:
        response = (
            host_connector.HostServicesConnector.install_system_update()
        )
        return response.error
    except host_connector.HostServicesException:
        logging.exception("Install system update raised an exception.")
        return "Failed to install update."

if __name__ == "__main__":
    if len(sys.argv) > 1:
        f = os.path.basename(sys.argv[1])
        locals()[f]()