如何改造随机字符生成代码，支持4种指定格式并随机选用？

核心实现思路 #

• 封装两个基础生成函数：一个生成指定长度的大写字母+数字混合字符串，另一个生成指定长度的纯数字字符串
• 针对每种格式，先生成对应长度的原始字符，再通过分段拼接的方式插入分隔符（-或空格）
• 用随机选择逻辑从4种格式中挑选一种生成结果

完整代码 #

import random
import string

def generate_alnum(length):
    """生成指定长度的大写字母+数字混合字符串"""
    chars = string.ascii_uppercase + string.digits
    return ''.join(random.choice(chars) for _ in range(length))

def generate_digits(length):
    """生成指定长度的纯数字字符串"""
    return ''.join(random.choice(string.digits) for _ in range(length))

def generate_random_format():
    """随机生成4种格式中的一种"""
    format_choice = random.randint(1, 4)
    
    if format_choice == 1:
        # XX-XXXX-XXXX-XXXX-XXXX 格式
        parts = [
            generate_alnum(2),
            generate_alnum(4),
            generate_alnum(4),
            generate_alnum(4),
            generate_alnum(4)
        ]
        return '-'.join(parts)
    elif format_choice == 2:
        # XXX XXX XXXX 格式（纯数字）
        parts = [
            generate_digits(3),
            generate_digits(3),
            generate_digits(4)
        ]
        return ' '.join(parts)
    elif format_choice == 3:
        # XXXX XXXX XXXX 格式（大写字母+数字）
        parts = [
            generate_alnum(4),
            generate_alnum(4),
            generate_alnum(4)
        ]
        return ' '.join(parts)
    else:
        # 16位无分隔符格式（大写字母+数字）
        return generate_alnum(16)

# 示例：生成10个随机格式字符串并写入文件
if __name__ == "__main__":
    with open("random_strings.txt", "w") as f:
        for _ in range(10):
            f.write(generate_random_format() + "\n")

关键格式处理说明 #

1. XX-XXXX-XXXX-XXXX-XXXX格式：
   分别生成2位、4位、4位、4位、4位的混合字符串片段，再用'-'.join(parts)自动将片段用短横线连接，无需手动拼接分隔符。
2. XXX XXX XXXX格式：
   生成3位、3位、4位的纯数字片段，通过' '.join(parts)插入空格分隔。
3. XXXX XXXX XXXX格式：
   生成三个4位的混合字符串片段，用空格连接即可。
4. 16位无分隔符格式：直接调用基础生成函数生成对应长度的字符串。

内容的提问来源于stack exchange，提问作者buji