1
# -*- coding: utf-8 -*-
2
import os
3
import re
4
import requests
5
from urllib.parse import urlparse
6
from concurrent.futures import ThreadPoolExecutor
7

8

9
def download_image(url, output_dir, ym, idx):
10
    """
11
    下载单个图片并保存到指定文件夹中，按照年/月的结构组织。
12
    """
13
    print(f"正在下载第 {idx} 张图片：{url}")
14
    try:
15
        response = requests.get(url, stream=True, timeout=10)
16
        if response.status_code == 200:
17
            # 获取文件扩展名
18
            parsed_url = urlparse(url)
19
            ext = os.path.splitext(parsed_url.path)[1]
20
            if not ext:
21
                ext = ".jpg"  # 默认使用 jpg 扩展名
22

23
            # 解析年月信息创建目录结构
24
            year, month = ym.split('-')
25
            year_month_dir = os.path.join(output_dir, year, month)
26

27
            # 创建年/月层级目录
28
            os.makedirs(year_month_dir, exist_ok=True)
29

30
            # 图片保存的路径和文件名 (格式: YYYY-MM_DD.ext)
31
            filename = f"{ym}_{idx:02d}{ext}"
32
            filepath = os.path.join(year_month_dir, filename)
33

34
            # 保存图片到文件
35
            with open(filepath, "wb") as f:
36
                for chunk in response.iter_content(chunk_size=1024):
37
                    if chunk:
38
                        f.write(chunk)
39
            print(f"保存到：{filepath}")
40
        else:
41
            print(f"下载失败，状态码：{response.status_code}")
42
    except Exception as e:
43
        print(f"下载图片时出现错误：{e}")
44

45

46
def get_url(md_file, output_dir, ym):
47
    # 读取 md 文件内容
48
    with open(md_file, "r", encoding="utf-8") as f:
49
        content = f.read()  # 使用正则表达式提取图片链接
50
    pattern = r'\[download 4k\]\((.*?)\)'
51
    urls = re.findall(pattern, content)
52
    print(f"在 {md_file} 中找到 {len(urls)} 个图片链接。")
53

54
    # 创建图片存储文件夹（如果不存在）
55
    os.makedirs(output_dir, exist_ok=True)
56

57
    # 使用线程池下载每个图片
58
    with ThreadPoolExecutor() as executor:
59
        for idx, url in enumerate(urls, start=1):
60
            executor.submit(download_image, url, output_dir, ym, idx)
61

62

63
def main(root_dir, output_dir):
64
    """
65
    遍历根目录下的所有子文件夹，查找 README.md 文件并下载图片。
66
    """
67
    # 遍历根目录下的所有子文件夹
68
    for root, dirs, files in os.walk(root_dir):
69
        for file in files:
70
            if file == "README.md":
71
                # 获取年月信息（文件夹名即为年月）
72
                folder_name = os.path.basename(root)
73
                # 确保文件夹名称是有效的年月格式
74
                if re.match(r'^\d{4}-\d{2}$', folder_name):
75
                    md_file_path = os.path.join(root, file)
76
                    print(f"正在处理 {md_file_path} ...")
77
                    # 下载该目录下的所有图片
78
                    get_url(md_file_path, output_dir, folder_name)
79

80

81
if __name__ == "__main__":
82
    # 根目录路径（wallpaper 文件夹路径）
83
    wallpaper_dir = "./bing-wallpaper/picture"
84
    # 指定图片保存的输出文件夹
85
    output_directory = "./wallpaper"
86
    # 遍历并下载所有图片
87
    main(wallpaper_dir, output_directory)