python之判断变量名是否合法

问题：

变量名是否合法:
         1.变量名可以由字母,数字或者下划线组成
         2.变量名只能以字母或者下划线开头
         s = 'hello@'

         1.判断变量名的第一个元素是否为字母或者下划线 s[0]
         2.如果第一个元素符合条件,判断除了第一个元素之外的其他元素s[1:]

思路：

         1.变量名的第一个字符是否为字母或下划线
         2.如果是，继续判断 --> 4
         3.如果不是，报错
         4.依次判断除了第一个字符之外的其他字符
         5.判断是否为字母数字或者下划线

while True:
    s = input('变量名:')
    #定义退出
    if s == 'exit':
        print('欢迎下次使用')
        break
    #判断字符串第一个变量是否满足条件
    if s[0].isalpha() or s[0] == '_':
        for i in s[1:]:
            #判断字符串以后的变量是否满足条件
            if not(i.isalnum() or i == '_'):
                print('%s变量名不合法' %s)
                break
        else:
            print('%s变量名合法' %s)
    else:
        print('%s变量名不合法' %s)