Python文件下载函数优化 - click_download_file

def click_download_file(self, urn, path): '点击下载按钮,下载文件' downloaded_files_success = list() # 下载成功的文件列表 downloaded_files_failed = list() # 下载失败的文件列表 links = self.browser.safe_get_link(By.CSS_SELECTOR, ATTACHMENT_VIEW_WINDOW_ELEMENT) for link in links: # 获取文件名 文件大小 file_title, download_time_limit, file_size = self.file_title_data(link, urn) self.browser.click_with_link(link) self.download_alert() # 文件下载过大 logger.info( '

【urn = {}】文件名{}文件大小{},下载时间{}'.format(urn, file_title, file_size, download_time_limit)) time.sleep(download_time_limit) # 文件下载时间

    # 文件下载
    while True:
        status = self.download_file_is_successfully(path, file_title, file_size)
        if status:
            downloaded_files_success.append(file_title)  # 下载成功
            break
        else:
            self.browser.click_with_link(link)
            self.download_alert()  # 重试需要点击文件下载过大
            time.sleep(download_time_limit)
    
    if status:
        logger.info('

【urn = {}】文件{}重试下载成功'.format(urn, file_title)) else: logger.info(' 【urn = {}】文件{}重试下载失败'.format(urn, file_title)) downloaded_files_failed.append(file_title) # 下载失败

if downloaded_files_failed:
    logger.info('

【urn = {}】下载失败文件列表：{}'.format(urn, downloaded_files_failed)) raise FileException(message=ErrorMsg.NO_DOWNLOADABLE_FILES, file_list=downloaded_files_failed) if downloaded_files_success: logger.info(' 【urn = {}】下载成功文件列表：{}'.format(urn, downloaded_files_success)) new_file_title = self.modify_file(path, urn, downloaded_files_success) # 下载成功修改文件名 return new_file_title