Python - Rply назначить несколько разных правил для нескольких разных функций - PullRequest
Новая
       30

Python - Rply назначить несколько разных правил для нескольких разных функций

0 голосов
/ 07 апреля 2019

Предположим, у меня есть код python-rply, который выглядит следующим образом (взято из здесь ): 

from rply import ParserGenerator, LexerGenerator
from rply.token import BaseBox

lg = LexerGenerator()
# Add takes a rule name, and a regular expression that defines the rule.
lg.add("PLUS", r"\+")
lg.add("MINUS", r"-")
lg.add("NUMBER", r"\d+")

lg.ignore(r"\s+")

# This is a list of the token names. precedence is an optional list of
# tuples which specifies order of operation for avoiding ambiguity.
# precedence must be one of "left", "right", "nonassoc".
# cache_id is an optional string which specifies an ID to use for
# caching. It should *always* be safe to use caching,
# RPly will automatically detect when your grammar is
# changed and refresh the cache for you.
pg = ParserGenerator(["NUMBER", "PLUS", "MINUS"],
        precedence=[("left", ['PLUS', 'MINUS'])], cache_id="myparser")

@pg.production("main : expr")
def main(p):
    # p is a list, of each of the pieces on the right hand side of the
    # grammar rule
    return p[0]

@pg.production("expr : expr PLUS expr")
@pg.production("expr : expr MINUS expr")
def expr_op(p):
    lhs = p[0].getint()
    rhs = p[2].getint()
    if p[1].gettokentype() == "PLUS":
        return BoxInt(lhs + rhs)
    elif p[1].gettokentype() == "MINUS":
        return BoxInt(lhs - rhs)
    else:
        raise AssertionError("This is impossible, abort the time machine!")

@pg.production("expr : NUMBER")
def expr_num(p):
    return BoxInt(int(p[0].getstr()))

lexer = lg.build()
parser = pg.build()

class BoxInt(BaseBox):
    def __init__(self, value):
        self.value = value

    def getint(self):
        return self.value

Это простой код, поэтому при вводе этого кода: 

parser.parse(lexer.lex("1 + 3"))

Он будет выполнен, давая вам 4 в качестве вывода и ответа.Это рабочий код, но все еще нуждается в улучшении.Часть кода, где @pg.production вызывается для сложения и вычитания, не очень эффективна;Под этим я подразумеваю, что если добавить к этому еще несколько операторов, это будет очень тесно.Есть ли хороший способ сделать не стесненную версию этой части, которая может выглядеть примерно так: 

@pg.production("expr : expr PLUS expr")
def plus(p):
    lhs = p[0].getint()
    rhs = p[2].getint()
    if p[1].gettokentype() == "PLUS":
        return BoxInt(lhs + rhs)
    else:
        raise AssertionError("This is impossible, abort the time machine!")

@pg.production("expr : expr MINUS expr")
def minus(p):
    lhs = p[0].getint()
    rhs = p[2].getint()

    if p[1].gettokentype() == "MINUS":
        return BoxInt(lhs - rhs)
    else:
        raise AssertionError("This is impossible, abort the time machine!")

ПРИМЕЧАНИЕ. Я использую rply , а не слой , но они очень похожи.

1 Ответ

1 голос
/ 07 апреля 2019

Если вы разделите функции так, чтобы у каждого производства была своя собственная функция - что, действительно, является наилучшей практикой, - тогда абсолютно бесполезно проверять тип токена оператора.Вы знаете, что это такое, потому что логика парсера означает, что функция будет вызываться только при совпадении с производством.

Так что вы можете написать достаточно компактный код: 

@pg.production("expr : expr PLUS expr")
def plus(p):
    return BoxInt(p[0].getint() +  p[2].getint())

@pg.production("expr : expr MINUS expr")
def minus(p):
    return BoxInt(p[0].getint() -  p[2].getint())
...