python --version  # Windows 系統(tǒng)
# 或
python3 --version  # macOS/Linux 系統(tǒng)

import random
import string
import os

# 定義默認配置（可直接修改，適配個人使用習慣）
DEFAULT_PASSWORD_LENGTH = 16  # 默認密碼長度
MIN_LENGTH = 8  # 最小密碼長度限制
MAX_LENGTH = 64  # 最大密碼長度限制
DEFAULT_BATCH_COUNT = 5  # 默認批量生成數量

def get_char_set(include_upper=True, include_lower=True, include_digits=True, include_symbols=True):
    """
    構建密碼字符集：根據用戶選擇組合對應的字符類型
    :param include_upper: 是否包含大寫字母（bool）
    :param include_lower: 是否包含小寫字母（bool）
    :param include_digits: 是否包含數字（bool）
    :param include_symbols: 是否包含特殊符號（bool）
    :return: 組合后的字符集字符串
    """
    char_set = ""
    
    # 拼接對應字符集
    if include_upper:
        char_set += string.ascii_uppercase  # 大寫字母 A-Z
    if include_lower:
        char_set += string.ascii_lowercase  # 小寫字母 a-z
    if include_digits:
        char_set += string.digits  # 數字 0-9
    if include_symbols:
        char_set += string.punctuation  # 特殊符號 !"#$%&'()*+,-./:;<=>?@[\]^_`{|}~
    
    # 驗證字符集是否為空
    if not char_set:
        raise ValueError("字符集不能為空！至少需要選擇一種字符類型用于生成密碼")
    
    return char_set

def generate_single_password(length=DEFAULT_PASSWORD_LENGTH, char_set=None):
    """
    生成單個安全密碼
    :param length: 密碼長度（int）
    :param char_set: 密碼字符集（str）
    :return: 生成的密碼字符串
    """
    # 驗證密碼長度是否合法
    if not (MIN_LENGTH <= length <= MAX_LENGTH):
        raise ValueError(f"密碼長度不合法！請輸入 {MIN_LENGTH}-{MAX_LENGTH} 之間的整數")
    
    # 若未傳入字符集，使用默認完整字符集
    if char_set is None:
        char_set = get_char_set()
    
    # 生成隨機密碼：使用 random.choices 從字符集中隨機選取指定長度的字符
    # 加入 random.shuffle 打亂字符順序，提升密碼隨機性
    password_chars = random.choices(char_set, k=length)
    random.shuffle(password_chars)  # 打亂字符排列，避免固定順序漏洞
    password = ''.join(password_chars)
    
    return password

def batch_generate_passwords(count=DEFAULT_BATCH_COUNT, length=DEFAULT_PASSWORD_LENGTH, char_set=None):
    """
    批量生成密碼
    :param count: 生成數量（int）
    :param length: 單個密碼長度（int）
    :param char_set: 密碼字符集（str）
    :return: 密碼列表
    """
    # 驗證批量生成數量是否合法
    if count <= 0 or not isinstance(count, int):
        raise ValueError("生成數量不合法！請輸入大于 0 的整數")
    
    password_list = []
    for _ in range(count):
        password = generate_single_password(length, char_set)
        password_list.append(password)
    
    return password_list

def save_passwords_to_file(password_list, filename="generated_passwords.txt"):
    """
    將生成的密碼列表保存到本地文本文件
    :param password_list: 密碼列表（list）
    :param filename: 保存文件名（str）
    """
    if not password_list:
        print("? 無有效密碼可保存！")
        return
    
    # 以追加模式寫入文件，避免覆蓋原有數據，同時添加時間戳便于區(qū)分
    with open(filename, 'a', encoding='utf-8') as f:
        # 寫入文件分隔符與時間戳
        current_time = str(random.random())  # 簡化時間戳，新手易理解（可替換為 datetime 精確時間）
        f.write(f"\n========== 密碼生成記錄 - {current_time} ==========\n")
        for index, password in enumerate(password_list, start=1):
            f.write(f"{index}. {password}\n")
    
    print(f"? 密碼已成功保存至 {filename}，共 {len(password_list)} 條記錄")

