Python实现的密码强度检测器示例

# coding: utf-8"""checkCheck if your password safe"""import re# 特征NUMBER = re.compile(r'[0-9]')LOWER_CASE = re.compile(r'[a-z]')UPPER_CASE = re.compile(r'[A-Z]')OTHERS = re.compile(r'[^0-9A-Za-z]')def load_common_password(): words = [] with open("10k_most_common.txt", "r") as f:  for word in f:   words.append(word.strip()) return wordsCOMMON_WORDS = load_common_password()# 管理密码强度的类class Strength(object): """ 密码强度三个属性：是否有效valid, 强度strength, 提示信息message """ def __init__(self, valid, strength, message):  self.valid = valid  self.strength = strength  self.message = message def __repr__(self):  return self.strength def __str__(self):  return self.message def __bool__(self):  return self.validclass Password(object): TERRIBLE = 0 SIMPLE = 1 MEDIUM = 2 STRONG = 3 @staticmethod def is_regular(input):  regular = ''.join(['qwertyuiop', 'asdfghjkl', 'zxcvbnm'])  return input in regular or input[::-1] in regular @staticmethod def is_by_step(input):  delta = ord(input[1]) - ord(input[0])  for i in range(2, len(input)):   if ord(input[i]) - ord(input[i - 1]) != delta:    return False  return True @staticmethod def is_common(input):  return input in COMMON_WORDS def __call__(self, input, min_length=6, min_type=3, level=STRONG):  if len(input) < min_length:   return Strength(False, "terrible", "密码太短了")  if self.is_regular(input) or self.is_by_step(input):   return Strength(False, "simple", "密码有规则")  if self.is_common(input):   return Strength(False, "simple", "密码很常见")  types = 0  if NUMBER.search(input):   types += 1  if LOWER_CASE.search(input):   types += 1  if UPPER_CASE.search(input):   types += 1  if OTHERS.search(input):   types += 1  if types < 2:   return Strength(level <= self.SIMPLE, "simple", "密码太简单了")  if types < min_type:   return Strength(level <= self.MEDIUM, "medium", "密码还不够强")  return Strength(True, "strong", "密码很强")class Email(object): def __init__(self, email):  self.email = email def is_valid_email(self):  if re.match("^.+@(//[?)[a-zA-Z0-9//-//.]+//.([a-zA-Z]{2,3}|[0-9]{1,3})(//]?)$", self.email):   return True  return False def get_email_type(self):  types = ['qq', '163', 'gmail', '126', 'sina']  email_type = re.search('@/w+', self.email).group()[1:]  if email_type in types:   return email_type  return 'wrong email'password = Password()

# coding: utf-8"""test for check"""import unittestimport checkclass TestCheck(unittest.TestCase): def test_regular(self):  rv = check.password("qwerty")  self.assertTrue(repr(rv) == "simple")  self.assertTrue('规则' in rv.message) def test_by_step(self):  rv = check.password("abcdefg")  self.assertTrue(repr(rv) == "simple")  self.assertTrue('规则' in rv.message) def test_common(self):  rv = check.password("password")  self.assertTrue(repr(rv) == "simple")  self.assertTrue('常见' in rv.message) def test_medium(self):  rv = check.password("ahj01a")  self.assertTrue(repr(rv) == 'medium')  self.assertTrue('不够强' in rv.message) def test_strong(self):  rv = check.password("asjka9AD")  self.assertTrue(repr(rv) == 'strong')  self.assertTrue('很强' in rv.message) # 测试邮箱 def test_email(self):  rv = check.Email("123@gmail.com")  self.assertEqual(rv.is_valid_email(), True) def test_email_type(self):  rv = check.Email("123@gmail.com")  types = ['qq', '163', 'gmail', '126', 'sina']  self.assertIn(rv.get_email_type(), types)if __name__ == '__main__': unittest.main()