Python批量处理微信公众号图片资源

背景

最近发现一些公众号的图片资源还是可以利用一下的。本着能让机器干的活绝不能自己动手的原则，于是决定使用Python批量处理，让机器来处理一下。

实现思路

首先需要找到微信公众号的图片外链的共同特点，例如域名格式等。然后通过匹配这些特点的外链，识别出需要的图片资源。接着，遍历匹配到的外链，进行批量下载。

实现代码

import os
import re
import requests
from urllib.parse import urlparse, parse_qs

# 定义读取文件、匹配网址并保存文件的函数
def extract_and_save_urls(file_name, output_folder='下载图片素材'):
    # 创建输出文件夹（如果不存在）
    if not os.path.exists(output_folder):
        os.makedirs(output_folder)

    # 读取文件内容
    with open(file_name, 'r', encoding='utf-8') as file:
        content = file.read()

    # 匹配网址的正则表达式
    url_pattern = re.compile(r'https://mmbiz\.qpic\.cn/mmbiz_png/[^\s]+')
    urls = url_pattern.findall(content)

    # 下载并保存文件
    for idx, url in enumerate(urls):
        try:
            response = requests.get(url)
            if response.status_code == 200:
                # 解析 URL 以获取文件扩展名
                parsed_url = urlparse(url)
                file_extension = parse_qs(parsed_url.query).get('wx_fmt', [''])[0]
                if not file_extension:  # 默认扩展名为 'png'，如果未指定 wx_fmt 参数
                    file_extension = 'png'
                file_name = f'image_{idx+1}.{file_extension}'
                file_name = file_name.replace('"', '').replace(',', '').replace('?', '') # 移除不必要的字符
                file_path = os.path.join(output_folder, file_name)
                with open(file_path, 'wb') as img_file:
                    img_file.write(response.content)
                print(f'Successfully saved {url} as {file_name}')
            else:
                print(f'Failed to download {url}, status code: {response.status_code}')
        except Exception as e:
            print(f'Error downloading {url}: {e}')

# 指定输入输出文件
file_name = 'data.js'  
extract_and_save_urls(file_name)