mkpasswd

#!/usr/bin/env python
#-*- coding:utf-8 -*-
# created:     Mon Sep 15 13:27:15 2014
# filename:    mkpasswd.py
# author:      juntao liu
# email:       [email protected]
# descritpion:

import datetime
import random
import argparse
import os


LOG_FILENAME = os.path.expanduser("~/.mkpasswd.log")

LENGTH = 8

CHOICES = {
    'number': '0123456789'*10, # number
    'upper': 'ABCDEFGHIJKLMNOPQRSTUVWXYZ'*5,  # captical letter
    'lower': 'abcdefghijklmnopqrstuvwxyz'*5,  # lower case
    'special': '`~!@#$%^&*()_+-=[{]}\\|;:\'\",<.>/?' # special character
}


def log_password(password):
    if os.path.exists(LOG_FILENAME):
        with open(LOG_FILENAME, "a") as f:
            now = datetime.datetime.now()
            f.write("%s %s\n"%( now.strftime("%Y-%m-%d %H:%M:%S"), password))


def contains(s, choice):
    for ch in s:
        if ch in choice:
            return True
    return False

def mkpasswd(length, **kwargs):
    choices = CHOICES.copy()
    for k,v in  kwargs.items():
        if v and k in choices:
            choices.pop(k)
    password = ''
    for _ in range(length):
        password += random.choice(''.join(choices.values()))

    for choice in choices.values():
        if not contains(password, choice):
            return mkpasswd(length, **kwargs)
    return password

if __name__ == "__main__":
    parser = argparse.ArgumentParser("mkpasswd")
    parser.add_argument('-n', '--no-number', action="store_true", default=False, help="password without number")
    parser.add_argument('-u', '--no-upper', action="store_true", default=False, help="password without captial letter")
    parser.add_argument('-l', '--no-lower', action="store_true", default=False, help="password without lowercase")
    parser.add_argument('-s', '--no-special', action="store_true", default=False, help="password without special character")
    parser.add_argument('length', metavar="length", nargs="?", type=int, default=LENGTH, help="password length")

    ns, _ = parser.parse_known_args()
    password =  mkpasswd(ns.length, number=ns.no_number, lower=ns.no_lower, upper=ns.no_upper, special=ns.no_special)
    log_password(password)
    print(password)

usage: mkpasswd [-h] [-n] [-u] [-l] [-s] [length]

positional arguments:
  length            password length

optional arguments:
  -h, --help        show this help message and exit
  -n, --no-number   password without number
  -u, --no-upper    password without captial letter
  -l, --no-lower    password without lowercase
  -s, --no-special  password without special character

write password to ~/.mkpasswd.log , only if file exists.

touch ~/.mkpasswd.log, enable this feature