Python itertools：从4位码生成含额外数字的6位排列

from itertools import product

# 生成所有两位数字的组合
# 例如 (0, 0), (0, 1), ..., (9, 9)
for x, y in product(range(10), repeat=2):
    # x, y 分别代表两位数字
    # print(f"{x}{y}")
    pass

entry = "1234"
x = 0
y = 0
new_entry = f"{entry}{x}{y}" # 结果为 "123400"

from itertools import permutations

# 对组合后的6位字符串进行全排列
for perm_tuple in permutations(new_entry):
    permutation_str = "".join(perm_tuple)
    # print(permutation_str)

all_permutations = set()
for perm_tuple in permutations(new_entry):
    all_permutations.add("".join(perm_tuple))
# results = list(all_permutations)

from itertools import product, permutations
from typing import Iterable, Set

def get_expanded_permutations(entry: str) -> Set[str]:
    """
    为给定的4位数字字符串生成所有包含两个额外数字的6位排列。

    Args:
        entry: 一个4位数字字符串，例如 "1234"。

    Returns:
        一个包含所有唯一6位排列字符串的集合。
    """
    if not (isinstance(entry, str) and len(entry) == 4 and entry.isdigit()):
        raise ValueError("Input entry must be a 4-digit string.")

    generated_permutations = set()

    # 步骤一：生成所有两位数字的组合 (00-99)
    for x, y in product(range(10), repeat=2):
        # 步骤二：将原始4位码与两位填充数字组合成一个6位字符串
        # 例如 "1234" + "0" + "0" -> "123400"
        combined_string = f"{entry}{x}{y}"

        # 步骤三：对组合后的6位字符串进行全排列
        for perm_tuple in permutations(combined_string):
            # 将元组形式的排列转换为字符串
            permutation_str = "".join(perm_tuple)
            # 步骤四：添加到集合中以自动去重
            generated_permutations.add(permutation_str)

    return generated_permutations

# 示例用法
input_code = "1234"
results = get_expanded_permutations(input_code)
print(f"为 '{input_code}' 生成了 {len(results)} 个唯一的6位排列。")
# 打印前10个结果作为示例
# for i, perm in enumerate(list(results)[:10]):
#     print(perm)

import os
import datetime
from itertools import product, permutations
from typing import Set

def get_expanded_permutations(entry: str) -> Set[str]:
    # ... (与上述函数定义相同) ...
    if not (isinstance(entry, str) and len(entry) == 4 and entry.isdigit()):
        raise ValueError("Input entry must be a 4-digit string.")

    generated_permutations = set()

    for x, y in product(range(10), repeat=2):
        combined_string = f"{entry}{x}{y}"
        for perm_tuple in permutations(combined_string):
            permutation_str = "".join(perm_tuple)
            generated_permutations.add(permutation_str)

    return generated_permutations

def process_files(input_filepath: str, output_filepath: str, log_filepath: str):
    """
    从输入文件读取4位码，生成6位排列，并写入输出文件和日志文件。
    """
    if not os.path.exists(input_filepath):
        raise FileNotFoundError(f"Input file not found: {input_filepath}")

    with open(input_filepath, 'r') as infile:
        input_data = [line.strip() for line in infile if line.strip()]

    total_entries = len(input_data)
    processed_count = 0

    with open(output_filepath, 'w') as outfile, \
         open(log_filepath, 'w') as logfile:

        logfile.write(f"Permutation generation log - {datetime.datetime.now()}\n\n")

        for entry in input_data:
            try:
                # 生成当前4位码的所有6位排列
                perms = get_expanded_permutations(entry)

                # 将所有排列一次性写入输出文件，每个排列占一行
                if perms: # 确保有排列生成
                    outfile.write("\n".join(sorted(list(perms)))) # 写入前排序，可选
                    outfile.write("\n") # 为下一个条目添加分隔符

                logfile.write(f"Generated permutations for entry: '{entry}' ({len(perms)} unique permutations)\n")
                processed_count += 1
                print(f"Processed {processed_count}/{total_entries}: '{entry}'")

            except ValueError as e:
                logfile.write(f"Error processing entry '{entry}': {e}\n")
                print(f"Error processing entry '{entry}': {e}")
            except Exception as e:
                logfile.write(f"An unexpected error occurred for entry '{entry}': {e}\n")
                print(f"An unexpected error occurred for entry '{entry}': {e}")

        logfile.write(f"\nProcessing complete. Total entries processed: {processed_count}\n")

    print("Permutation generation completed.")

if __name__ == "__main__":
    # 模拟输入文件
    with open("input.txt", "w") as f:
        f.write("1234\n")
        f.write("5678\n")
        f.write("abcd\n") # 故意放入一个无效条目

    output_file = "output.txt"
    log_file = f"permutation_log_{datetime.datetime.now().strftime('%Y%m%d%H%M%S')}.log"

    try:
        process_files("input.txt", output_file, log_file)
        print(f"Results written to {output_file}")
        print(f"Log written to {log_file}")
    except Exception as e:
        print(f"An error occurred during file processing: {e}")