利用python爬取深圳证券交易所财报

  开始爬取前需要安装一下 pandas 和 request 库,我用的是 pip 安装。代码如下:

pip install pandas
pip install requests

  安装完成后就要正式开始爬取数据了。

  打开深圳证券交易所官网:

  http://www.szse.cn/

   在信息披露->上市公司信息->上市公司公告这里可以找到所有上市公司的财报,也可以加上一些筛选条件,比如我这次爬取的就是2020.1.1-2020.12.31日之间的所有半年报。按F12打开开发者调试工具(我用的是chrome)。输入筛选条件。

           

   点查询后就可以看到浏览器向服务器发送的请求了,点开红线画出来的地方,如下。可以看到浏览器请求了一个Payload,里面有seDate等内容,多尝试几次可以发现一些规律。seDate里就是筛选的时间范围,bigCategoryId里面就是筛选的类别,可以是年度报告,季度报告等等。半年财报就是‘010303’。随便找到一个财报下载下来,可以在开发者调试工具里看到下载的链接,多找几个数据就可以发现他们有一个共同的头:‘http://disc.static.szse.cn/download/’。

 

  接下来只需要简简单单的写一个程序就好了。

# 定义爬取函数,参数为爬取第几页数据
def get_pdf_address(pageNum):
    url = 'http://www.szse.cn/api/disc/announcement/annList?random=%s' % random.random()
    #headers= {'User-Agent':str(UserAgent().random)}

    headers = {'Accept': 'application/json, text/javascript, */*; q=0.01'
    ,'Accept-Encoding': 'gzip, deflate'
    ,'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8'
    ,'Content-Type': 'application/json'
    ,'Host': 'www.szse.cn'
    ,'Origin': 'http://www.szse.cn'
    ,'Proxy-Connection': 'close'
    ,'Referer': 'http://www.szse.cn/disclosure/listed/fixed/index.html'
    ,'User-Agent': 'Mozilla/5.0(Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko)Chrome/74.0.3729.169 Safari/537.36'
    ,'X-Request-Type': 'ajax'
    ,'X-Requested-With': 'XMLHttpRequest'}

    pagenum = int(pageNum)
    payload = {"seDate":["2020-01-01","2020-12-31"],"channelCode":["fixed_disc"],"bigCategoryId":["010303"],"pageSize":30,"pageNum":pagenum}
    response = requests.post(url,headers =headers,data = json.dumps(payload)) #使用json格式
    result = response.json()
    return result

  具体爬取下来的内容可以尝试打印出来,这里就不放图了。我们需要把获得的数据整理一下,我用了pandas库,它可以直接写到excel里面,很方便。调用at()方法可以往里面加数据。这样我们就可以把股票代码,名称,下载链接等数据整理出来保存在excel里了。直接上完整版的代码。

'''
爬取深圳证券交易所财报地址
每一页有30个财报
每10页手动保存一次,防止被发现
'''
import requests
import time
import pandas as pd
import random
import os
import json

# 定义爬取函数
def get_pdf_address(pageNum):
    url = 'http://www.szse.cn/api/disc/announcement/annList?random=%s' % random.random()
    #headers= {'User-Agent':str(UserAgent().random)}

    headers = {'Accept': 'application/json, text/javascript, */*; q=0.01'
    ,'Accept-Encoding': 'gzip, deflate'
    ,'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8'
    ,'Content-Type': 'application/json'
    ,'Host': 'www.szse.cn'
    ,'Origin': 'http://www.szse.cn'
    ,'Proxy-Connection': 'close'
    ,'Referer': 'http://www.szse.cn/disclosure/listed/fixed/index.html'
    ,'User-Agent': 'Mozilla/5.0(Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko)Chrome/74.0.3729.169 Safari/537.36'
    ,'X-Request-Type': 'ajax'
    ,'X-Requested-With': 'XMLHttpRequest'}

    pagenum = int(pageNum)
    payload = {"seDate":["2020-01-01","2020-12-31"],"channelCode":["fixed_disc"],"bigCategoryId":["010303"],"pageSize":30,"pageNum":pagenum}
    response = requests.post(url,headers =headers,data = json.dumps(payload)) #使用json格式
    result = response.json()
    return result
 
#创建一个DataFrame储存爬取信息
pdf_infor= pd.DataFrame(columns =['secCode','secName','url','title','publishTime'])

# 下载域名的前缀
count = 0
url_head ='http://disc.static.szse.cn/download/'

#起始页数时page_a,我一次只爬了10页,所以截至页数是page_b = page_a + 10
page_a = 150
page_b = page_a + 10

path_xlsx = 'download_url_' + str(page_a) + '_' + str(page_b-1) + '.xlsx' #保存为excel的文件名

