1
1
mirror of https://github.com/KenanZhu/AutoLibrary.git synced 2026-06-18 15:33:03 +08:00

refactor(pages): 引入 Page Object 模式重构全部页面模块,变量统一为 snake_case

将原始 Selenium 操作脚本重构为三层 Page Object 架构:
- Page Objects(LoginPage/ReserveView/RecordsView/MainShell)
- Component Objects(Overlay 基类 + SeatMapOverlay/ReserveResultDialog 等对话框)
- Flow 状态机(ReserveFlow/CheckinFlow/RenewFlow)
- Services(CaptchaHandler/ReserveValidator/RecordChecker)

变量命名统一为 snake_case,方法名保持 camelCase,类名保持 PascalCase。

Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
2026-05-26 12:39:21 +08:00
parent 106463b9e5
commit 2226e8ac90
18 changed files with 3007 additions and 0 deletions
+266
View File
@@ -0,0 +1,266 @@
# -*- coding: utf-8 -*-
"""
Copyright (c) 2026 KenanZhu.
All rights reserved.
This software is provided "as is", without any warranty of any kind.
You may use, modify, and distribute this file under the terms of the MIT License.
See the LICENSE file for details.
"""
import time
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.remote.webdriver import WebDriver
from selenium.common.exceptions import (
ElementNotInteractableException,
NoSuchElementException,
StaleElementReferenceException,
TimeoutException,
)
from pages._dialogs import SeatMapOverlay, ReserveResultDialog
class ReserveView:
DATE_SELECT = (By.ID, "onDate_select")
DATE_OPTION_FMT = "p#options_onDate a[value='{value}']"
DATE_XPATH_FMT = "//p[@id='options_onDate']/a[@value='{value}']"
PLACE_SELECT = (By.ID, "display_building")
PLACE_OPTION_FMT = "p#options_building a[value='{value}']"
PLACE_XPATH_FMT = "//p[@id='options_building']/a[@value='{value}']"
FLOOR_SELECT = (By.ID, "floor_select")
FLOOR_OPTION_FMT = "p#options_floor a[value='{value}']"
FLOOR_XPATH_FMT = "//p[@id='options_floor']/a[@value='{value}']"
FIND_ROOM_BTN = (By.ID, "findRoom")
ROOM_BTN_FMT = "room_{room}"
SEAT_LAYOUT = (By.ID, "seatLayout")
SEAT_ITEMS = (By.CSS_SELECTOR, "li[id^='seat_']")
RESERVE_BTN = (By.ID, "reserveBtn")
START_TIME_OPTS = (By.CSS_SELECTOR, "#startTime ul li a")
END_TIME_OPTS = (By.CSS_SELECTOR, "#endTime ul li a")
RESULT_DIALOG = (By.CLASS_NAME, "layoutSeat")
RESULT_TITLE = (By.CSS_SELECTOR, ".layoutSeat dt")
RESULT_DETAIL = (By.CSS_SELECTOR, ".layoutSeat dd")
FLOOR_MAP = {"2": "二层", "3": "三层", "4": "四层", "5": "五层"}
ROOM_MAP = {
"1": "二层内环", "2": "二层西区", "3": "三层内环", "4": "三层外环",
"5": "四层内环", "6": "四层外环", "7": "四层期刊", "8": "五层考研",
}
def __init__(
self,
driver: WebDriver,
) -> None:
self._driver = driver
def selectDate(
self,
date_str: str,
) -> bool:
if self._clickOptionByJS(
trigger_id="onDate_select",
option_css=self.DATE_OPTION_FMT.format(value=date_str),
):
return True
return self._clickOption(
trigger=self.DATE_SELECT,
option=(By.XPATH, self.DATE_XPATH_FMT.format(value=date_str)),
)
def selectPlace(
self,
place: str = "1",
) -> bool:
if self._clickOptionByJS(
trigger_id="display_building",
option_css=self.PLACE_OPTION_FMT.format(value=place),
):
return True
return self._clickOption(
trigger=self.PLACE_SELECT,
option=(By.XPATH, self.PLACE_XPATH_FMT.format(value=place)),
)
def selectFloor(
self,
floor: str,
) -> bool:
if self._clickOptionByJS(
trigger_id="floor_select",
option_css=self.FLOOR_OPTION_FMT.format(value=floor),
):
return True
return self._clickOption(
trigger=self.FLOOR_SELECT,
option=(By.XPATH, self.FLOOR_XPATH_FMT.format(value=floor)),
)
def selectRoom(
self,
room: str,
) -> bool:
try:
WebDriverWait(self._driver, 2).until(
EC.element_to_be_clickable(self.FIND_ROOM_BTN)
).click()
except (TimeoutException, ElementNotInteractableException):
return False
except Exception:
return False
try:
WebDriverWait(self._driver, 2).until(
EC.element_to_be_clickable((By.ID, self.ROOM_BTN_FMT.format(room=room)))
).click()
return True
except (TimeoutException, ElementNotInteractableException):
return False
except Exception:
return False
def openSeatMap(
self,
) -> SeatMapOverlay:
return SeatMapOverlay(self._driver)
def selectSeat(
self,
seat_id: str,
) -> str | None:
try:
WebDriverWait(self._driver, 2).until(
EC.presence_of_element_located(self.SEAT_LAYOUT)
)
WebDriverWait(self._driver, 2).until(
EC.presence_of_all_elements_located(self.SEAT_ITEMS)
)
except TimeoutException:
return None
except Exception:
return None
try:
all_seats = self._driver.find_elements(*self.SEAT_ITEMS)
seat_id_upper = seat_id.lstrip('0').upper()
for seat in all_seats:
if not seat_id_upper == seat.text.lstrip('0'):
continue
seat_link = seat.find_element(By.TAG_NAME, "a")
WebDriverWait(self._driver, 2).until(
EC.element_to_be_clickable(seat_link)
)
seat_link.click()
return seat_link.get_attribute("title")
return None
except (NoSuchElementException, TimeoutException,
StaleElementReferenceException, ElementNotInteractableException):
return None
except Exception:
return None
def submitReserve(
self,
) -> bool:
try:
WebDriverWait(self._driver, 2).until(
EC.element_to_be_clickable(self.RESERVE_BTN)
).click()
return True
except (TimeoutException, ElementNotInteractableException):
return False
except Exception:
return False
def waitResultDialog(
self,
) -> ReserveResultDialog:
return ReserveResultDialog(self._driver)
def getAvailableTimeOptions(
self,
time_id: str,
) -> list:
try:
WebDriverWait(self._driver, 2).until(
EC.presence_of_all_elements_located(
(By.CSS_SELECTOR, f"#{time_id} ul li a")
)
)
except TimeoutException:
return []
except Exception:
return []
return self._driver.find_elements(
By.CSS_SELECTOR,
f"#{time_id} ul li a",
)
def refresh(
self,
) -> None:
self._driver.refresh()
def _clickOptionByJS(
self,
trigger_id: str,
option_css: str,
) -> bool:
script = f"""
try {{
var trigger = document.getElementById('{trigger_id}');
if (trigger) {{
trigger.click();
var option = document.querySelector("{option_css}");
if (option) {{
option.click();
return true;
}}
return false;
}}
return false;
}} catch (e) {{
return false;
}}
"""
result = self._driver.execute_script(script)
time.sleep(0.1)
return result
def _clickOption(
self,
trigger: tuple,
option: tuple,
) -> bool:
try:
WebDriverWait(self._driver, 2).until(
EC.element_to_be_clickable(trigger)
).click()
WebDriverWait(self._driver, 2).until(
EC.element_to_be_clickable(option)
).click()
return True
except (TimeoutException, ElementNotInteractableException):
return False
except Exception:
return False