aboutsummaryrefslogtreecommitdiff
path: root/ql/parser/lexer.py
blob: 6e53f121a61f8678cd4e17e5a76b10f2d687503b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
from re import findall

from ply.lex import lex


class QLLexer:
    def __init__(self):
        self.__errors = []
        self.__lexer = None

    @property
    def lexer(self):
        return self.__lexer

    @property
    def errors(self):
        return self.__errors

    def build(self):
        self.__lexer = lex(module=self)

    def tokenize(self, data):
        self.__errors = []
        self.lexer.input(data)
        token = self.lexer.token()
        tokens = [token]

        while token:
            token = self.lexer.token()
            tokens.append(token)

        return tokens

    tokens = [
        'PLUS', 'MINUS', 'TIMES', 'DIVIDE', 'COLON',
        'ASSIGN',
        'EQUALS', 'NOT_EQUALS',
        'LESS_EQUALS', 'LESS_THAN', 'GREATER_EQUALS', 'GREATER_THAN',
        'AND', 'OR',
        'DOLLAR', 'RUBLE',
        'LEFT_BRACE', 'RIGHT_BRACE',
        'LEFT_BRACKET', 'RIGHT_BRACKET',
        'INTEGER_LITERAL', 'DECIMAL_LITERAL',
        'TRUE', 'FALSE',
        'DATE_LITERAL',
        'STRING_LITERAL',
        'IDENTIFIER',
    ]

    reserved_keywords = {
        'form':     'FORM',
        'if':       'IF',
        'not':      'NOT',

        # Types
        'boolean':  'BOOLEAN',
        'date':     'DATE',
        'decimal':  'DECIMAL',
        'integer':  'INTEGER',
        'money':    'MONEY',
        'string':   'STRING',
    }

    tokens += list(reserved_keywords.values())

    # Regular expression rules for simple tokens
    t_ignore = ' \t'

    t_PLUS = r'\+'
    t_MINUS = r'-'
    t_TIMES = r'\*'
    t_DIVIDE = r'/'
    t_COLON = r':'
    t_ASSIGN = r'='

    t_LESS_EQUALS = r'<='
    t_LESS_THAN = r'<'
    t_GREATER_EQUALS = r'>='
    t_GREATER_THAN = r'>'
    t_EQUALS = r'=='
    t_NOT_EQUALS = r'!='
    t_AND = r'&&'
    t_OR = r'\|\|'

    t_DOLLAR = r'\$'
    t_RUBLE = r'\₽'

    t_LEFT_BRACE = r'\{'
    t_RIGHT_BRACE = r'\}'

    t_LEFT_BRACKET = r'\('
    t_RIGHT_BRACKET = r'\)'

    # Literals
    @staticmethod
    def t_FALSE(token):
        r'False'
        token.value = False
        return token

    @staticmethod
    def t_TRUE(token):
        r'True'
        token.value = True
        return token

    def t_DATE_LITERAL(self, token):
        r'date\(\s*\d{1,2}\s*,\s*\d{1,2}\s*,\s*\d{1,4}\s*\)'

        try:
            token.value = findall(r'\d\d*', token.value)
            return token
        except SyntaxError:
            self.errors.append('Invalid date.')

    @staticmethod
    def t_DECIMAL_LITERAL(token):
        r'\d+[.]\d+'
        token.value = token.value
        return token

    @staticmethod
    def t_INTEGER_LITERAL(token):
        r'\d+'
        token.value = token.value
        return token

    @staticmethod
    def t_STRING_LITERAL(token):
        r'\"(.+?)\"'
        token.value = token.value[1:-1]
        return token

    # Other
    def t_IDENTIFIER(self, token):
        r'[a-z][a-zA-Z_0-9]*'
        token.type = self.reserved_keywords.get(token.value, 'IDENTIFIER')
        return token

    @staticmethod
    def t_comment(token):
        r'//.*'
        pass

    @staticmethod
    def t_newline(token):
        r'\n+'
        token.lexer.lineno += len(token.value)

    # Error handling
    def t_error(self, token):
        self.errors.append("Illegal character '%s'" % token.value[0])
        token.lexer.skip(1)