parser/parser.py

   1 # Parse query
   2
   3 from ply import lex, yacc
   4
   5 literals = '()'
   6
   7 tokens = ('NAME', 'AND_OP', 'OR_OP', 'NOT_OP', 'SP1')
   8
   9 t_NAME = '[a-z][a-z0-9_]*'
  10
  11 t_AND_OP = '&'
  12
  13 t_OR_OP = r'\|'
  14
  15 t_NOT_OP = '!'
  16
  17 t_SP1 = '[ \t]+'
  18
  19 def t_error(t):
  20     """Avoid warnings on stderr"""
  21
  22 lexer = lex.lex()
  23
  24 def p_expression_name(p):
  25     """expression : NAME"""
  26     p[0] = ('NAME', p[1])
  27
  28 def p_expression_and_and(p):
  29     """expression : expression SP0 AND_OP AND_OP SP0 expression"""
  30     p[0] = ('AND', p[1], p[6])
  31
  32 def p_expression_and(p):
  33     """expression : expression SP0 AND_OP SP0 expression"""
  34     p[0] = ('AND', p[1], p[5])
  35
  36 def p_expression_and_word(p):
  37     """expression : l_expression and_word r_expression"""
  38     p[0] = ('AND', p[1], p[3])
  39
  40 def p_expression_not(p):
  41     """expression : NOT_OP SP0 expression"""
  42     p[0] = ('NOT', p[3])
  43
  44 def p_expression_or_or(p):
  45     """expression : expression SP0 OR_OP OR_OP SP0 expression"""
  46     p[0] = ('OR', p[1], p[6])
  47
  48 def p_expression_or(p):
  49     """expression : expression SP0 OR_OP SP0 expression"""
  50     p[0] = ('OR', p[1], p[5])
  51
  52 def p_expression_in_parens(p):
  53     """expression : expression_parens"""
  54     p[0] = p[1]
  55
  56 def p_l_expression(p):
  57     """l_expression : expression_parens
  58                     | expression SP1
  59     """
  60     if len(p) == 2:
  61         p[0] = p[1]
  62     elif len(p) == 3:
  63         p[0] = p[1]
  64     else:
  65         raise ValueError(p)
  66
  67 def p_r_expression(p):
  68     """r_expression : expression_parens
  69                     | SP1 expression
  70     """
  71     if len(p) == 2:
  72         p[0] = p[1]
  73     elif len(p) == 3:
  74         p[0] = p[2]
  75     else:
  76         raise ValueError(p)
  77
  78 def p_expression_parens(p):
  79     """expression_parens : '(' SP0 expression SP0 ')'"""
  80     p[0] = ('PARENS', p[3])
  81
  82 def p_and_word(p):
  83     """and_word : NAME"""
  84     if p[1] in ('AND', 'and'):
  85         p[0] = p[1]
  86     else:
  87         raise SyntaxError
  88
  89 def p_SP0(p):
  90     """SP0 : SP1
  91            | empty
  92     """
  93
  94 def p_empty(p):
  95     """empty :"""
  96
  97 def p_error(p):
  98     """Avoid warnings on stderr"""
  99     yacc.restart()
 100
 101 precedence = (
 102     ('left', 'OR_OP'),
 103     ('left', 'AND_OP'),
 104     ('right', 'NOT_OP'),
 105 )
 106
 107 parser = yacc.yacc()