python 爬取百度文库并下载(免费文章限定)_Python

python 爬取百度文库并下载(免费文章限定)

2021-08-08 00:14Jack-Cherish Python

这篇文章主要介绍了python 爬取百度文库并下载的示例，帮助大家更好的理解和学习python 爬虫的相关知识，感兴趣的朋友可以了解下

				?

									import requests

									import re

									import json

									import os

									session = requests.session()

									def fetch_url(url):

									 return session.get(url).content.decode('gbk')

									def get_doc_id(url):

									 return re.findall('view/(.*).html', url)[0]

									def parse_type(content):

									 return re.findall(r"docType.*?\:.*?\'(.*?)\'\,", content)[0]

									def parse_title(content):

									 return re.findall(r"title.*?\:.*?\'(.*?)\'\,", content)[0]

									def parse_doc(content):

									 result = ''

									 url_list = re.findall('(https.*?0.json.*?)\\\\x22}', content)

									 url_list = [addr.replace("\\\\\\/", "/") for addr in url_list]

									 for url in url_list[:-5]:

									  content = fetch_url(url)

									  y = 0

									  txtlists = re.findall('"c":"(.*?)".*?"y":(.*?),', content)

									  for item in txtlists:

									   if not y == item[1]:

									    y = item[1]

									    n = '\n'

									   else:

									    n = ''

									   result += n

									   result += item[0].encode('utf-8').decode('unicode_escape', 'ignore')

									 return result

									def parse_txt(doc_id):

									 content_url = 'https://wenku.baidu.com/api/doc/getdocinfo?callback=cb&doc_id=' + doc_id

									 content = fetch_url(content_url)

									 md5 = re.findall('"md5sum":"(.*?)"', content)[0]

									 pn = re.findall('"totalPageNum":"(.*?)"', content)[0]

									 rsign = re.findall('"rsign":"(.*?)"', content)[0]

									 content_url = 'https://wkretype.bdimg.com/retype/text/' + doc_id + '?rn=' + pn + '&type=txt' + md5 + '&rsign=' + rsign

									 content = json.loads(fetch_url(content_url))

									 result = ''

									 for item in content:

									  for i in item['parags']:

									   result += i['c'].replace('\\r', '\r').replace('\\n', '\n')

									 return result

									def parse_other(doc_id):

									 content_url = "https://wenku.baidu.com/browse/getbcsurl?doc_id=" + doc_id + "&pn=1&rn=99999&type=ppt"

									 content = fetch_url(content_url)

									 url_list = re.findall('{"zoom":"(.*?)","page"', content)

									 url_list = [item.replace("\\", '') for item in url_list]

									 if not os.path.exists(doc_id):

									  os.mkdir(doc_id)

									 for index, url in enumerate(url_list):

									  content = session.get(url).content

									  path = os.path.join(doc_id, str(index) + '.jpg')

									  with open(path, 'wb') as f:

									   f.write(content)

									 print("图片保存在" + doc_id + "文件夹")

									def save_file(filename, content):

									 with open(filename, 'w', encoding='utf8') as f:

									  f.write(content)

									  print('已保存为:' + filename)

									# test_txt_url = 'https://wenku.baidu.com/view/cbb4af8b783e0912a3162a89.html?from=search'

									# test_ppt_url = 'https://wenku.baidu.com/view/2b7046e3f78a6529657d5376.html?from=search'

									# test_pdf_url = 'https://wenku.baidu.com/view/dd6e15c1227916888586d795.html?from=search'

									# test_xls_url = 'https://wenku.baidu.com/view/eb4a5bb7312b3169a551a481.html?from=search'

									def main():

									 url = input('请输入要下载的文库URL地址')

									 content = fetch_url(url)

									 doc_id = get_doc_id(url)

									 type = parse_type(content)

									 title = parse_title(content)

									 if type == 'doc':

									  result = parse_doc(content)

									  save_file(title + '.txt', result)

									 elif type == 'txt':

									  result = parse_txt(doc_id)

									  save_file(title + '.txt', result)

									 else:

									  parse_other(doc_id)

									if __name__ == "__main__":

									 main()