-#!/usr/bin/env python
+#!/usr/bin/env python3
-import random, string, sys
+import random, sys
-chars = filter(lambda c: c.isalnum(), map(chr, range(256)))
-chars = chars * 2
-chars += "!@#%^&*(){}[]/=?+_-;:,.<>"
-
-def generate(length):
- return "".join([random.choice(chars) for i in xrange(length)])
-
-if __name__ == "__main__":
- try:
- n = int(sys.argv[1])
- except:
- print "Usage: %s length" % sys.argv[0]
- sys.exit(1)
-
- print generate(n)
+alnum_chars = list(filter(lambda c: c.isalnum(), map(chr, range(128))))
+full_chars = alnum_chars * 2 + list("!@#%^&*(){}[]/=?+_-;:,.<>")
+def generate(chars, length):
+ return "".join([random.choice(chars) for i in range(length)])
+def print_usage():
+ print("Usage: %s [length] [alnum|full]" % sys.argv[0])
+ sys.exit(1)
+if __name__ == "__main__":
+ if len(sys.argv) <= 1:
+ n = 20
+ elif sys.argv[1].isdigit():
+ n = int(sys.argv[1])
+ else:
+ print_usage()
+
+ if len(sys.argv) <= 2:
+ chars = alnum_chars
+ elif sys.argv[2] == "alnum":
+ chars = alnum_chars
+ elif sys.argv[2] == "full":
+ chars = full_chars
+ else:
+ print_usage()
+
+ print(generate(chars, n))