def main()

in floresv1/scripts/shuf.py [0:0]

• 10 lines of code
• 4 McCabe index (conditional complexity)

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--num-lines', '-n', default=None, help='Output the first n lines after shuffling', type=int)
    parser.add_argument('--seed', '-s', default=42, help='Random seed', type=int)
    args = parser.parse_args()

    lines = [line for line in sys.stdin]
    args.num_lines = min(args.num_lines or len(lines), len(lines))

    np.random.seed(args.seed)
    for i in np.random.choice(len(lines), args.num_lines, replace=False):
        print(lines[i], end='')