利用Python实现批量下载上市公司财务报表

3、确保把IE的默认下载路径改成本文件所在目录

4、下载后立即最小化运行窗口，避免遮挡屏幕导致pyautogui无法定位图像位置没有安装的话可以在命令提示符里面输入“pip install 库名”进行安装：pip install pyautogui pip install pywin32 pip install pandas pip install xlwings"""

import pyautogui from win32com.client import DispatchEx import pandas import xlwings import time import os

# 同花顺网站下载{链接的固定字段:文件名固定字段}ref = {'main&type=report': 'main_report.xls','main&type=year': 'main_year.xls','main&type=simple': 'main_simple.xls','debt&type=report': 'debt_report.xls','debt&type=year': 'debt_year.xls','benefit&type=report': 'benefit_report.xls','benefit&type=year': 'benefit_year.xls','benefit&type=simple': 'benefit_simple.xls','cash&type=report': 'cash_report.xls','cash&type=year': 'cash_year.xls','cash&type=simple': 'cash_simple.xls'}

df = pandas.read_excel(r'./从THS批量下载上市公司财报.xlsm', sheet_name='Main', dtype='str', header=0)

# 把表格中不足6位的公司代码补全成6位，比如把'2'补全成'000002'df['公司代码'] = df['公司代码'].apply(lambda x: str('000000')[:(6 - len(x))] + str(x))

# 创建一个{下载文件名：下载链接}的字典urls = {}for i in ref.keys():for j in df['公司代码']:# 如果原来下载过就略过以节约时间if not os.path.exists(f'{j}_{ref[i]}'):urls[f'{j}_{ref[i]}'] = str(f'http://basic.10jqka.com.cn/api/stock/export.php?export={i}&code={j}')

wb = xlwings.App(visible=False, add_book=False).books.open(r'./从THS批量下载上市公司财报.xlsm')

# 这里用Excel宏的一个退出IE功能,比win32com的功能更好用# 也是因为本人技术太渣，不知道怎么用Python完全退出IE# QuitIE的宏代码在后面QuitIE = wb.macro('QuitIE')

# 这里用Excel宏的XMLHTTP功能，下载效率更高# XMLHTTP的宏代码在后面XMLHTTP = wb.macro('XMLHTTP')

# IE浏览器下载button的截图img = r'./capture.png'

def IEDownload(url):ie = DispatchEx('InternetExplorer.Application') ie.Navigate(url)

# 最多尝试查找5次，避免死循环times = 0 while times < 5:location = pyautogui.locateCenterOnScreen(img, confidence=0.9) if location is not None:pyautogui.click(location.x, location.y, clicks=1, button='left', duration=0.01, interval=0.01) break times += 1

windows = 0 for filename in urls:# 每7次调用一次xmlhttp，胆子大可以把这个值设小点if windows % 7 == 0:XMLHTTP(filename, urls[filename]) if not os.path.exists(f'./{filename}'):IEDownload(urls[filename]) windows += 1 else:IEDownload(urls[filename]) windows += 1# 每7次关闭IE的所有窗口，释放内存if windows % 7 == 0:time.sleep(0.05) QuitIE() time.sleep(0.05)

QuitIE() wb.close() xlwings.App().quit()