QT/superset2
1
0
Fork 0
mirror of https://github.com/apache/superset.git synced 2026-04-19 08:04:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(pre-commit): Add pyupgrade and pycln hooks (#24197)

Browse Source
This commit is contained in:
John Bodley
2023-06-01 12:01:10 -07:00
committed by GitHub
parent 7d7ce63970
commit a4d5d7c6b9
448 changed files with 3084 additions and 3305 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
superset/utils/webdriver.py
View File

@@ -20,7 +20,7 @@ from __future__ import annotations
import logging
from enum import Enum
from time import sleep
from typing import Any, Dict, List, Optional, Tuple, TYPE_CHECKING
from typing import Any, TYPE_CHECKING
from flask import current_app
from selenium.common.exceptions import (
@@ -37,7 +37,7 @@ from selenium.webdriver.support.ui import WebDriverWait
from superset.extensions import machine_auth_provider_factory
from superset.utils.retries import retry_call
WindowSize = Tuple[int, int]
WindowSize = tuple[int, int]
logger = logging.getLogger(__name__)
@@ -56,7 +56,7 @@ class ChartStandaloneMode(Enum):
SHOW_NAV = 0
def find_unexpected_errors(driver: WebDriver) -> List[str]:
def find_unexpected_errors(driver: WebDriver) -> list[str]:
error_messages = []
try:
@@ -111,7 +111,7 @@ def find_unexpected_errors(driver: WebDriver) -> List[str]:
class WebDriverProxy:
def __init__(self, driver_type: str, window: Optional[WindowSize] = None):
def __init__(self, driver_type: str, window: WindowSize | None = None):
self._driver_type = driver_type
self._window: WindowSize = window or (800, 600)
self._screenshot_locate_wait = current_app.config["SCREENSHOT_LOCATE_WAIT"]
@@ -124,7 +124,7 @@ class WebDriverProxy:
options = firefox.options.Options()
profile = FirefoxProfile()
profile.set_preference("layout.css.devPixelsPerPx", str(pixel_density))
kwargs: Dict[Any, Any] = dict(options=options, firefox_profile=profile)
kwargs: dict[Any, Any] = dict(options=options, firefox_profile=profile)
elif self._driver_type == "chrome":
driver_class = chrome.webdriver.WebDriver
options = chrome.options.Options()
@@ -164,13 +164,11 @@ class WebDriverProxy:
except Exception: # pylint: disable=broad-except
pass
def get_screenshot(
self, url: str, element_name: str, user: User
) -> Optional[bytes]:
def get_screenshot(self, url: str, element_name: str, user: User) -> bytes | None:
driver = self.auth(user)
driver.set_window_size(*self._window)
driver.get(url)
img: Optional[bytes] = None
img: bytes | None = None
selenium_headstart = current_app.config["SCREENSHOT_SELENIUM_HEADSTART"]
logger.debug("Sleeping for %i seconds", selenium_headstart)
sleep(selenium_headstart)