343 lines
14 KiB
Python
343 lines
14 KiB
Python
# test_more_download_page.py
|
||
from appium.webdriver.common.appiumby import AppiumBy
|
||
from selenium.webdriver.support.ui import WebDriverWait
|
||
from selenium.webdriver.support import expected_conditions as EC
|
||
from selenium.common.exceptions import TimeoutException, NoSuchElementException
|
||
import logging
|
||
import time
|
||
|
||
class MoreDownloadPage:
|
||
def __init__(self, driver, wait,device_id):
|
||
self.driver = driver
|
||
self.wait = wait
|
||
self.device_id = device_id
|
||
self.logger = logging.getLogger(__name__)
|
||
|
||
def is_on_more_download_page(self):
|
||
"""通过下载历史数据按钮来判断是否在更多下载页面"""
|
||
try:
|
||
# 使用下载历史数据按钮的resource-id来检查
|
||
download_history_locator = (AppiumBy.ID, "com.bjjw.cjgc:id/download_history")
|
||
self.wait.until(EC.presence_of_element_located(download_history_locator))
|
||
self.logger.info("已确认在更多下载页面")
|
||
return True
|
||
except TimeoutException:
|
||
self.logger.warning("未找到下载历史数据按钮，不在更多下载页面")
|
||
return False
|
||
except Exception as e:
|
||
self.logger.error(f"检查更多下载页面时发生意外错误: {str(e)}")
|
||
return False
|
||
|
||
def click_download_button(self):
|
||
"""点击下载按钮"""
|
||
try:
|
||
# 点击下载历史数据按钮
|
||
download_button = self.wait.until(
|
||
EC.element_to_be_clickable((AppiumBy.ID, "com.bjjw.cjgc:id/download_history"))
|
||
)
|
||
download_button.click()
|
||
self.logger.info("已点击下载历史数据按钮")
|
||
|
||
# 等待下载操作开始
|
||
# time.sleep(3)
|
||
|
||
return True
|
||
|
||
except TimeoutException:
|
||
self.logger.error("等待下载按钮可点击超时")
|
||
return False
|
||
except Exception as e:
|
||
self.logger.error(f"点击下载按钮时出错: {str(e)}")
|
||
return False
|
||
|
||
def click_download_original_data(self):
|
||
"""点击下载原始数据按钮并处理日期选择"""
|
||
try:
|
||
# 点击下载原始数据按钮
|
||
download_original_btn = self.wait.until(
|
||
EC.element_to_be_clickable((AppiumBy.ID, "com.bjjw.cjgc:id/download_org"))
|
||
)
|
||
download_original_btn.click()
|
||
self.logger.info("已点击下载原始数据按钮")
|
||
|
||
# 等待日期选择弹窗出现
|
||
# time.sleep(2)
|
||
|
||
# 点击选择开始日期
|
||
start_date_btn = self.wait.until(
|
||
EC.element_to_be_clickable((AppiumBy.ID, "com.bjjw.cjgc:id/date"))
|
||
)
|
||
start_date_btn.click()
|
||
self.logger.info("已点击选择开始日期")
|
||
|
||
# 等待日期选择器出现
|
||
# time.sleep(2)
|
||
|
||
# 滑动年份选择器 - 向上滑动1/5的距离
|
||
if not self._swipe_year_wheel():
|
||
self.logger.error("滑动年份选择器失败")
|
||
return False
|
||
|
||
# 点击日期选择器的确定按钮
|
||
confirm_btn = self.wait.until(
|
||
EC.element_to_be_clickable((AppiumBy.ID, "com.bjjw.cjgc:id/okBtn"))
|
||
)
|
||
confirm_btn.click()
|
||
self.logger.info("已确认日期选择")
|
||
|
||
# 等待日期选择器关闭
|
||
# time.sleep(2)
|
||
|
||
# 假设弹窗有确定按钮，点击它开始下载
|
||
try:
|
||
# 尝试查找并点击下载弹窗的确定按钮
|
||
download_confirm_btn = WebDriverWait(self.driver, 5).until(
|
||
EC.element_to_be_clickable((AppiumBy.XPATH, "//android.widget.Button[contains(@text, '确定') or contains(@text, '下载')]"))
|
||
)
|
||
download_confirm_btn.click()
|
||
self.logger.info("已点击下载确认按钮")
|
||
except TimeoutException:
|
||
self.logger.warning("未找到下载确认按钮，可能不需要确认")
|
||
|
||
# 等待下载开始
|
||
# time.sleep(3)
|
||
|
||
return True
|
||
|
||
except TimeoutException:
|
||
self.logger.error("等待下载原始数据按钮可点击超时")
|
||
return False
|
||
except Exception as e:
|
||
self.logger.error(f"点击下载原始数据时出错: {str(e)}")
|
||
return False
|
||
|
||
def click_download_result_data(self):
|
||
"""点击下载成果数据按钮并处理日期选择"""
|
||
try:
|
||
# 点击下载成果数据按钮
|
||
download_result_btn = self.wait.until(
|
||
EC.element_to_be_clickable((AppiumBy.ID, "com.bjjw.cjgc:id/download_result"))
|
||
)
|
||
download_result_btn.click()
|
||
self.logger.info("已点击下载成果数据按钮")
|
||
|
||
# 等待日期选择弹窗出现
|
||
# time.sleep(2)
|
||
|
||
# 点击选择开始日期
|
||
start_date_btn = self.wait.until(
|
||
EC.element_to_be_clickable((AppiumBy.ID, "com.bjjw.cjgc:id/date"))
|
||
)
|
||
start_date_btn.click()
|
||
self.logger.info("已点击选择开始日期")
|
||
|
||
# 等待日期选择器出现
|
||
# time.sleep(2)
|
||
|
||
# 滑动年份选择器 - 向上滑动1/5的距离
|
||
if not self._swipe_year_wheel():
|
||
self.logger.error("滑动年份选择器失败")
|
||
return False
|
||
|
||
# 点击日期选择器的确定按钮
|
||
confirm_btn = self.wait.until(
|
||
EC.element_to_be_clickable((AppiumBy.ID, "com.bjjw.cjgc:id/okBtn"))
|
||
)
|
||
confirm_btn.click()
|
||
self.logger.info("已确认日期选择")
|
||
|
||
# 等待日期选择器关闭
|
||
# time.sleep(2)
|
||
|
||
# 假设弹窗有确定按钮，点击它开始下载
|
||
try:
|
||
# 尝试查找并点击下载弹窗的确定按钮
|
||
download_confirm_btn = WebDriverWait(self.driver, 5).until(
|
||
EC.element_to_be_clickable((AppiumBy.XPATH, "//android.widget.Button[contains(@text, '确定') or contains(@text, '下载')]"))
|
||
)
|
||
download_confirm_btn.click()
|
||
self.logger.info("已点击下载确认按钮")
|
||
except TimeoutException:
|
||
self.logger.warning("未找到下载确认按钮，可能不需要确认")
|
||
|
||
# 等待下载开始
|
||
# time.sleep(3)
|
||
|
||
return True
|
||
|
||
except TimeoutException:
|
||
self.logger.error("等待下载成果数据按钮可点击超时")
|
||
return False
|
||
except Exception as e:
|
||
self.logger.error(f"点击下载成果数据时出错: {str(e)}")
|
||
return False
|
||
|
||
def _swipe_year_wheel(self):
|
||
"""滑动年份选择器的滚轮"""
|
||
try:
|
||
# 获取年份选择器滚轮元素
|
||
year_wheel = self.driver.find_element(AppiumBy.ID, "com.bjjw.cjgc:id/wheelView1")
|
||
|
||
# 获取滚轮的位置和尺寸
|
||
location = year_wheel.location
|
||
size = year_wheel.size
|
||
|
||
# 计算滚轮中心点坐标
|
||
center_x = location['x'] + size['width'] // 2
|
||
center_y = location['y'] + size['height'] // 2
|
||
|
||
# 计算滑动距离 - 滚轮高度的1/5
|
||
swipe_distance = size['height'] // 5
|
||
|
||
# 执行滑动操作 - 从中心向上滑动1/5高度
|
||
self.driver.swipe(center_x, center_y - swipe_distance, center_x, center_y, 500)
|
||
|
||
self.logger.info("已滑动年份选择器")
|
||
return True
|
||
|
||
except Exception as e:
|
||
self.logger.error(f"滑动年份选择器时出错: {str(e)}")
|
||
return False
|
||
|
||
def wait_for_loading_dialog(self, timeout=900):
|
||
"""
|
||
检查特定结构的加载弹窗的出现和消失
|
||
|
||
参数:
|
||
timeout: 最大等待时间，默认10分钟(600秒)
|
||
|
||
返回:
|
||
bool: 如果加载弹窗出现并消失返回True，否则返回False
|
||
"""
|
||
try:
|
||
self.logger.info("开始检查加载弹窗...")
|
||
|
||
# 首先检查加载弹窗是否出现
|
||
start_time = time.time()
|
||
loading_appeared = False
|
||
|
||
# 等待加载弹窗出现（最多等待30秒）
|
||
while time.time() - start_time < 30:
|
||
try:
|
||
# 根据提供的结构查找加载弹窗
|
||
# 查找包含ProgressBar和"loading..."文本的弹窗
|
||
loading_indicators = [
|
||
(AppiumBy.XPATH, "//android.widget.FrameLayout[@resource-id='android:id/content']/android.widget.LinearLayout[@resource-id='android:id/parentPanel']//android.widget.ProgressBar"),
|
||
(AppiumBy.XPATH, "//android.widget.TextView[@resource-id='android:id/message' and @text='loading...']"),
|
||
(AppiumBy.XPATH, "//android.widget.FrameLayout[@resource-id='android:id/content']//android.widget.ProgressBar"),
|
||
(AppiumBy.XPATH, "//*[contains(@text, 'loading...')]")
|
||
]
|
||
|
||
for by, value in loading_indicators:
|
||
try:
|
||
element = self.driver.find_element(by, value)
|
||
if element.is_displayed():
|
||
loading_appeared = True
|
||
self.logger.info("数据下载已开始")
|
||
self.logger.info("检测到加载弹窗出现")
|
||
break
|
||
except:
|
||
continue
|
||
|
||
if loading_appeared:
|
||
break
|
||
|
||
except Exception as e:
|
||
pass
|
||
|
||
time.sleep(1)
|
||
|
||
# 如果加载弹窗没有出现，直接返回True
|
||
if not loading_appeared:
|
||
self.logger.info("未检测到加载弹窗，继续执行")
|
||
return True
|
||
|
||
# 等待加载弹窗消失
|
||
self.logger.info("等待加载弹窗消失...")
|
||
disappearance_start_time = time.time()
|
||
|
||
while time.time() - disappearance_start_time < timeout:
|
||
try:
|
||
# 检查加载弹窗是否还存在
|
||
loading_still_exists = False
|
||
|
||
for by, value in loading_indicators:
|
||
try:
|
||
element = self.driver.find_element(by, value)
|
||
if element.is_displayed():
|
||
loading_still_exists = True
|
||
break
|
||
except:
|
||
continue
|
||
|
||
if not loading_still_exists:
|
||
self.logger.info("加载弹窗已消失")
|
||
return True
|
||
|
||
# 每1分钟记录一次状态
|
||
if int(time.time() - disappearance_start_time) % 60 == 0:
|
||
elapsed_time = int(time.time() - disappearance_start_time)
|
||
self.logger.info(f"加载弹窗仍在显示，已等待{elapsed_time//60}分钟")
|
||
|
||
except Exception as e:
|
||
# 如果出现异常，可能弹窗已经消失
|
||
self.logger.info("加载弹窗可能已消失")
|
||
return True
|
||
|
||
time.sleep(1)
|
||
|
||
# 如果超时，记录错误并返回False
|
||
self.logger.error(f"加载弹窗在{timeout}秒后仍未消失")
|
||
return False
|
||
|
||
except Exception as e:
|
||
self.logger.error(f"检查加载弹窗时出错: {str(e)}")
|
||
return False
|
||
|
||
def more_download_page_manager(self):
|
||
"""执行更多下载页面管理操作"""
|
||
try:
|
||
self.logger.info("开始执行更多下载页面操作")
|
||
|
||
# 检查是否在更多下载页面
|
||
if not self.is_on_more_download_page():
|
||
self.logger.error("不在更多下载页面")
|
||
return False
|
||
|
||
# 点击下载历史数据按钮
|
||
if not self.click_download_button():
|
||
self.logger.error("点击下载历史数据按钮失败")
|
||
return False
|
||
|
||
# 等待下载历史数据页面加载完成
|
||
# time.sleep(3)
|
||
|
||
# 点击下载原始数据按钮
|
||
if not self.click_download_original_data():
|
||
self.logger.error("点击下载原始数据按钮失败")
|
||
return False
|
||
|
||
# 等待下载操作完成
|
||
time.sleep(1)
|
||
|
||
# 使用wait_for_loading_dialog函数等待下载过程中的加载弹窗消失
|
||
if not self.wait_for_loading_dialog():
|
||
self.logger.warning("下载过程中的加载弹窗未在预期时间内消失，但操作已完成")
|
||
|
||
# 等待一段时间，确保原始数据下载完成
|
||
time.sleep(1)
|
||
|
||
# 点击下载成果数据按钮
|
||
if not self.click_download_result_data():
|
||
self.logger.error("点击下载成果数据按钮失败")
|
||
return False
|
||
|
||
# 使用wait_for_loading_dialog函数等待下载过程中的加载弹窗消失
|
||
if not self.wait_for_loading_dialog():
|
||
self.logger.warning("成果数据下载过程中的加载弹窗未在预期时间内消失，但操作已完成")
|
||
|
||
self.logger.info("更多下载页面操作执行完成")
|
||
return True
|
||
except Exception as e:
|
||
self.logger.error(f"执行更多下载页面操作时出错: {str(e)}")
|
||
return False |