Python检测密码强度

一般地，可以作为密码字符的主要有数字、小写字母、大写字母和几个标点符号。密码安全强度主要和字符串的复杂程度有关系，字符串中包含的字符种类越多，认为其安全强度越高。按照这个标准，可以把安全强度分为强密码、中高、中低、弱密码。其中强密码表示字符串中同时含有数字、小写字母、大写字母、标点符号这四类自负，而弱密码表示字符串中仅包含4类字符中的一种。

编写程序，输入一个字符串，输出该字符串作为密码时的安全强度。

from string import digits, ascii_lowercase, ascii_uppercase


def check(pwd):
    # 密码必须至少包含6个字符
    if not isinstance(pwd, str) or len(pwd) < 6:
        return 'not suitable for password'
    # 密码强度等级与包含的种类的对应关系
    d = {1: 'weak', 2: 'below middle', 3: 'above middle', 4: 'strong'}
    # 分别用来标记pwd是否含有数字，大小写字母
    # 大写字母和指定的标点符号
    r = [False] * 4

    for ch in pwd:
        # 是否包含数字
        if not r[0] and ch in digits:
            r[0] = True
        # 是否包含小写字母
        elif not r[1] and ch in ascii_lowercase:
            r[1] = True
        # 是否包含大写字母
        elif not r[2] and ch in ascii_lowercase:
            r[2] = True
        # 是否包含指定的标点符号
        elif not r[3] and ch in ',.!;?<>':
            r[2] = True
    # 统计包含的字符种类，返回密码强度
    return d.get(r.count(True), 'error')


print(check('a2Cd,abc'))