from appium import webdriver
from appium.webdriver.common.appiumby import AppiumBy
from selenium.webdriver.support.ui import WebDriverWait

from permission_dialog_options import build_options, permission_payload


options = build_options("android")
driver = webdriver.Remote("http://127.0.0.1:4723", options=options)

try:
    print(f"session={driver.session_id}")
    print(f"permission_payload={permission_payload(options)}")
    driver.find_element(AppiumBy.ACCESSIBILITY_ID, "Use Location").click()
    status = WebDriverWait(driver, 10).until(
        lambda item: item.find_element(AppiumBy.ACCESSIBILITY_ID, "Location enabled")
    )
    print(f"permission_flow={status.text}")
finally:
    driver.quit()