quiz.py 文件源码

def hard(args):
    ref = collections.defaultdict(list)

    with open(args.ref, "rb") as f:
        b = f.read(1)
        i = 0
        while b:
            ref[b].append(i)
            i = i + 1
            b = f.read(1)

    if len(ref) < 256:
        raise Exception

    output = []

    with open(args.input, "rb") as f:
        b = f.read(1)
        while b:
            output.append(ref[b][random.randrange(len(ref[b]))])
            b = f.read(1)

    with open(args.output, "w") as f:
        for item in output:
            f.write(str(item) + os.linesep)