Skip to content

Commit 1744d4f

Browse files
committed
variable can have digit in the end (@var1)
1 parent 0698724 commit 1744d4f

File tree

1 file changed

+2
-10
lines changed

1 file changed

+2
-10
lines changed

mindsdb_sql_parser/lexer.py

Lines changed: 2 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -350,11 +350,7 @@ def DQUOTE_STRING(self, t):
350350
def ignore_newline(self, t):
351351
self.lineno += len(t.value)
352352

353-
@_(r'@[a-zA-Z_.$]+',
354-
r"@'[a-zA-Z_.$][^']*'",
355-
r"@`[a-zA-Z_.$][^`]*`",
356-
r'@"[a-zA-Z_.$][^"]*"'
357-
)
353+
@_(r'@(?:([a-zA-Z_$0-9]*[a-zA-Z_$]+[a-zA-Z_$0-9]*)|(?:`([^`]+)`))')
358354
def VARIABLE(self, t):
359355
t.value = t.value.lstrip('@')
360356

@@ -366,11 +362,7 @@ def VARIABLE(self, t):
366362
t.value = t.value.strip('`')
367363
return t
368364

369-
@_(r'@@[a-zA-Z_.$]+',
370-
r"@@'[a-zA-Z_.$][^']*'",
371-
r"@@`[a-zA-Z_.$][^`]*`",
372-
r'@@"[a-zA-Z_.$][^"]*"'
373-
)
365+
@_(r'@@(?:([a-zA-Z_$0-9]*[a-zA-Z_$]+[a-zA-Z_$0-9]*)|(?:`([^`]+)`))')
374366
def SYSTEM_VARIABLE(self, t):
375367
t.value = t.value.lstrip('@')
376368

0 commit comments

Comments
 (0)