def main_interactive():
    """
    交互式主菜單，提供用戶友好的操作入口
    """
    print("?? 歡迎使用 Python 安全密碼生成器")
    print(f"?? 默認配置：密碼長度 {DEFAULT_PASSWORD_LENGTH} 位，包含大小寫字母、數字、特殊符號")
    print(f"?? 密碼長度限制：{MIN_LENGTH}-{MAX_LENGTH} 位，批量生成默認 {DEFAULT_BATCH_COUNT} 個\n")
    
    while True:
        # 打印主菜單
        print("========== 密碼生成器主菜單 ==========")
        print("1. 生成單個密碼")
        print("2. 批量生成密碼")
        print("3. 自定義字符集生成密碼")
        print("4. 退出程序")
        print("======================================")
        
        # 獲取用戶選擇
        try:
            choice = int(input("請輸入您的操作選擇（1-4）："))
        except ValueError:
            print("? 輸入無效！請輸入 1-4 之間的整數\n")
            continue
        
        # 處理用戶選擇
        if choice == 1:
            # 生成單個密碼
            try:
                length = int(input(f"請輸入密碼長度（默認 {DEFAULT_PASSWORD_LENGTH}，范圍 {MIN_LENGTH}-{MAX_LENGTH}）：").strip() or DEFAULT_PASSWORD_LENGTH)
                password = generate_single_password(length)
                print(f"\n? 單個密碼生成成功：{password}\n")
                
                # 詢問是否保存
                save_choice = input("是否將該密碼保存到文件？（y/n，默認 n）：").strip().lower()
                if save_choice == 'y':
                    save_passwords_to_file([password])
            except Exception as e:
                print(f"? 生成失?。簕str(e)}\n")
        
        elif choice == 2:
            # 批量生成密碼
            try:
                count = int(input(f"請輸入生成數量（默認 {DEFAULT_BATCH_COUNT}，大于 0）：").strip() or DEFAULT_BATCH_COUNT)
                length = int(input(f"請輸入密碼長度（默認 {DEFAULT_PASSWORD_LENGTH}，范圍 {MIN_LENGTH}-{MAX_LENGTH}）：").strip() or DEFAULT_PASSWORD_LENGTH)
                password_list = batch_generate_passwords(count, length)
                
                print(f"\n? 批量密碼生成成功，共 {count} 個：")
                for index, pwd in enumerate(password_list, start=1):
                    print(f"{index}. {pwd}")
                print()
                
                # 詢問是否保存
                save_choice = input("是否將這些密碼保存到文件？（y/n，默認 n）：").strip().lower()
                if save_choice == 'y':
                    save_passwords_to_file(password_list)
            except Exception as e:
                print(f"? 生成失?。簕str(e)}\n")
        
        elif choice == 3:
            # 自定義字符集生成密碼
            print("\n請選擇需要包含的字符類型（輸入 y 表示包含，n 表示不包含，直接回車默認 y）")
            try:
                upper_choice = input("是否包含大寫字母（A-Z）？：").strip().lower() or 'y'
                lower_choice = input("是否包含小寫字母（a-z）？：").strip().lower() or 'y'
                digits_choice = input("是否包含數字（0-9）？：").strip().lower() or 'y'
                symbols_choice = input("是否包含特殊符號（!@#$ 等）？：").strip().lower() or 'y'
                
                # 轉換為布爾值
                include_upper = (upper_choice == 'y')
                include_lower = (lower_choice == 'y')
                include_digits = (digits_choice == 'y')
                include_symbols = (symbols_choice == 'y')
                
                # 構建自定義字符集
                char_set = get_char_set(include_upper, include_lower, include_digits, include_symbols)
                
                # 選擇生成單個或批量
                gen_type = input("\n請選擇生成類型（1-單個，2-批量，默認 1）：").strip() or '1'
                if gen_type == '1':
                    length = int(input(f"請輸入密碼長度（默認 {DEFAULT_PASSWORD_LENGTH}，范圍 {MIN_LENGTH}-{MAX_LENGTH}）：").strip() or DEFAULT_PASSWORD_LENGTH)
                    password = generate_single_password(length, char_set)
                    print(f"\n? 自定義字符集密碼生成成功：{password}\n")
                    
                    save_choice = input("是否將該密碼保存到文件？（y/n，默認 n）：").strip().lower()
                    if save_choice == 'y':
                        save_passwords_to_file([password])
                elif gen_type == '2':
                    count = int(input(f"請輸入生成數量（默認 {DEFAULT_BATCH_COUNT}，大于 0）：").strip() or DEFAULT_BATCH_COUNT)
                    length = int(input(f"請輸入密碼長度（默認 {DEFAULT_PASSWORD_LENGTH}，范圍 {MIN_LENGTH}-{MAX_LENGTH}）：").strip() or DEFAULT_PASSWORD_LENGTH)
                    password_list = batch_generate_passwords(count, length, char_set)
                    
                    print(f"\n? 自定義字符集批量密碼生成成功，共 {count} 個：")
                    for index, pwd in enumerate(password_list, start=1):
                        print(f"{index}. {pwd}")
                    print()
                    
                    save_choice = input("是否將這些密碼保存到文件？（y/n，默認 n）：").strip().lower()
                    if save_choice == 'y':
                        save_passwords_to_file(password_list)
                else:
                    print("? 生成類型選擇無效！\n")
            except Exception as e:
                print(f"? 生成失敗：{str(e)}\n")
        
        elif choice == 4:
            # 退出程序
            print("?? 程序已退出，感謝使用！生成的密碼已保存至 generated_passwords.txt（若選擇保存）")
            break
        
        else:
            print("? 選擇無效！請輸入 1-4 之間的整數\n")

# 程序入口
if __name__ == "__main__":
    main_interactive()

python password_generator.py  # Windows 系統(tǒng)
# 或
python3 password_generator.py  # macOS/Linux 系統(tǒng)