Задача по языку формальных грамматик

import sys
x=int(0)
y=int(0)
z=int(0)
a=input('Введите слово ')
if a.isalpha()==False:
        l=int(len(a))
        if a[0]=="0":
            for i in range (0,l):
                if a[i]=="0":
                    x+=1
                else: print("Введены некорректные цифорки")
                break
                if a[i]=="1":
                    y+=1
                else: print("Введены некорректные цифорки")
                break
                if a[i]=="2":
                    z+=1
                else: print("Введены некорректные цифорки")
                break
        else: print("Введены некорректные цифорки")
        sys.exit(0)
        if x==y==z:
            print("True")
        else:
            print("False")
else: print('Введены буковки')

def foo(word):
 
    if not set(word) or set(word) - {'0', '1', '2'}:
        return False
 
    tmp = len(word) // 3
    if word[:tmp] == '0' * tmp and\
            word[tmp:tmp * 2] == '1' * tmp and\
            word[tmp * 2:] == '2' * tmp:
        return True
    return False
 
 
print(foo(input()))

def check(word):
    n=len(word)
    k=n//3
    test="0"*k+"1"*k+"2"*k
    return test==word
 
print(check("001122"))
print(check("0000111122220"))

word = list(input('Введите слово:'))
sword = sorted(word)
x = word.count('0')
y = word.count('1')
z = word.count('2')
if((x == y == z) and word == sword): print('Слово принадлежит языку')
else: print('Слово не принадлежит языку')