random_sample.py

random_sample.py

#!/usr/bin/env python3

import argparse
import random


def main(path, samplesize):
    """
    Prints random samples in order of selection.

    :param path: file path of newline-separated els.
    :param samplesize: size of sample to select. bounded by (1, len(els)).
    """
    with open(path, "r") as f:
        els = f.read().splitlines()

    # Cap samples to els length, no negatives
    samplesize = max(1, min(len(els), samplesize))
    sample = random.sample(els, k=samplesize)

    for i, s in enumerate(sample):
        print(f"Selection {i+1}: {s}")


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Random sample from a file")
    parser.add_argument(
        "-f",
        "--file",
        type=str,
        dest="path",
        help="Newline separated file of elements",
        default=None,
        required=True,
    )
    parser.add_argument(
        "-s",
        "--size",
        type=int,
        dest="samplesize",
        help="Sample size",
        default=1,
        required=False,
    )
    main(**parser.parse_args().__dict__)