uzume

#!/usr/bin/env python
import random, string
import argparse
import os
import sys

arg_to_charset = {
        'uppercase': string.ascii_letters,
        'numbers': string.digits,
        'symbols': string.punctuation
}

parser = argparse.ArgumentParser()
parser.add_argument("length", default=10, type=int, nargs='?', help='Length of the generated password. Defaults to 10')
parser.add_argument("-u", "--uppercase", action='store_true', required=False, help='Include uppercase letters')
parser.add_argument("-n", "--numbers", action='store_true', required=False, help='Include numbers')
parser.add_argument("-s", "--symbols", action='store_true', required=False, help='Include symbols')

a = parser.parse_args()

def randompass():
    length = a.length
    charsets = [string.ascii_lowercase]

    for arg, charset in arg_to_charset.items():
        if getattr(a, arg):
            charsets.append(charset)

    output = ''.join(random.choice(''.join(charsets)) for _ in range(length))

    if "NO_COLOR" in os.environ or not sys.stdout.isatty():
        return output
    else:
        return '\033[93m' + output + '\033[0m' # colored output


if __name__ == '__main__':
    print(randompass())