mirror of
https://github.com/KenanZhu/AutoLibrary.git
synced 2026-06-20 16:33:03 +08:00
78 lines
1.8 KiB
Python
78 lines
1.8 KiB
Python
# -*- 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.
|
|
"""
|
|
from selenium.common.exceptions import (
|
|
NoSuchElementException,
|
|
StaleElementReferenceException,
|
|
)
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.remote.webdriver import WebDriver
|
|
|
|
from pages.components.Dialog import Dialog
|
|
|
|
|
|
class ReserveResultDialog(Dialog):
|
|
"""
|
|
Reservation result dialog shown after submitting a reserve request.
|
|
"""
|
|
|
|
ROOT = (By.CLASS_NAME, "layoutSeat")
|
|
|
|
def __init__(
|
|
self,
|
|
driver: WebDriver,
|
|
) -> None:
|
|
|
|
super().__init__(driver, self.ROOT, auto_close_on_exit=False)
|
|
|
|
def _titleLocator(
|
|
self,
|
|
) -> tuple:
|
|
|
|
return (By.CSS_SELECTOR, ".layoutSeat dt")
|
|
|
|
def getTitle(
|
|
self,
|
|
) -> str:
|
|
|
|
try:
|
|
return self._find(*self._titleLocator()).text
|
|
except (NoSuchElementException, StaleElementReferenceException):
|
|
return ""
|
|
|
|
def isSuccess(
|
|
self,
|
|
) -> bool:
|
|
|
|
title = self.getTitle()
|
|
return any(
|
|
kw in title
|
|
for kw in ("预定好了", "预约成功", "操作成功")
|
|
)
|
|
|
|
def isFailure(
|
|
self,
|
|
) -> bool:
|
|
|
|
contents = self.getDetailTexts()
|
|
return any(
|
|
"预约失败" in msg or "已有1个有效预约" in msg
|
|
for msg in contents
|
|
)
|
|
|
|
def getDetailTexts(
|
|
self,
|
|
) -> list[str]:
|
|
|
|
try:
|
|
elements = self._findAll(By.CSS_SELECTOR, ".layoutSeat dd")
|
|
return [el.text for el in elements if el.text.strip()]
|
|
except (NoSuchElementException, StaleElementReferenceException):
|
|
return []
|