OliveTin/docs/modules/ROOT/images/solutions/container-control-panel/setup_preview.py

#!/usr/bin/env python3
"""Open the My Containers dashboard for the container control panel solution."""

import time

from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait


def _wait_for_dashboard(driver, timeout=30):
    WebDriverWait(driver, timeout).until(
        lambda d: d.execute_script("return !!window.client")
    )
    WebDriverWait(driver, timeout).until(
        lambda d: bool(d.find_element(By.TAG_NAME, "body").get_attribute("loaded-dashboard"))
    )


def _wait_for_my_containers_dashboard(driver, timeout=30):
    def ready(d):
        required_titles = [
            "Start minecraft",
            "Stop minecraft",
            "Start brave_shirley",
            "Stop brave_shirley",
        ]
        for title in required_titles:
            try:
                button = d.find_element(By.CSS_SELECTOR, f'[title="{title}"]')
            except Exception:
                return False
            if not button.is_displayed():
                return False
        return True

    WebDriverWait(driver, timeout).until(ready)


def run(driver):
    _wait_for_dashboard(driver)
    _wait_for_my_containers_dashboard(driver)
    time.sleep(0.2)