dump @@ to string

Author: StpMax

mindsdb_sql_parser/utils.py

@@ -81,10 +81,18 @@ def tokens_to_string(tokens):
         # filling space between tokens
         line += ' '*(token.index - shift - len(line))
 
+        match token.type:
+            case 'VARIABLE':
+                token_value = '@' + token.value
+            case 'SYSTEM_VARIABLE':
+                token_value = '@@' + token.value
+            case _:
+                token_value = token.value
+
         # add token
-        line += token.value
+        line += token_value
 
-        last_pos = token.index + len(token.value)
+        last_pos = token.index + len(token_value)
 
     # last line
     content += line