rag-gateway/app/utils/password_handle.py

import asyncio
import random
import string


async def generate_password(length=10):
    if length < 6:  # 至少需要3位密码以包含所有必要的字符
        raise ValueError("Password length should be at least 3")

    # 确保密码中至少包含一个字母和一个数字
    password = [
        random.choice(string.ascii_uppercase),  # 大写字母
        random.choice(string.ascii_lowercase),  # 小写字母
        random.choice(string.digits)  # 数字
    ]

    # 添加剩余的随机字符
    characters = string.ascii_letters + string.digits
    password.extend(random.choice(characters) for _ in range(length - 3))

    # 打乱密码以确保随机性
    random.shuffle(password)

    # 将列表转换为字符串
    return ''.join(password)

if __name__ == "__main__":
    # 生成一个10位的密码
    asyncio.run(generate_password(10))
    # password = generate_password(10)
    # print(password)