SanYeCao-Nonebot/src/image/get_image.py

import os
import yaml
import random
import requests
from PIL import Image, ImageDraw,ImageFont
from src.configs.path_config import image_local_path,image_local_qq_image_path,rua_png

with open(os.getcwd() +'/src/image/config/image.yaml', 'r', encoding='utf-8') as f:
    image = yaml.load(f.read(), Loader=yaml.FullLoader).get('image')
    # image_local_path = image.get('image_local_path')
    # image_local_qq_image_path = image.get('image_local_qq_image_path')
    smms_token = image.get('smms_token')
    smms_image_upload_history = image.get('smms_image_upload_history')
    ju_he_token = image.get('ju_he_token')
    ju_he_image_list = image.get('ju_he_image_list')
    app_id = image.get('app_id')
    bot_account = image.get('bot_account')

# qq_image_save__path = os.getcwd()+'/'+image_local_qq_image_path

"""本地图片"""
def get_image_names():
    image_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp']  # 定义常见的图片文件扩展名
    image_names = []
    for root, dirs, files in os.walk(image_local_path):
        for file in files:
            if any(file.endswith(ext) for ext in image_extensions):  # 检查文件是否是图片文件
                image_names.append(file)
    random.choice(image_names)  # 随机选取一张图片
    local_image_path = image_local_path + '/' + random.choice(image_names)  # 随机选取一张图片的路径
    return local_image_path

"""获取QQ头像"""
def download_qq_image(member_open_id):
    if not os.path.exists(image_local_qq_image_path):
        os.makedirs(image_local_qq_image_path)

    save_path = image_local_qq_image_path + '/' + member_open_id + '.jpg'
    size = 640 #尺寸 40、100、140、640
    url = f"https://q.qlogo.cn/qqapp/{app_id}/{member_open_id}/{size}"
    response = requests.get(url)  # 发送 GET 请求获取图片资源
    if response.status_code == 200:  # 判断请求是否成功
        with open(save_path, 'wb') as file:  # 以二进制写入模式打开文件
            file.write(response.content)  # 将响应内容写入文件
        return save_path

"""获取QQ头像"""
def download_qq_image_by_account(account):
    if not os.path.exists(image_local_qq_image_path):
        os.makedirs(image_local_qq_image_path)
    if account is None:
        account = bot_account
    save_path = image_local_qq_image_path + '/' + account + '.jpg'
    size = 640  # 尺寸 40、100、140、640
    url = f"https://q2.qlogo.cn/headimg_dl?dst_uin={account}&spec={size}"
    response = requests.get(url)  # 发送 GET 请求获取图片资源
    if response.status_code == 200:  # 判断请求是否成功
        with open(save_path, 'wb') as file:  # 以二进制写入模式打开文件
            file.write(response.content)  # 将响应内容写入文件
        return save_path

"""删除QQ头像"""
def qq_image_delete():
    for root, dirs, files in os.walk(image_local_qq_image_path):
        for file in files:
            file_path = os.path.join(root, file)
            os.remove(file_path)

""" sm.ms  图床"""
def get_smms_image_url():
    # 定义请求的参数
    data = requests.get(smms_image_upload_history, headers={'Authorization': smms_token}, params={"page": "1"}).json().get('data')
    urls = [item['url'] for item in data]
    random_url = random.choice(urls)
    return random_url

"""聚合图床"""
def get_juhe_image_url():
    # 定义请求的参数
    params = {"token": ju_he_token,"f": "json","categories": "猫羽雫","page": 1, "size": 400}
    random_url = random.choice(requests.get(ju_he_image_list, params=params).json().get('docs', [])).get('url')
    return random_url


""" rua 头动图生成"""
class rua():
    def __init__(self, img_file):
        self.author = Image.open(img_file)

    def add_png(self, png_d):
        # 重置图片大小
        author = self.author.resize((png_d[0], png_d[1] - png_d[2]))

        # 载入素材
        rua_p1 = Image.open(png_d[3])

        # 创建背景模板
        rua_png1 = Image.new('RGBA', (110, 110), (255, 255, 255, 255))

        # 使用预定义的参数：jd，合成一帧的样例
        rua_png1.paste(author, (110 - png_d[0], 110 - png_d[1] + png_d[2]), author)
        rua_png1.paste(rua_p1, (0, 110 - png_d[1] - png_d[2]), rua_p1)
        return rua_png1

    def add_gif(self):

        # 获取素材列表
        pst = os.listdir(rua_png)
        for i in range(len(pst)):
            pst[i] = rua_png + pst[i]

        # 预调试好的参数，传入素材列表
        jd = [[90, 90, 5, pst[0]],
              [90, 87, 5, pst[2]],
              [90, 84, 10, pst[3]],
              [90, 81, 8, pst[4]],
              [90, 78, 5, pst[5]],
              [90, 75, 5, pst[6]],
              [90, 72, 8, pst[7]],
              [90, 74, 8, pst[8]],
              [90, 77, 9, pst[9]],
              [90, 80, 8, pst[1]]]

        # 重置要生成的图片大小
        self.author = self.author.resize((90, 90))

        # 绘制模板
        alpha_layer = Image.new('L', (90, 90), 0)
        draw = ImageDraw.Draw(alpha_layer)
        draw.ellipse((0, 0, 90, 90), fill=255)
        self.author.putalpha(alpha_layer)

        # gif列表
        gifs = []
        for i in range(len(jd)):
            # 将参数传递给生成方法
            # 添加到gif列表
            gifs.append(self.add_png(jd[i]))

        # 文件名,是否保存所有,图片列表,fps/ms
        gifs[0].save(image_local_qq_image_path + '/rua.gif', "GIF", save_all=True, append_images=gifs, duration=35, loop=0)
        self.author.close()
        return image_local_qq_image_path + '/rua.gif'

"""
图文合成
"""
def add_text_to_image(image_path, text, output_path, font_size):
    # 打开图片
    image = Image.open(image_path)
    # 创建一个可用于绘制的对象
    draw = ImageDraw.Draw(image)

    # 设置字体和字体大小
    font = ImageFont.truetype("arial.ttf", font_size)  # 这里使用 Arial 字体，你可以替换为其他字体文件路径
    # 设置文本的位置和颜色
    position = (50, 50)
    text_color = (255, 0, 0)  # 红色

    # 在图片上绘制文本
    draw.text(position, text, font=font, fill=text_color)

    # 保存合成后的图片
    image.save(output_path)
    print(f"合成后的图片已保存到 {output_path}")


if __name__ == '__main__':
    print(get_smms_image_url())
    print(get_juhe_image_url())
    print(get_image_names())
    file_path = '8A91A2F3BE5B5AF3FEC97FB5AA6D9B38.jpg'
    au = rua(file_path).add_gif()
    image_path = "Justice.jpg"
    text = "Hello, World!"
    output_path = "output.jpg"
    add_text_to_image(image_path, text, output_path, font_size=48)