from lark import Lark
gramatica = """
    start: atribuicao
    atribuicao: ID "=" NUMERO
    ID: /[a-zA-Z_]+/
    NUMERO: /[0-9]+/
    %import common.WS
    %ignore WS
"""
parser = Lark(gramatica)
print(parser.parse("x = 50").pretty())