kaqing 2.0.68__py3-none-any.whl → 2.0.70__py3-none-any.whl

adam/sql/sql_completer.py

@@ -194,10 +194,10 @@ class SqlCompleter(Completer):
             'select_a_comma_                 > name|*           > select_a                      ^ *',
             'select_a_                       > from             > select_from                   ^ from',
             'select_from_                    > name             > select_from_x                 ^ (select,tables',
-            '-                               > (                > select_from_x_lp_',
+            '-                               > {                > select_from_x_lp_',
             'select_from_x_lp_               > select           > select',
             'select_from_x                   > ,                > select_from_x_comma_',
-            '-                               > )                > select_from_x_rp_',
+            '-                               > }                > select_from_x_rp_',
             'select_from_x_rp__              > as               > select_from_x_as              ^ as',
             'select_from_x_comma_            > name             > select_from_x                 ^ tables',
             'select_from_x_                                                                     ^ as,where,inner join,left outer join,right outer join,full outer join,group by,limit',
@@ -219,6 +219,7 @@ class SqlCompleter(Completer):
             '-                               > full outer join  > select_join',
             'select_from_x_as_               > name             > select_from_x_as_x            ^ x,y,z',
             'select_from_x_as_x              > ,                > select_from_x_as_x_comma_',
+            '-                               > }                > select_where_sc_re_',
             'select_from_x_as_x_comma_       > name             > select_from_x                 ^ tables',
             'select_where_                   > name             > select_where_a                ^ id,x.,y.,z.',
             'select_where_a                  > comparison       > select_where_a_op             ^ =,<,<=,>,>=,<>,like,not,in',
@@ -237,7 +238,15 @@ class SqlCompleter(Completer):
             'select_where_a_not_op           > name|single|num  > select_where_sc               ^ single',
             'select_where_a_op               > name|single|num  > select_where_sc               ^ single',
             'select_where_sc                 > )                > select_where_sc_rp_',
-            'select_where_sc_rp__            > as               > select_from_x_as              ^ as',
+            'select_where_sc                 > }                > select_where_sc_re_',
+            'select_where_sc_re__            > as               > select_from_x_as              ^ as',
+            'select_where_sc_rp_             > }                > select_where_sc_re_',
+            'select_where_sc_rp__            > as               > select_from_x_as              ^ and,or,group by,limit',
+            '-                               > and|or           > select_where',
+            '-                               > group            > select_group',
+            '-                               > group by         > select_group_by',
+            '-                               > limit            > select_where_sc_limit',
+            '-                               > }                > select_where_sc_re_',
             'select_where_sc_                > and|or           > select_where                  ^ and,or,group by,limit',
             '-                               > group            > select_group',
             '-                               > group by         > select_group_by',
@@ -245,6 +254,7 @@ class SqlCompleter(Completer):
             'select_group_                   > by               > select_group_by               ^ by',
             'select_group_by_                > name             > select_group_by_a             ^ id,x.,y.,z.',
             'select_group_by_a               > ,                > select_group_by_a_comma_',
+            '-                               > }                > select_where_sc_re_',
             'select_group_by_a_comma_        > name             > select_group_by_a             ^ id,x.,y.,z.',
             'select_group_by_a_              > limit            > select_where_sc_limit         ^ limit',
             'select_where_sc_limit_          > num              > select_where_sc_limit_num     ^ 1',
@@ -269,6 +279,7 @@ class SqlCompleter(Completer):
             'select_x_join_y_on_a            > comparison       > select_x_join_y_on_a_op       ^ =',
             'select_x_join_y_on_a_op         > name             > select_x_join_y_on_a_op_b     ^ id,x.,y.,z.',
             'select_x_join_y_on_a_op_b       > _                > select_from_x_as_x_',
+            '-                               > }                > select_where_sc_re_',
 
 
             '                                > insert           > insert',
@@ -392,7 +403,7 @@ class SqlCompleter(Completer):
 
         return (machine, completers)
 
-    def traverse_tokens(self, text: str, tokens: list[Token], state: str = '', indent=0):
+    def traverse_tokens(self, text: str, tokens: list[Token], state: str = '', indent=0, lp_level = 0, lp_levels_to_match: set = set()):
         keywords = [
             'select', 'from', 'as', 'not', 'in', 'where',
             'and', 'or', 'group', 'by', 'group by', 'limit',
@@ -406,6 +417,20 @@ class SqlCompleter(Completer):
             'delete'
         ]
 
+        def is_opening_embrace():
+            if f'{state} > ' + '{' in self.machine[0]:
+                lp_levels_to_match.add(lp_level)
+                return '{'
+
+            return '('
+
+        def is_closing_embrace():
+            if f'{state} > ' + '}' in self.machine[0] and lp_level in lp_levels_to_match:
+                lp_levels_to_match.remove(lp_level)
+                return '}'
+
+            return ')'
+
         for token in tokens:
             if self.debug:
                 if token.ttype == T.Whitespace:
@@ -422,7 +447,7 @@ class SqlCompleter(Completer):
             # print("  " * indent + f"Token: {token.value}, Type: {token.ttype}@{token.ttype.__class__}")
 
             if token.is_group:
-                state = self.traverse_tokens(text, token.tokens, state, indent + 1)
+                state = self.traverse_tokens(text, token.tokens, state, indent + 1, lp_level)
             else:
                 it = ''
                 if (t := token.value.lower()) in keywords:
@@ -441,9 +466,12 @@ class SqlCompleter(Completer):
                     if token.value == ',':
                         it = ','
                     elif token.value == '(':
-                        it = '('
+                        it = is_opening_embrace()
+                        lp_level += 1
                     elif token.value == ')':