for i in range(page_a,page_b):

    print("爬取深交所年报下载地址第{}页".format(i))
    result = get_pdf_address(i)
    num = len(result['data'])
    for each in range(num):
        #each = 1
        pdf_infor.at[count,'secCode'] = result['data'][each]['secCode'][0]
        pdf_infor.at[count,'secName'] = result['data'][each]['secName'][0]
        pdf_infor.at[count,'url'] = url_head + result['data'][each]['attachPath']
        pdf_infor.at[count,'title'] = result['data'][each]['title']
        pdf_infor.at[count,'publishTime'] = result['data'][each]['publishTime']
        count += 1
    print('获取完成')
    time.sleep(random.uniform(2,3)) #控制访问速度
 
# 提取title中字符串获取年份
pdf_infor['Year'] = pdf_infor['title'].str.extract('([0-9]{4})')
pdf_infor.to_excel(path_xlsx) #保存为excel

  手动改一改page_a的值就可以得到所有的urli链接了,当然也可以写一个循环自动去十页十页的爬。我分开处理是为了防止出现问题,方便调试。在爬取url链接的时候还是很顺利的,服务器一次也没有拒绝我的请求。

 

 随便打开一个excel,可以看到密密麻麻的全是url链接。但是仔细看就会发现里面不止是财报,还有年度报告的摘要,所以我们小小的处理一下。

 

# pandsa删除摘要
#保存的时候会默认把股票代码开头的0去掉
import pandas as pd

file_list = os.listdir()
path_xlsx = '1_delete.xlsx'
pdf_infor = pd.read_excel('1.xlsx')

#print(type(pdf_infor.at[2,'title']))


for i in range(pdf_infor.shape[0]):
    zhaiyao = pdf_infor.at[i,'title']
    if zhaiyao.find('摘要') != -1:
        pdf_infor =  pdf_infor.drop(i)

pdf_infor.to_excel(path_xlsx)

  处理完成后会发现所有的股票代码前面的0都不见了,我是在所有财报下载完成后统一修改文件名的。处理方法有很多种,这也不是什么大问题。

  最后就是下载了,下载的时候会偶尔遇到服务器拒绝我的请求,所以只能用try方法,被拒绝的时候把没有成功下载的文件序号记下来,等全部跑完了再单独处理就好了。话不多说,上代码。

'''
pandas 读取数据批量下载
无法下载的保存在txt中,晚点单独下载
'''
import requests
import time
import pandas as pd
import random
import os
import json

file_path= "pdf_1/"

pdf_infor = pd.read_excel('1.xlsx')

headers ={'Upgrade-Insecure-Requests':'1',
        'User-Agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.71 Safari/537.36'}

#下载列表里第i个url
def download(i,f_url):
    f_url = pdf_infor.at[i,'url']
    return requests.get(f_url,headers = headers)

for each in range(pdf_infor.shape[0]):
    Stkcd = pdf_infor.at[each,'secCode']
    firm_name = pdf_infor.at[each,'secName'].replace("*","")
    Year = pdf_infor.at[each,'Year']
    pdf_url = pdf_infor.at[each,'url']

    file_name = "{}{}{}.pdf".format(Stkcd,firm_name,Year)
    file_full_name = os.path.join(file_path,file_name)
    print("开始下载{},股票代码{}的{}年报".format(firm_name,Stkcd,Year))

    try:
        time.sleep(random.uniform(4,6)) #控制访问速度
        rs = download(each,pdf_url)      
        with open(file_full_name, "wb") as fp:
            for chunk in rs.iter_content(chunk_size=10240):
                if chunk:
                    fp.write(chunk)
    except Exception as e:
        with open('log.txt', "a") as f:
            f.write(str(each)+'\n') #当然也可以直接把url给保存到错误日志里
        print(e)   

    print("===================下载完成==========================")

  最后看一下爬取的效果吧,这里就只剩下股票代码有点问题了。我干脆就全部给加上吧。代码如下

 

 

#因为之前去掉excel里摘要url的时候把所有股票代码开头的0都去掉了,现在补上
import os
import re

i = 'test'#保存pdf的路径
#补全开头的0
def fix_name(name):
	for i in range(6-len(name)):
		name = '0' + name
	return name

file_path = os.listdir(i)
for file in file_path:
	#print(file)
	stock_num_wrong = re.findall(r'\d+',file)#去掉年份提出数字

	rest_file = file[len(stock_num_wrong[0]):]#去掉提取出的数字

	stock_num_new = fix_name(str(stock_num_wrong[0]))#第一个是股票代码第二个是年份

	file_name_new = stock_num_new + rest_file
	os.rename(os.path.join(i,file),os.path.join(i,file_name_new))
	print(rest_file)

  补全以后就大功告成了。当然这个方法有挺多地方可以整合改进的,但是我太懒了,反正数据也有了,就这样吧!

 

 

  

posted @ 2021-10-26 10:08  流心蛋黄酥  阅读(3001)  评论(1)    收藏  举报