Selenium: wait until an element can be found by relative locator

wait_relative.py

from selenium.common import NoSuchElementException
from selenium.webdriver.common.by import By
from selenium.webdriver.remote.webdriver import WebDriver
from selenium.webdriver.remote.webelement import WebElement
from selenium.webdriver.support.relative_locator import RelativeBy, locate_with

class WaitRelative:
    def __init__(self, by: RelativeBy):
        self._by = by

    def __call__(self, driver: WebDriver) -> WebElement | Literal[False]:
        try:
            return driver.find_element(self._by)
        except NoSuchElementException:
            return False

def wait_relative(driver: WebDriver, by: RelativeBy, timeout: float = 30
  ) -> WebElement:
    return WebDriverWait(driver, timeout).until(WaitRelative(by))

e = wait_relative(
  locate_with(By.CSS_SELECTOR, 'div[data-name="Shared"]').below(other))