pyton 实在不会 怎么做呀

年终奖应缴纳税款计算:
年终奖小于或等于4万元时，按3%缴纳税款;
年终奖大于4万，小于或等于10万元时，4万元以下的按3%缴纳，超出4万的，按10%缴纳。
年终奖10万到20万之间时，4万元以下的按3%缴纳，4万到10万的按10%缴纳，高于10万元按20%缴纳。
年终奖20万以上时，4万元以下的按3%缴纳，4万到10万
的按10%，10万到20万的按20%，高于20万元按50%缴纳。从键盘输入年终奖，求应缴纳税金?

flag = 1
def nsjs(njz):
    sk_num = 0.0
    temp = njz
    if temp > 200000:
        sk = (temp - 200000) * 0.5
        sk_num = sk_num + sk
        temp = 200000
    if temp > 100000:
        sk = (temp - 100000) * 0.2
        sk_num = sk_num + sk
        temp = 100000
    if temp > 40000:
        sk = (temp - 40000) * 0.1
        sk_num = sk_num + sk
        temp = 40000
    sk = temp * 0.03
    sk_num = sk_num + sk
    return sk_num

while flag == 1:
    try:
        njz = int(input("请输入年终奖（单位元），0退出："))
        if njz == 0:
            break
        print("年终奖{}元，应缴纳税款{}元".format(njz,nsjs(njz)))
    except Exception as e:
        print("错误，",repr(e))