mirror of
https://github.com/ClovertaTheTrilobita/SanYeCao-Nonebot.git
synced 2026-04-01 22:04:51 +00:00
e50a197c18
- 将 `get_image.py` 中的功能拆分为多个独立模块：`qq_image.py`, `rua.py`, `add_text_to_image.py`, `delete_file.py`, `download_image.py` - 更新相关插件的导入路径以适配新模块结构 - 新增 `nai_loong.py` 和 `platform.py` 插件
33 lines
1.4 KiB
Python
33 lines
1.4 KiB
Python
import os
|
|
import random
|
|
import requests
|
|
|
|
from src.configs.path_config import image_local_path
|
|
from src.configs.api_config import smms_token,smms_image_upload_history,ju_he_token,ju_he_image_list
|
|
|
|
"""本地图片"""
|
|
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
|
|
|
|
""" 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
|