微信公众号文章采集工具，可采集文章文字内容信息及图片

#微信文章页采集
# -*- coding: UTF-8 -*-
#by 微信：huguo00289
import requests
import re,time,os
from bs4 import BeautifulSoup
from baocun import bctp,bcwb
from fake_useragent import UserAgent

def ua():
    ua = UserAgent()
    headers = {"User-Agent": ua.random}
    return headers

def get_content(url):
    headers=ua()
    respnese=requests.get(url,headers=headers).text
    soup=BeautifulSoup(respnese,'lxml')
    #获取标题
    h2=soup.find('h2',class_="rich_media_title").get_text()
    h2=h2.replace('\n','')
    h2 = h2.replace(' ', '')
    h2 = re.sub(r'[\|\/\<\>\:\*\?\\\"]', "_", h2)  # 剔除不合法字符
    print(f'微信公众号文章标题：{h2}')
    os.makedirs(f'weixin/{h2}/',exist_ok=True)
    lj =f'weixin/{h2}/'
    ljj=f'weixin/{h2}/{h2}.txt'
    author=soup.find('div',class_="rich_media_meta_list").find('a',id="js_name").get_text()
    author = author.replace('\n', '')
    author = author.replace(' ', '')
    author=f'来源：{author}'
    print(author)
    i=1
    text=''
    ps=soup.find('div',class_="rich_media_content").find_all('p')
    for p in ps:
        if "img" in str(p):
            try:
                img_url=p.find('img')['data-src']
                print(img_url)
                if "jpeg" == img_url[-4:]:
                    img_name=f'{i}.{img_url[-4:]}'
                else:
                    img_name = f'{i}.{img_url[-3:]}'
                bctp(lj, img_url, img_name)
                p_content=img_name
                i=i+1
            except Exception as e:
                print(f"获取图片数据失败，错误代码：{e}")
                pass
        else:
            p_content=p.get_text()
        text = '%s%s%s' % (text, '\n',p_content)
    texts='%s%s%s%s%s'%(h2,'\n',author,'\n',text)
    print(texts)
    bcwb(ljj, texts)

if __name__ == '__main__':
    url=input("请输入要采集的微信公众号文章链接:")
    print(f'爬虫启动中，请稍后......')
    get_content(url)
    print(f'采集完毕，程序5s后自动关闭！')
    time.sleep(5)

#微信公众号文章 完整版内容抓取
#by 微信：huguo00289
def cs(url):
    headers = ua()
    respnese = requests.get(url, headers=headers).text
    soup = BeautifulSoup(respnese, 'lxml')
    get_article(soup)


def get_article(soup):
    # 获取标题
    h2 = soup.find('h2', class_="rich_media_title").get_text()
    h2 = h2.replace('\n', '')
    h2 = h2.replace(' ', '')
    h2 = re.sub(r'[\|\/\<\>\:\*\?\\\"]', "_", h2)  # 剔除不合法字符
    print(f'微信公众号文章标题：{h2}')
    os.makedirs(f'weixin/{h2}/', exist_ok=True)
    lj = f'weixin/{h2}/'
    ljj = f'weixin/{h2}/{h2}.txt'
    author = soup.find('div', class_="rich_media_meta_list").find('a', id="js_name").get_text()
    author = author.replace('\n', '')
    author = author.replace(' ', '')
    author = f'来源：{author}'
    print(author)
    # 获取文字内容
    texts = soup.find('div', class_="rich_media_content").get_text()
    texts = '\n'.join(texts.split('。'))  # 以句号 分割文本
    print(texts)
    texts = '%s%s%s%s%s' % (h2, '\n', author, '\n', texts)
    print(texts)
    bcwb(ljj, texts)
    # 获取所有图片
    i = 1
    imgs = soup.find('div', class_="rich_media_content").find_all('img')
    for img in imgs:
        img_url = img['data-src']
        print(img_url)
        if "jpeg" == img_url[-4:]:
            img_name = f'{i}.{img_url[-4:]}'
        else:
            img_name = f'{i}.{img_url[-3:]}'
        bctp(lj, img_url, img_name)
        i = i + 1

#by 微信：huguo00289
#存储内容

import requests
import time
from fake_useragent import UserAgent

def ua():
    ua = UserAgent()
    headers = {"User-Agent": ua.random}
    return headers

#下载图片
def bctp(lj,img_url,img_name):
    print("开始下载图片！")
    try:
        r = requests.get(img_url,headers=ua(),timeout=5)
        with open(f'{lj}/{img_name}', 'wb') as f:
            f.write(r.content)
            print(f'下载{img_name}图片成功！')
            time.sleep(1)
    except Exception as e:
        if "port=443): Read timed out" in str(e):
            time.sleep(2)
            try:
                r = requests.get(img_url, headers=ua(),timeout=5)
                with open(f'{lj}/{img_name}', 'wb') as f:
                    f.write(r.content)
                    print(f'下载{img_name}图片成功！')
            except Exception as e:
                print(f'下载{img_name}图片失败！')
                print(f'错误代码：{e}')
                with open(f'{lj}/spider.txt', 'a+', encoding='utf-8') as f:
                    f.write(f'错误代码：{e}---下载 {img_url} 图片失败\n')
        else:
            print(f'下载{img_name}图片失败！')
            print(f'错误代码：{e}')
            with open(f'{lj}/spider.txt', 'a+', encoding='utf-8') as f:
                f.write(f'错误代码：{e}---下载 {img_url} 图片失败\n')

#保存文本内容
def bcwb(ljj,texts):
    print("开始保存文本")
    with open(ljj, 'w', encoding='utf-8') as f:
        f.write(texts)
    print(f'保存文本内容成功！')