检验密码的强度，一直无效语法

Question

检验密码强度，我写的总是无效语法，没有思路了，指点一下呀，这种定义函数，真的是好难

Answer 1

import string
def main(pwd):
    dig=lCase =hCase=punnctuation=  0
    if len(pwd)<6:
        print('weak')
    for ch in pwd:
        if ch in string.digits:
            dig=1
        elif ch in string.ascii_lowercase:
            lCase=1
        elif ch in string.ascii_uppercase:
            hCase=1
        elif ch in string.punctuation:
            punnctuation=1
    if dig+lCase+hCase+punnctuation == 1:
        print("weak")
    elif dig+lCase+hCase+punnctuation == 2:
        print("below_middle")
    elif dig+lCase+hCase+punnctuation == 3:
        print("above_middle")
    elif dig+lCase+hCase+punnctuation == 4:
        print("strong")
    else:
        print('weak')

main(input("input code:"))