Use a single password_hash function
Using two different functions makes no sense in Python. A single function with an optional argument is more Pythonistic.
This commit is contained in:
parent
cd287242b0
commit
935b0acf26
2 changed files with 2 additions and 5 deletions
|
@ -63,10 +63,7 @@ class LibreAuthPassError(Exception):
|
||||||
else:
|
else:
|
||||||
self.message = 'unknown error'
|
self.message = 'unknown error'
|
||||||
|
|
||||||
def password_hash(password):
|
def password_hash(password, standard=NOSTANDARD):
|
||||||
return password_hash_standard(password, NOSTANDARD)
|
|
||||||
|
|
||||||
def password_hash_standard(password, standard):
|
|
||||||
pass_len = len(password)
|
pass_len = len(password)
|
||||||
if pass_len < PASSWORD_MIN_LEN:
|
if pass_len < PASSWORD_MIN_LEN:
|
||||||
raise LibreAuthPassError(1)
|
raise LibreAuthPassError(1)
|
||||||
|
|
|
@ -45,7 +45,7 @@ class PasswordTestCase(unittest.TestCase):
|
||||||
def test_std(self):
|
def test_std(self):
|
||||||
p = b'my super password'
|
p = b'my super password'
|
||||||
for std in (NOSTANDARD, NIST80063B, ):
|
for std in (NOSTANDARD, NIST80063B, ):
|
||||||
h = password_hash_standard(p, NIST80063B)
|
h = password_hash(p, standard=std)
|
||||||
self.assertTrue(h.startswith('$'))
|
self.assertTrue(h.startswith('$'))
|
||||||
self.assertEqual(len(h.split('$')), 5)
|
self.assertEqual(len(h.split('$')), 5)
|
||||||
self.assertTrue(is_valid(p, h))
|
self.assertTrue(is_valid(p, h))
|
||||||
|
|
Reference in a new issue