roach57
Python,Linux,运维自动化,Shell

导航

 
1 2 4 8
密码存在大写字母 a赋值1:
密码存在小写字母 b赋值2:
密码存在数字        c赋值4:
密码存在特殊字符 d赋值8:

当 len(pwd) >= 8  &&  a+b+c+d = 7 or 11 or 13 or 14的时候密码强度符合要求;
7  11 13 14  


密码强度

密码强度如何量化呢? 
一个密码可以有以下几种类型:长度、大写字母、小写字母、数字以及特殊符号。 
显然,密码包含的特征越多、长度越长,其强度也就越高。 
我们设置几个等级来评测密码强度,分别是:terriblesimple
mediumstrong。 
不同的应用可能对密码强度的要求不一样,我们引入最小程度(min_length)和最小特征数(min_types),作为可配置选项。 
这样我们就可以检测密码包含的特征,特征与密码之间的关系可以简单定义为:

特征数强度
小于最小长度terrible
常用密码或规则的密码simple
小于最小特征数medium
大于或等于最小特征数strong

常用的密码可以从这里获取。

代码实现

check.py

# coding: utf-8

"""
check
Check 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 words

COMMON_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.valid


class 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()

test_check.py: 用于单元测试

# coding: utf-8

"""
test for check
"""
import unittest
import check


class 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()




posted on 2017-05-11 11:10  roach57  阅读(515)  评论(0编辑  收藏  举报