liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

秀人网图片随机优化

Browse Source
This commit is contained in:
liuwei
2025-03-06 17:57:58 +08:00
parent a698edb7ff
commit 48f7426937
1 changed files with 8 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
xiuren/random_pic.py
View File

@@ -3,24 +3,17 @@ import random
def get_random_file_from_dir(directory): def get_random_file_from_dir(directory):
selected_file = None
file_count = 0
# 定义图片文件的扩展名列表
image_extensions = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp'} image_extensions = {'.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp'}
image_files = []
# 遍历目录及其所有子目录 with os.scandir(directory) as entries:
for root, dirs, files in os.walk(directory): for entry in entries:
for file in files: if entry.is_file():
# 获取文件扩展名 _, ext = os.path.splitext(entry.name)
_, ext = os.path.splitext(file) if ext.lower() in image_extensions:
# 只考虑图片文件 image_files.append(entry.path)
if ext.lower() in image_extensions:
file_count += 1
if random.randint(1, file_count) == 1:
# 每个图片文件有1/file_count的概率被选中
selected_file = os.path.join(root, file)
return selected_file return random.choice(image_files) if image_files else None
def get_xiuren_pic(): def get_xiuren_pic():