python -- 密码生成器

代码可能写的特别粗糙， 请自行优化.

#!/usr/bin/env python

import itertools
import string
import sys


def usage():
    print "Password generator - v1.0 "
    print "\t%s [type] [length]" % sys.argv[0]
    print "\ttype: l for letters, d for digits, s for singals"


def generator(lsttype, length):
    """
    generator passwords with letters and signals
    Create by hap.ddup@gmail.com
    """
    pass_lst = []
    chars = {'l': list(string.letters),
             'd': list(string.digits),
             's': list("#$%&\'()*+,-./:;<=>?@[\\]^_`{|}~ ")
             }

    for i in lsttype:
        try:
            pass_lst.extend(chars[i])
        except KeyError:
            usage()
            sys.exit(1)

    for i in itertools.product(pass_lst, repeat=length):
        print "".join(i)

if __name__ == '__main__':
    if sys.argv.__len__() != 3:
        usage()
    else:
        generator(sys.argv[1], int(sys.argv[2]))