JezK
Edit File: crypt.py
"""Wrapper to the POSIX crypt library call and associated functionality. Note that the ``methods`` and ``METHOD_*`` attributes are non-standard extensions to Python 2.7, backported from 3.3""" import _crypt import string as _string from random import SystemRandom as _SystemRandom from collections import namedtuple as _namedtuple _saltchars = _string.ascii_letters + _string.digits + './' _sr = _SystemRandom() class _Method(_namedtuple('_Method', 'name ident salt_chars total_size')): """Class representing a salt method per the Modular Crypt Format or the legacy 2-character crypt method.""" def __repr__(self): return '<crypt.METHOD_%s>' % self.name def mksalt(method=None): """Generate a salt for the specified method. If not specified, the strongest available method will be used. This is a non-standard extension to Python 2.7, backported from 3.3 """ if method is None: method = methods[0] s = '$%s$' % method.ident if method.ident else '' s += ''.join(_sr.sample(_saltchars, method.salt_chars)) return s def crypt(word, salt=None): """Return a string representing the one-way hash of a password, with a salt prepended. If ``salt`` is not specified or is ``None``, the strongest available method will be selected and a salt generated. Otherwise, ``salt`` may be one of the ``crypt.METHOD_*`` values, or a string as returned by ``crypt.mksalt()``. Note that these are non-standard extensions to Python 2.7's crypt.crypt() entrypoint, backported from 3.3: the standard Python 2.7 crypt.crypt() entrypoint requires two strings as the parameters, and does not support keyword arguments. """ if salt is None or isinstance(salt, _Method): salt = mksalt(salt) return _crypt.crypt(word, salt) # available salting/crypto methods METHOD_CRYPT = _Method('CRYPT', None, 2, 13) METHOD_MD5 = _Method('MD5', '1', 8, 34) METHOD_SHA256 = _Method('SHA256', '5', 16, 63) METHOD_SHA512 = _Method('SHA512', '6', 16, 106) methods = [] for _method in (METHOD_SHA512, METHOD_SHA256, METHOD_MD5): _result = crypt('', _method) if _result and len(_result) == _method.total_size: methods.append(_method) methods.append(METHOD_CRYPT) del _result, _method