dotfiles/bin/genpw

#!/usr/bin/env python
"""
Module to generate a random string

Attributes:
    LENGTH (int): Defaults to a length of 16
"""

import argparse
import base64
import os
import sys

LENGTH = 16


def generate(length: int) -> str:
    """Simple function to call OpenSSL

    Args:
        length (int): length of string to return
    """
    return base64.encodebytes(os.urandom(256)).decode()[:length]


if __name__ == '__main__':

    def main() -> int:
        """Main function to call"""
        parser = argparse.ArgumentParser()
        parser.add_argument('-n', dest='length', default=LENGTH, type=int)
        args = parser.parse_args()

        sys.stdout.write(generate(args.length))

        return 0

    sys.exit(main())