随机组合名字的小程序

#!/usr/bin/env python3
# coding=utf-8

import os
import copy

def make_name(line, surname):
    L = [line.strip(),"\n"]
    count = 0
    line = line.replace("，","")
    line = line.replace("。","").strip()
    for first in line:
        new_line = list(copy.copy(line))
        new_line.remove(first)
        for second in new_line:
            L.append(surname+first+second+"  ")
            count += 1
            if count%10==0:
                L.append("\n")
    return L

def write_txt(path, surname="程"):
    try:
        os.remove("待选.txt")
    except:
        pass
    with open("待选.txt","a") as f:
        for line in open(path, "r"):
            f.writelines(make_name(line, surname))
            f.write("\n\n")


if __name__ == "__main__":
    write_txt("poem.txt")