1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85
| import requests import os from hashlib import md5 from requests.exceptions import RequestException from bs4 import BeautifulSoup
headers = {'If-None-Match': 'W/"5cc2cd8f-2c58"', "Referer": "http://www.mzitu.com/all/", 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/50.0.2661.102 UBrowser/6.1.2107.204 SafarMozilla/5.0 (Windows NT 6.1) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/71.0.3578.98 Safari/537.36'}
def get_page(url): try: response = requests.get(url, headers=headers) if response.status_code == 200: return response.text return None except RequestException: print('获取索引页失败') return None
def parse_page(html): soup = BeautifulSoup(html, 'lxml') items = soup.select('#pins li') for link in items: href = link.a['href'] get_detail_page(href)
def get_detail_page(href): for i in range(1,100): detail_url = href + '/' + str(i) if requests.get(detail_url, headers=headers).status_code == 200: parse_detail_page(detail_url) else: print('已至末尾页') return None
def parse_detail_page(detail_url): try: response = requests.get(detail_url, headers=headers) if response.status_code == 200: print('获取详情页成功') detail_html = response.text get_image(detail_html) return None except RequestException: print('获取详情页失败') return None
def get_image(detail_html): soup = BeautifulSoup(detail_html, 'lxml') items= soup.select('.main-image') for item in items: image = item.img['src'] save_image(image)
def save_image(image): response = requests.get(image,headers=headers) if response.status_code == 200: data = response.content file_path = '{0}/{1}.{2}'.format(os.getcwd(), md5(data).hexdigest(), 'jpg') print(file_path) if not os.path.exists(file_path): with open(file_path, 'wb') as f: f.write(data) f.close() print('保存成功') else: print('保存失败')
def main(): url = 'https://www.mzitu.com/tag/youhuo/' html = get_page(url) parse_page(html)
if __name__ == '__main__': main()
|
v1.5.2