add:全随机/random目录

Author: BakaCookie520

app/routes/images.py

@@ -5,7 +5,7 @@
 import logging
 from flask import Blueprint, redirect, send_from_directory, abort, request, render_template
 from ..utils.security import get_safe_path, is_banned
-from ..utils.cache import get_random_image, invalidate_cache
+from ..utils.cache import get_random_image, get_random_image_from_all_folders, invalidate_cache
 from ..config.config import Config
 
 # 配置日志
@@ -14,10 +14,57 @@
 # 创建蓝图
 images_bp = Blueprint('images', __name__)
 
+@images_bp.route('/random')
+def serve_random_from_all():
+    """
+    从所有文件夹中随机选择并返回图片
+    """
+    # 检查封禁状态
+    client_ip = request.remote_addr
+    current_path = request.path
+    banned, remaining, end_time = is_banned(client_ip, current_path, Config.BAN_DURATION)
+
+    if banned:
+        return render_template('too_many_requests.html',
+                               retry_after=int(remaining),
+                               end_time=int(end_time),
+                               client_ip=client_ip,
+                               target_url=current_path), 429
+
+    max_attempts = 3  # 最大重试次数
+    attempt = 0
+
+    while attempt < max_attempts:
+        # 从所有文件夹中随机选择图片
+        folder, image = get_random_image_from_all_folders(Config.IMAGE_BASE, Config.IMAGE_EXTENSIONS)
+        
+        if not folder or not image:
+            abort(404)  # 无有效图像
+
+        # 检查图像文件是否存在
+        folder_path = get_safe_path(Config.IMAGE_BASE, folder)
+        image_path = get_safe_path(folder_path, image)
+        
+        if image_path and os.path.isfile(image_path):
+            # 重定向到实际图像URL
+            return redirect(f'/{folder}/{image}')
+
+        # 文件不存在：记录日志并尝试重建缓存
+        logger.warning(f"图像文件不存在: {image_path}, 尝试 {attempt + 1}/{max_attempts}")
+        attempt += 1
+
+        # 使缓存失效
+        invalidate_cache(folder)
+
+    # 多次尝试后仍失败
+    logger.error("无法从所有文件夹中找到有效图像")
+    abort(404)
+
+
 @images_bp.route('/<path:folder>')
 def serve_sequential_image(folder):
     """
-    顺序服务图像：按预定义的随机顺序循环显示文件夹中的图像
+    随机服务图像：从指定文件夹中随机返回图像
     """
     # 首先检查封禁状态
     client_ip = request.remote_addr
@@ -46,7 +93,7 @@ def serve_sequential_image(folder):
     attempt = 0
 
     while attempt < max_attempts:
-        # 获取随机图像
+        # 获取随机图像（真随机）
         image = get_random_image(Config.IMAGE_BASE, folder, Config.IMAGE_EXTENSIONS)
         if not image:
             abort(404)  # 无有效图像

app/utils/cache.py

@@ -49,7 +49,7 @@ def init_folder_cache(image_base, folder, image_extensions):
 
 def get_random_image(image_base, folder, image_extensions):
     """
-    获取文件夹中的随机图像
+    获取文件夹中的随机图像（真随机）
     
     Args:
         image_base: 图像基础目录
@@ -60,37 +60,70 @@ def get_random_image(image_base, folder, image_extensions):
         随机图像文件名或None
     """
     with cache_lock:  # 线程安全操作
-        # 如果缓存中没有该文件夹
+        # 如果缓存中没有该文件夹，初始化缓存
         if folder not in folder_cache:
-            images = init_folder_cache(image_base, folder, image_extensions)  # 初始化缓存
+            images = init_folder_cache(image_base, folder, image_extensions)
             if not images:
                 return None  # 无有效图像
-
-            # 创建基于文件夹和时间的随机种子
-            seed = hash(f"{folder}-{time.time()}") % (2 ** 32)
-            random.seed(seed)
-            shuffled = images.copy()
-            random.shuffle(shuffled)  # 随机打乱图像列表
-
-            # 将打乱后的列表存入缓存
+            # 存储图像列表到缓存
             folder_cache[folder] = {
-                'images': shuffled,
-                'index': 0,  # 当前索引位置
-                'seed': seed  # 使用的随机种子
+                'images': images
             }
 
         cache = folder_cache[folder]
         if not cache['images']:
             del folder_cache[folder]  # 空列表则删除缓存项
             return None
 
-        # 获取当前图像并更新索引
-        current_index = cache['index']
-        image = cache['images'][current_index]
-        # 循环索引（到达末尾后回到开头）
-        cache['index'] = (current_index + 1) % len(cache['images'])
+        # 真随机：每次都随机选择一个图像
+        return random.choice(cache['images'])
+
+
+def get_random_image_from_all_folders(image_base, image_extensions):
+    """
+    从所有文件夹中随机选择一张图片
+    
+    Args:
+        image_base: 图像基础目录
+        image_extensions: 支持的图像扩展名列表
+        
+    Returns:
+        (文件夹名称, 图像文件名) 或 (None, None)
+    """
+    with cache_lock:
+        # 获取所有子文件夹
+        try:
+            subfolders = [d for d in os.listdir(image_base)
+                         if os.path.isdir(get_safe_path(image_base, d))]
+        except Exception as e:
+            logger.error(f"获取子文件夹列表失败: {str(e)}")
+            return None, None
+        
+        if not subfolders:
+            logger.warning("没有找到任何子文件夹")
+            return None, None
+        
+        # 收集所有文件夹中的所有图片
+        all_images = []  # 格式: [(folder, image), ...]
+        
+        for folder in subfolders:
+            # 如果缓存中没有该文件夹，初始化缓存
+            if folder not in folder_cache:
+                images = init_folder_cache(image_base, folder, image_extensions)
+                if images:
+                    folder_cache[folder] = {'images': images}
+            
+            # 从缓存中获取图片列表
+            if folder in folder_cache and folder_cache[folder]['images']:
+                for image in folder_cache[folder]['images']:
+                    all_images.append((folder, image))
+        
+        if not all_images:
+            logger.warning("没有找到任何图片")
+            return None, None
 
-        return image
+        # 真随机：随机选择一张图片
+        return random.choice(all_images)
 
 
 def invalidate_cache(folder):