diff --git a/src/operators/LibChecker.py b/src/operators/LibChecker.py index 3621304..e296527 100644 --- a/src/operators/LibChecker.py +++ b/src/operators/LibChecker.py @@ -13,6 +13,7 @@ import queue from datetime import datetime, timedelta from selenium.webdriver.common.by import By +from selenium.webdriver.chrome.webdriver import WebDriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC @@ -25,7 +26,7 @@ class LibChecker(LibOperator): self, input_queue: queue.Queue, output_queue: queue.Queue, - driver: any + driver: WebDriver ): super().__init__(input_queue, output_queue) diff --git a/src/operators/LibCheckin.py b/src/operators/LibCheckin.py index 565bfbd..7e5ffd6 100644 --- a/src/operators/LibCheckin.py +++ b/src/operators/LibCheckin.py @@ -11,6 +11,7 @@ import time import queue from selenium.webdriver.common.by import By +from selenium.webdriver.chrome.webdriver import WebDriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC @@ -23,7 +24,7 @@ class LibCheckin(LibOperator): self, input_queue: queue.Queue, output_queue: queue.Queue, - driver: any + driver: WebDriver ): super().__init__(input_queue, output_queue) diff --git a/src/operators/LibCheckout.py b/src/operators/LibCheckout.py index b978e3c..bf637d3 100644 --- a/src/operators/LibCheckout.py +++ b/src/operators/LibCheckout.py @@ -13,6 +13,7 @@ import queue from datetime import datetime, timedelta from selenium.webdriver.common.by import By +from selenium.webdriver.chrome.webdriver import WebDriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC @@ -25,7 +26,7 @@ class LibCheckout(LibOperator): self, input_queue: queue.Queue, output_queue: queue.Queue, - driver: any + driver: WebDriver ): super().__init__(input_queue, output_queue) diff --git a/src/operators/LibLogin.py b/src/operators/LibLogin.py index e1277c0..31cf2de 100644 --- a/src/operators/LibLogin.py +++ b/src/operators/LibLogin.py @@ -13,6 +13,7 @@ import base64 import ddddocr from selenium.webdriver.common.by import By +from selenium.webdriver.chrome.webdriver import WebDriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC @@ -25,7 +26,7 @@ class LibLogin(LibOperator): self, input_queue: queue.Queue, output_queue: queue.Queue, - driver: any + driver: WebDriver ): super().__init__(input_queue, output_queue) diff --git a/src/operators/LibLogout.py b/src/operators/LibLogout.py index 2a71bdc..c907835 100644 --- a/src/operators/LibLogout.py +++ b/src/operators/LibLogout.py @@ -10,8 +10,7 @@ See the LICENSE file for details. import queue from selenium.webdriver.common.by import By -from selenium.webdriver.support.ui import WebDriverWait -from selenium.webdriver.support import expected_conditions as EC +from selenium.webdriver.chrome.webdriver import WebDriver from base.LibOperator import LibOperator @@ -22,7 +21,7 @@ class LibLogout(LibOperator): self, input_queue: queue.Queue, output_queue: queue.Queue, - driver: any + driver: WebDriver ): super().__init__(input_queue, output_queue) diff --git a/src/operators/LibRenew.py b/src/operators/LibRenew.py index 36fc1e2..238addb 100644 --- a/src/operators/LibRenew.py +++ b/src/operators/LibRenew.py @@ -10,6 +10,7 @@ See the LICENSE file for details. import queue from selenium.webdriver.common.by import By +from selenium.webdriver.chrome.webdriver import WebDriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC @@ -22,7 +23,7 @@ class LibRenew(LibOperator): self, input_queue: queue.Queue, output_queue: queue.Queue, - driver: any + driver: WebDriver ): super().__init__(input_queue, output_queue) diff --git a/src/operators/LibReserve.py b/src/operators/LibReserve.py index 965424d..a58335b 100644 --- a/src/operators/LibReserve.py +++ b/src/operators/LibReserve.py @@ -12,6 +12,7 @@ import queue from datetime import datetime from selenium.webdriver.common.by import By +from selenium.webdriver.chrome.webdriver import WebDriver from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC @@ -24,7 +25,7 @@ class LibReserve(LibOperator): self, input_queue: queue.Queue, output_queue: queue.Queue, - driver: any + driver: WebDriver ): super().__init__(input_queue, output_queue)