import re
 
# 给定的 Markdown 文本
markdown_text = """
1. [如果历史是一群喵](https://mp.weixin.qq.com/s/WVwlW3bhcirBSn2KjsMEYA)
2. [如果历史是一群喵 | 祖先喵们的统一战争](https://mp.weixin.qq.com/s/HwkfqalZD7Ugvpt5lcPyuQ)
3. [吃饭穿衣，竟然都是皇上教的！](https://mp.weixin.qq.com/s/MRvf9c2_8RPX8ouSaNiP1Q)
4. [他年少丧父，被迫子承父业，最后靠湿身建立了一个国家](https://mp.weixin.qq.com/s/l-nWXU45sku88ML4NCjn7Q)
5. [天下都是我家哒！](https://mp.weixin.qq.com/s/PVl-eHCmZSMHLQhm_pRAvw)
6. [跟你讲一下三千多年前的狗血剧](https://mp.weixin.qq.com/s/VoNhyDXoy8Yg-cBCzDcdzg)
...
"""
 
# 使用正则表达式提取所有 URL
urls = re.findall(r'\[.*?\]\((https?://[^\s]+)\)', markdown_text)

from tqdm import tqdm
import requests  
from bs4 import BeautifulSoup
import time
import random
import os
 
headers = {
    'cookie':'pgv_pvid=6670082751; RK=WMxp6iw+3H; ptcz=831e2d5114bbf9b46ee7956fedb62717ee910417ecd992f3e0027f034213caf1; o_cookie=2925851543; pac_uid=1_2925851543; iip=0; tvfe_boss_uuid=94828b35f56c4131; LW_uid=01d6E8a1d0T8Y6S87134I123O2; eas_sid=J116c8t1G078b6f8N1u4m24059; LW_sid=6166y891k1d2s4h7v9M5A8K6e8; rewardsn=; wxtokenkey=777; wwapp.vid=; wwapp.cst=; wwapp.deviceid=',
    'user-agent':'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36 Edg/112.0.1722.48'
}
 
index = 0
 
for url in urls:
    index += 1
    print("第 " + str(index) + ' 章')
    try:
        response = requests.get(url, headers=headers)
        # print(response.status_code)  # 打印响应状态码
        # 获取网页数据
        html = response.text
 
        soup = BeautifulSoup(html, 'html.parser')
        img_list = soup.find_all('img')
    except Exception as e:
        print(f"{index}：错误 - {str(e)}")
 
    for img_url in tqdm(img_list):
        try:
            name = str(img_list.index(img_url))
            # print(img_url)
            img_link = img_url.get('data-src')
            if img_link != None:
                # print(img_link)
                response2 = requests.get(img_link)
                # 图片是二进制数据，获取要用content，文本文件用text
                img_content = response2.content
                # 设置休眠时间，防止速度过快被封
                time.sleep(random.uniform(3, 5))
                # 保存文件
                folder_path = './如果历史是一群喵/' + '第 '+ str(index) + ' 章'
                if not os.path.exists(folder_path):
                    # 如果文件夹不存在，则创建它
                    os.makedirs(folder_path)
                with open(folder_path + '/' + name + '.' + img_link.split('=')[-1],'wb+') as f:
                     f.write(img_content)
                     f.close()
                     # print(f'第 {name} 张图片下载成功')
        except Exception as e:
            print('第 '+ str(index) + ' 章：错误：' + str(e))

import os
 
def rename_files_in_folder(folder_path):
    # 遍历文件夹中的所有文件
    for filename in os.listdir(folder_path):
        # 获取完整文件路径
        old_file = os.path.join(folder_path, filename)
        
        if os.path.isfile(old_file):
            # 如果文件后缀是 .appmsg，改为 .jpeg
            if filename.endswith('.appmsg'):
                new_file = os.path.join(folder_path, filename.replace('.appmsg', '.jpeg'))
                os.rename(old_file, new_file)
            
            # 如果文件后缀是 .1，改为 .gif
            elif filename.endswith('.1'):
                new_file = os.path.join(folder_path, filename.replace('.1', '.gif'))
                os.rename(old_file, new_file)
 
