编程新手求助：完善文件上传函数，实现文件夹文件数限制处理

Hey there! Great job getting the basic upload function working—let's build out the folder management logic you need to handle the 10000-file limit and auto-create numbered folders. I'll walk you through this step by step with a concrete example (I'll use Python since it's super accessible for beginners, but the logic translates to other languages too).

1. Core Logic Overview #

Here's the plan we'll implement:

• Start with your base target folder (e.g., ./uploads/folder)
• Check how many files are in the current target folder (ignore subfolders, we only care about uploaded files)
• If the count hits 10000, create a new numbered folder (like folder 1, then folder 2, etc.)
• Save the uploaded file to the first folder that has space (or the new one we just created)

2. Full Code Example #

import os
# Optional: For handling duplicate filenames later
import time

def save_uploaded_file(uploaded_file, base_folder="./uploads/folder"):
    # Make sure the base folder exists (create it if not)
    os.makedirs(base_folder, exist_ok=True)
    
    target_folder = base_folder
    folder_suffix = 0
    
    while True:
        # Count only files (skip subfolders like our numbered ones)
        # Using os.scandir is faster than os.listdir for large folders
        file_count = sum(1 for entry in os.scandir(target_folder) if entry.is_file())
        
        if file_count < 10000:
            break
        
        # Folder is full—move to the next numbered one
        folder_suffix += 1
        target_folder = f"{base_folder} {folder_suffix}"
        # Create the new folder if it doesn't exist
        os.makedirs(target_folder, exist_ok=True)
    
    # Save the file to the target folder
    # Optional: Add a timestamp to avoid overwriting duplicate filenames
    filename, ext = os.path.splitext(uploaded_file.filename)
    safe_filename = f"{filename}_{int(time.time())}{ext}"
    file_path = os.path.join(target_folder, safe_filename)
    
    # Write the file content—adjust this based on your upload setup
    # If you're using Flask, uploaded_file is a FileStorage object; for raw bytes, use uploaded_file directly
    with open(file_path, "wb") as f:
        f.write(uploaded_file.read())
    
    return file_path

3. Key Details & Optimizations #

• Avoid counting subfolders: We use entry.is_file() (from os.scandir) to make sure we only count actual uploaded files, not the numbered subfolders we create. This keeps our count accurate.
• Faster folder scanning: os.scandir is way faster than os.listdir for large folders because it gets file metadata (like whether it's a file or folder) in one go, instead of making separate system calls.
• Cross-platform safety: Using os.path.join ensures your paths work on Windows, macOS, and Linux without having to manually handle / vs \.

4. Bonus: Handling Duplicate Filenames #

I added a timestamp to the filename in the example because if two users upload a file with the same name (like vacation.jpg), the second one would overwrite the first. Adding a Unix timestamp ensures every filename is unique. If you prefer, you could also use a random string instead of a timestamp.

If you're working with a different language (like JavaScript/Node.js, Java, C#), just let me know—I can adapt this logic to fit your stack!

内容的提问来源于stack exchange，提问作者Achilles Balaba Revalliv