-                        it = ')'
+                        lp_level -= 1
+                        it = is_closing_embrace()
+
                 elif token.ttype == T.Operator.Comparison:
                     it = 'comparison'
 

adam/sql/term_completer.py

@@ -1,6 +1,6 @@
 from typing import Callable, Iterable, List, Mapping, Optional, Pattern, Union
 
-from prompt_toolkit.completion import CompleteEvent, Completer, Completion, WordCompleter
+from prompt_toolkit.completion import CompleteEvent, Completion, WordCompleter
 from prompt_toolkit.document import Document
 from prompt_toolkit.formatted_text import AnyFormattedText
 

adam/version.py

@@ -1,5 +1,5 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
-__version__ = "2.0.68"  #: the working version
+__version__ = "2.0.70"  #: the working version
 __release__ = "1.0.0"  #: the release version

{kaqing-2.0.68.dist-info → kaqing-2.0.70.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: kaqing
-Version: 2.0.68
+Version: 2.0.70
 Summary: UNKNOWN
 Home-page: UNKNOWN
 License: UNKNOWN

{kaqing-2.0.68.dist-info → kaqing-2.0.70.dist-info}/RECORD

@@ -14,7 +14,7 @@ adam/repl_commands.py,sha256=WA90Rl27Juctzr3U3kfCDk5N-oYMKlfWbZeafUgk7k0,4723
 adam/repl_session.py,sha256=uIogcvWBh7wd8QQ-p_JgLsyJ8YJgINw5vOd6JIsd7Vo,472
 adam/repl_state.py,sha256=nZO5CucSIGz68f4JreadGphRzMXPUzkTn-4XcoJJOug,8643
 adam/utils.py,sha256=2DoWsrcaioFFH0-RjT30qelVRPUJqCGTfz_ucfE7F8g,7406
-adam/version.py,sha256=VZJ1V0tn0r8qo89qxufoxCjB67gaQg6kLNEAEkIKX5g,139
+adam/version.py,sha256=yckbK4TX1cs1x8SSakRP16TVFCz-kWBTek3kfgTkIHA,139
 adam/checks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 adam/checks/check.py,sha256=Qopr3huYcMu2bzQgb99dEUYjFzkjKHRI76S6KA9b9Rk,702
 adam/checks/check_context.py,sha256=FEHkQ32jY1EDopQ2uYWqy9v7aEEX1orLpJWhopwAlh4,402
@@ -164,8 +164,8 @@ adam/k8s_utils/services.py,sha256=EOJJGACVbbRvu5T3rMKqIJqgYic1_MSJ17EA0TJ6UOk,31
 adam/k8s_utils/statefulsets.py,sha256=5g7KxGRHgEewT8rnZneDTaJDylUf-dHH2edWJEoorr8,4667
 adam/k8s_utils/volumes.py,sha256=RIBmlOSWM3V3QVXLCFT0owVOyh4rGG1ETp521a-6ndo,1137
 adam/sql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adam/sql/sql_completer.py,sha256=hjUzwMUrfqPAO8vufC74f9PiBuuvu3_FNVd5L2--xAQ,25133
-adam/sql/term_completer.py,sha256=17fuRyW2Uk-CoKySAh0S_UEzRQu9a-NxWII8jQnpAPU,2550
+adam/sql/sql_completer.py,sha256=VQfSPgNenkzIPKdER2p7NkU-82yyM-TDiSPHs_NkSJA,26705
+adam/sql/term_completer.py,sha256=bNnHAVf9NZl52xS_BQpikbOK39gDBJADnT9gSvG0iqI,2539
 adam/sso/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 adam/sso/authenticator.py,sha256=BCm16L9zf5aLU47-sTCnudn2zLPwd8M2wwRminJfsqw,615
 adam/sso/authn_ad.py,sha256=fDW8UR3WWykny5Awa5dQjjBUSFzIDz4aMn-lwXoABl8,5857
@@ -176,8 +176,8 @@ adam/sso/idp.py,sha256=fvcwUw_URTgsO6ySaqTIw0zQT2qRO1IPSGhf6rPtybo,5804
 adam/sso/idp_login.py,sha256=QAtCUeDTVWliJy40RK_oac8Vgybr13xH8wzeBoxPaa8,1754
 adam/sso/idp_session.py,sha256=9BUHNRf70u4rVKrVY1HKPOEmOviXvkjam8WJxmXSKIM,1735
 adam/sso/sso_config.py,sha256=5N8WZgIJQBtHUy585XLRWKjpU87_v6QluyNK9E27D5s,2459
-kaqing-2.0.68.dist-info/METADATA,sha256=3pgPlx0IZRVQJ24RUGpkCJrmXEbGedBmYAcUCXPw-cc,132
-kaqing-2.0.68.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-kaqing-2.0.68.dist-info/entry_points.txt,sha256=SkzhuQJUWsXOzHeZ5TgQ2c3_g53UGK23zzJU_JTZOZI,39
-kaqing-2.0.68.dist-info/top_level.txt,sha256=8_2PZkwBb-xDcnc8a2rAbQeJhXKXskc7zTP7pSPa1fw,5
-kaqing-2.0.68.dist-info/RECORD,,
+kaqing-2.0.70.dist-info/METADATA,sha256=CXfFNyxJYztdM8ZE6MW7MSPJ6LmxM63_fNGPBAH4THU,132
+kaqing-2.0.70.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+kaqing-2.0.70.dist-info/entry_points.txt,sha256=SkzhuQJUWsXOzHeZ5TgQ2c3_g53UGK23zzJU_JTZOZI,39
+kaqing-2.0.70.dist-info/top_level.txt,sha256=8_2PZkwBb-xDcnc8a2rAbQeJhXKXskc7zTP7pSPa1fw,5
+kaqing-2.0.70.dist-info/RECORD,,