# 使用示例
folder_path = r'./如果历史是一群喵/第 184 章'  # 替换为目标文件夹的路径
rename_files_in_folder(folder_path)

import os
import re
from tqdm import tqdm
from PIL import Image
 
load_path = "./如果历史是一群喵"
save_path = "./如果历史是一群喵 resized"
 
# A4 纵向尺寸在 72 DPI 下的像素大小
A4_width = 595
A4_height = 840
A4_aspect_ratio = A4_width / A4_height
 
item_list = os.listdir(load_path)
item_list.sort(key=lambda x: int(re.search(r'第 (\d+) 章', x).group(1)))
 
# 遍历顶层子文件夹
for item in item_list:
    item_path = os.path.join(load_path, item)
    print(item)
    if os.path.isdir(item_path):
        # 自定义排序函数，提取文件名中的数字部分并按数字排序
        file_list = os.listdir(item_path)
        file_list.sort(key=lambda x: int(re.match(r"(\d+)", x).group(0)))
        # 找到第一个后缀不是 '.jpeg' 的文件，移除该元素及之后的所有元素
        for i, file in enumerate(file_list):
            if not file.endswith('.jpeg'):
                file_list = file_list[:i]  # 切片操作，保留列表中 i 之前的元素
                break
        # 创建一个 pdf 对象
        pdf = FPDF()
        for image_file in tqdm(file_list):
            image = Image.open(os.path.join(item_path, image_file))
            # 获取原始图片的宽度和高度
            original_width, original_height = image.size
 
            # 计算新尺寸比例，保持宽高比不变
            aspect_ratio = original_width / original_height
            
            # 根据宽高比决定如何调整图片
            if aspect_ratio > A4_aspect_ratio:
                # 图片宽度超出比例限制，调整宽度为 A4 宽度，计算高度
                new_width = A4_width
                new_height = int(new_width / aspect_ratio)
            else:
                # 图片高度超出比例限制，调整高度为 A4 高度，计算宽度
                new_height = A4_height
                new_width = int(new_height * aspect_ratio)
 
            # 调整图片大小
            resized_image = image.resize((new_width, new_height))
 
            # 创建一个白色的 A4 背景图像
            background = Image.new('RGB', (A4_width, A4_height), (255, 255, 255))
            # 计算将图片居中放置在 A4 背景中的位置
            left = (A4_width - new_width) // 2
            top = (A4_height - new_height) // 2
 
            # 将调整后的图片粘贴到背景图像上
            background.paste(resized_image, (left, top))
            # 获取图片的宽度和高度
            width, height = background.size
            
            folder_path = os.path.join(os.path.join(save_path, item))
            if not os.path.exists(folder_path):
                # 如果文件夹不存在，则创建它
                os.makedirs(folder_path)
            background.save(os.path.join(folder_path, image_file), format="JPEG", quality=80)

from fpdf import FPDF
import re
import os
from tqdm import tqdm
 
load_path = './如果历史是一群喵 resized/'
save_path = './如果历史是一群喵 pdf/'
 
# 将图像文件转换为 PDF
def images_to_pdf(image_paths, output_pdf_path):
    pdf = FPDF()
    image_paths.sort(key=lambda x: int(re.search(r'(\d+)\.jpeg', x).group(1)))
    # 遍历所有图像文件
    for image_path in image_paths:
        # 添加新的一页
        pdf.add_page()
        # 插入图片，设置为 A4 尺寸的图像（宽度 210mm，高度 297mm）
        pdf.image(image_path, x=0, y=0, w=210, h=297)
    
    # 输出 PDF 文件
    pdf.output(output_pdf_path)
 
for folder in tqdm(os.listdir(load_path)):
    image_folder = os.path.join(load_path, folder)
    image_paths = [os.path.join(image_folder, f) for f in os.listdir(image_folder)]
 
    output_pdf_path = os.path.join(save_path, folder + '.pdf')  # 输出 PDF 的路径
    images_to_pdf(image_paths, output_pdf_path)