gen http auth passwd file

nginx_auth.py

import os
import sys
import random

# more info
# http://trac.edgewall.org/export/10770/trunk/contrib/htpasswd.py

# We need a crypt module, but Windows doesn't have one by default.  Try to find
# one, and tell the user if we can't.
try:
    import crypt
except ImportError:
    try:
        import fcrypt as crypt
    except ImportError:
        sys.stderr.write("Cannot find a crypt module.  "
                         "Possibly http://carey.geek.nz/code/python-fcrypt/\n")
        sys.exit(1)


def salt():
    """Returns a string of 2 randome letters"""
    letters = 'abcdefghijklmnopqrstuvwxyz' \
              'ABCDEFGHIJKLMNOPQRSTUVWXYZ' \
              '0123456789/.'
    return random.choice(letters) + random.choice(letters)


def gen(username, password):
    """Replace the entry for the given user, or add it if new."""
    pwhash = crypt.crypt(password, salt())
    return "{}:{}".format(username, pwhash)


def usage():
    sys.stderr.write("Usage: python nginx_auth.py username password\n")


def main():
    try:
        username = sys.argv[1]
        password = sys.argv[2]
    except IndexError:
        usage()
        sys.exit(1)
    print gen(username, password)


if __name__ == "__main__":
    main()