张翔
146 lines
5.2 KiB
Python
146 lines
5.2 KiB
Python
"""
|
|
日历页面
|
|
|
|
Uniapp日历页面的页面对象模型。
|
|
"""
|
|
|
|
from playwright.sync_api import Page
|
|
from ..base_page import BasePage
|
|
|
|
|
|
class CalendarPage(BasePage):
|
|
"""日历页面"""
|
|
|
|
# 页面路径
|
|
PATH = "/pages/calendar/index"
|
|
|
|
# 元素定位器
|
|
LOCATORS = {
|
|
"page": ".page",
|
|
"calendar_grid": ".calendar-grid-container",
|
|
"calendar_card": ".calendar-card",
|
|
"month_title": ".month-title",
|
|
"year_display": ".year-display",
|
|
"prev_month_btn": ".nav-button",
|
|
"next_month_btn": ".nav-button",
|
|
"today_btn": ".today-button",
|
|
"day_number": ".day-number",
|
|
"day_lunar": ".day-lunar",
|
|
"selected_date": ".calendar-card-selected",
|
|
"current_date": ".calendar-card-today",
|
|
"weekday_header": ".weekday",
|
|
"tab_calendar": ".nav-item:nth-child(1)",
|
|
"tab_almanac": ".nav-item:nth-child(2)",
|
|
"tab_user": ".nav-item:nth-child(3)",
|
|
}
|
|
|
|
def __init__(self, page: Page, base_url: str = ""):
|
|
super().__init__(page, base_url)
|
|
|
|
def navigate(self, path: str = "") -> None:
|
|
"""导航到日历页面"""
|
|
target_path = path if path else self.PATH
|
|
self.page.goto(f"{self.base_url}{target_path}")
|
|
self.wait_for_load()
|
|
|
|
def is_loaded(self) -> bool:
|
|
"""检查页面是否加载完成"""
|
|
return self.is_element_visible(self.LOCATORS["calendar_grid"])
|
|
|
|
def get_month_display(self) -> str:
|
|
"""获取月份显示"""
|
|
return self.get_text(self.LOCATORS["month_title"])
|
|
|
|
def get_year_display(self) -> str:
|
|
"""获取年份显示"""
|
|
month_text = self.get_text(self.LOCATORS["month_title"])
|
|
if month_text:
|
|
import re
|
|
match = re.search(r'(\d{4})年', month_text)
|
|
if match:
|
|
return match.group(1)
|
|
return ""
|
|
|
|
def click_prev_month(self) -> None:
|
|
"""点击上一月"""
|
|
nav_buttons = self.page.locator(self.LOCATORS["prev_month_btn"]).all()
|
|
if len(nav_buttons) > 0:
|
|
nav_buttons[0].click()
|
|
|
|
def click_next_month(self) -> None:
|
|
"""点击下一月"""
|
|
nav_buttons = self.page.locator(self.LOCATORS["next_month_btn"]).all()
|
|
if len(nav_buttons) > 1:
|
|
nav_buttons[1].click()
|
|
|
|
def click_today(self) -> None:
|
|
"""点击今天按钮"""
|
|
self.click_element(self.LOCATORS["today_btn"])
|
|
|
|
def click_date(self, day: int) -> None:
|
|
"""点击指定日期"""
|
|
calendar_cards = self.page.locator(self.LOCATORS["calendar_card"]).all()
|
|
for card in calendar_cards:
|
|
day_number = card.locator(self.LOCATORS["day_number"]).text_content()
|
|
if day_number and int(day_number) == day:
|
|
card.click()
|
|
break
|
|
|
|
def get_date_cells_count(self) -> int:
|
|
"""获取日期单元格数量"""
|
|
return self.get_elements_count(self.LOCATORS["calendar_card"])
|
|
|
|
def get_selected_date(self) -> str:
|
|
"""获取选中的日期"""
|
|
import time
|
|
time.sleep(1)
|
|
selected = self.page.locator(self.LOCATORS["selected_date"]).first
|
|
if selected and selected.count() > 0:
|
|
day_number = selected.locator(self.LOCATORS["day_number"])
|
|
if day_number.count() > 0:
|
|
return day_number.text_content() or ""
|
|
return ""
|
|
|
|
def has_lunar_text(self, day: int) -> bool:
|
|
"""检查指定日期是否有农历显示"""
|
|
calendar_cards = self.page.locator(self.LOCATORS["calendar_card"]).all()
|
|
for card in calendar_cards:
|
|
day_number = card.locator(self.LOCATORS["day_number"]).text_content()
|
|
if day_number and int(day_number) == day:
|
|
return card.locator(self.LOCATORS["day_lunar"]).count() > 0
|
|
return False
|
|
|
|
def get_lunar_text(self, day: int) -> str:
|
|
"""获取指定日期的农历文本"""
|
|
calendar_cards = self.page.locator(self.LOCATORS["calendar_card"]).all()
|
|
for card in calendar_cards:
|
|
day_number = card.locator(self.LOCATORS["day_number"]).text_content()
|
|
if day_number and int(day_number) == day:
|
|
return card.locator(self.LOCATORS["day_lunar"]).text_content() or ""
|
|
return ""
|
|
|
|
def click_tab_calendar(self) -> None:
|
|
"""点击万年历Tab"""
|
|
self.click_element(self.LOCATORS["tab_calendar"])
|
|
|
|
def click_tab_almanac(self) -> None:
|
|
"""点击黄历Tab"""
|
|
self.click_element(self.LOCATORS["tab_almanac"])
|
|
|
|
def click_tab_user(self) -> None:
|
|
"""点击我的Tab"""
|
|
self.click_element(self.LOCATORS["tab_user"])
|
|
|
|
def has_holiday_mark(self, day: int) -> bool:
|
|
"""检查指定日期是否有节假日标记"""
|
|
calendar_cards = self.page.locator(self.LOCATORS["calendar_card"]).all()
|
|
for card in calendar_cards:
|
|
day_number = card.locator(self.LOCATORS["day_number"]).text_content()
|
|
if day_number and int(day_number) == day:
|
|
return card.locator(self.LOCATORS["current_date"]).count() > 0
|
|
return False
|
|
|
|
def wait_for_calendar_load(self, timeout: int = 10000) -> None:
|
|
"""等待日历加载完成"""
|
|
self.wait_for_selector(self.LOCATORS["calendar_card"], timeout=timeout)
|