kaqing 2.0.55__py3-none-any.whl → 2.0.56__py3-none-any.whl

adam/sql/sql_completer.py

@@ -7,7 +7,10 @@ from sqlparse import tokens as T
 
 from adam.sql.term_completer import TermCompleter
 
+columns = TermCompleter(['id', 'x.', 'y.', 'z.'])
+
 class SqlCompleter(Completer):
+
     def __init__(self, tables: Callable[[], list[str]], dml: str = None, debug = False):
         super().__init__()
         self.dml = dml
@@ -45,11 +48,17 @@ class SqlCompleter(Completer):
             elif state == "select_a_from_":
                 completer = TermCompleter(self.tables())
             elif state == "select_a_from_x_":
+                completer = TermCompleter(['as', 'where', 'inner', 'left', 'right', 'full', 'group', 'limit'])
+            elif state == "select_a_from_x_as_x_":
                 completer = TermCompleter(['where', 'inner', 'left', 'right', 'full', 'group', 'limit'])
             elif state == "select_a_from_x,":
                 completer = TermCompleter(self.tables())
+            elif state == "select_a_from_x_as_":
+                completer = TermCompleter(['x', 'y', 'z'])
+            elif state == "select_a_from_x_as_x,":
+                completer = TermCompleter(self.tables())
             elif state == "select_a_from_x_where_":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "select_a_from_x_where_id":
                 completer = TermCompleter(['=', '<', '<=', '>', '>=', '<>', 'like'])
             elif state == "select_a_from_x_where_id=":
@@ -61,9 +70,9 @@ class SqlCompleter(Completer):
             elif state == "select_a_from_x_group_":
                 completer = TermCompleter(['by'])
             elif state == "select_a_from_x_group_by_":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "select_a_from_x_group_by_a,":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "select_a_from_x_group_by_a_":
                 completer = TermCompleter(['limit'])
             elif state == "select_a_from_x_group_by_a_limit_":
@@ -77,11 +86,11 @@ class SqlCompleter(Completer):
             elif state == "select_a_from_x_inner_join_y_":
                 completer = TermCompleter(['on'])
             elif state == "select_a_from_x_inner_join_y_on_":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "select_a_from_x_inner_join_y_on_a":
                 completer = TermCompleter(['='])
             elif state == "select_a_from_x_inner_join_y_on_a=":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "select_a_from_x_inner_join_y_on_a=b_":
                 completer = TermCompleter(['where', 'group', 'limit'])
             elif state == "select_a_from_x_left_":
@@ -104,9 +113,9 @@ class SqlCompleter(Completer):
             elif state == "insert_into_x_":
                 completer = TermCompleter(['values'])
             elif state == "insert_into_x(":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "insert_into_x(a,":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "insert_into_x(a)_":
                 completer = TermCompleter(['values('])
             elif state == "insert_into_x_values":
@@ -119,7 +128,7 @@ class SqlCompleter(Completer):
             elif state == "update_x_":
                 completer = TermCompleter(['set'])
             elif state in ["update_x_set_", "update_x_set_a=v,"]:
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "update_x_set_a":
                 completer = TermCompleter(['='])
             elif state == "update_x_set_a=":
@@ -127,7 +136,7 @@ class SqlCompleter(Completer):
             elif state == "update_x_set_a=v_":
                 completer = TermCompleter(['where'])
             elif state == "update_x_set_a=v_where_":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "update_x_set_a=v_where_id":
                 completer = TermCompleter(['='])
             elif state == "update_x_set_a=v_where_id=v_":
@@ -140,7 +149,7 @@ class SqlCompleter(Completer):
             elif state == "delete_from_x_":
                 completer = TermCompleter(['where'])
             elif state == "delete_from_x_where_":
-                completer = TermCompleter(['id'])
+                completer = columns
             elif state == "delete_from_x_where_id":
                 completer = TermCompleter(['='])
             elif state == "delete_from_x_where_id=":
@@ -215,8 +224,12 @@ class SqlCompleter(Completer):
                 elif state == 'select_a_from_x,':
                     if token.ttype == T.Name:
                         state = 'select_a_from_x'
-                elif state == 'select_a_from_x_':
-                    if token.ttype == T.Keyword and token.value.lower() == 'where':
+                elif state in ['select_a_from_x_', 'select_a_from_x_as_x_']:
+                    if token.ttype == T.Punctuation and token.value == ',':
+                        state = 'select_a_from_x,'
+                    elif token.ttype == T.Keyword and token.value.lower() == 'as':
+                        state = 'select_a_from_x_as'
+                    elif token.ttype == T.Keyword and token.value.lower() == 'where':
                         state = 'select_a_from_x_where'
                     elif token.ttype == T.Keyword and token.value.lower() == 'limit':
                         state = 'select_a_from_x_where_id=v_limit'
@@ -240,6 +253,20 @@ class SqlCompleter(Completer):
                         state = 'select_a_from_x_full'
                     elif token.ttype == T.Keyword and token.value.lower() == 'full outer join':
                         state = 'select_a_from_x_inner_join'
+                elif state == 'select_a_from_x_as':
+                    if token.ttype == T.Text.Whitespace:
+                        state = 'select_a_from_x_as_'
+                elif state == 'select_a_from_x_as_':
+                    if token.ttype == T.Name:
+                        state = 'select_a_from_x_as_x'
+                elif state == 'select_a_from_x_as_x':
+                    if token.ttype == T.Text.Whitespace:
+                        state = 'select_a_from_x_as_x_'
+                    elif token.ttype == T.Punctuation and token.value == ',':
+                        state = 'select_a_from_x_as_x,'
+                elif state == 'select_a_from_x_as_x,':
+                    if token.ttype == T.Name:
+                        state = 'select_a_from_x'
                 elif state == 'select_a_from_x_where':
                     if token.ttype == T.Text.Whitespace:
                         state = 'select_a_from_x_where_'

adam/sql/term_completer.py

@@ -46,15 +46,22 @@ class TermCompleter(WordCompleter):
             if self.ignore_case:
                 word = word.lower()
 
-            if word_before_cursor in ['(', ',', '=']:
-                return True
-
             if self.match_middle:
                 return word_before_cursor in word
             else:
                 return word.startswith(word_before_cursor)
 
         for a in words:
+            if word_before_cursor in ['(', ',', '=']:
+                display = self.display_dict.get(a, a)
+                display_meta = self.meta_dict.get(a, "")
+                yield Completion(
+                    a,
+                    0,
+                    display=display,
+                    display_meta=display_meta,
+                )
+
             if word_matches(a):
                 display = self.display_dict.get(a, a)
                 display_meta = self.meta_dict.get(a, "")

adam/version.py

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

{kaqing-2.0.55.dist-info → kaqing-2.0.56.dist-info}/METADATA

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

{kaqing-2.0.55.dist-info → kaqing-2.0.56.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=591d7gV6uQSFtm7IWdlIYAHjfAzs9bdvIkwlIAeKddE,7540
 adam/utils.py,sha256=2DoWsrcaioFFH0-RjT30qelVRPUJqCGTfz_ucfE7F8g,7406
-adam/version.py,sha256=1igvrn67AjTOcviySrpDWebyfe7wHN1EWQt56kFGM58,139
+adam/version.py,sha256=_cSpSCO9CTcCYlbRGHCj0bLXdSaXLCYsyqeNu1RWB_s,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,9 +164,8 @@ adam/k8s_utils/services.py,sha256=EOJJGACVbbRvu5T3rMKqIJqgYic1_MSJ17EA0TJ6UOk,31
 adam/k8s_utils/statefulsets.py,sha256=hiBOmJZ3KTI6_naAFzNoW1NoYnnBG35BZ7RMdPhNC6o,4664
 adam/k8s_utils/volumes.py,sha256=RIBmlOSWM3V3QVXLCFT0owVOyh4rGG1ETp521a-6ndo,1137
 adam/sql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adam/sql/sql_completer.py,sha256=rl7UlL5kthXOmcPBK3LFD1H0S2yDZMIQPbov1x8Wo9w,29819
-adam/sql/sql_utils.py,sha256=MLoxB9h22WZiANu0SZzmBJcY2gtb6f_D4jnoJEjLgPc,171
-adam/sql/term_completer.py,sha256=O-sUxFVc1DJmPQCdT-XPK2SSkUe4AL1dw8ZPXXicATY,2273
+adam/sql/sql_completer.py,sha256=VMMM29z6AnUCYIM6VThdggFkiz8elQEI96EM8FO6RLo,31170
+adam/sql/term_completer.py,sha256=VCCMPUzvh_9lwIOwqQmeijM0V2JPADkBbNV4HQIqDCw,2538
 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
@@ -177,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.55.dist-info/METADATA,sha256=3kKBRQ9yQ_KbdoGuD42fhkxDprEvWUEsRe5UO67FhvA,132
-kaqing-2.0.55.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-kaqing-2.0.55.dist-info/entry_points.txt,sha256=SkzhuQJUWsXOzHeZ5TgQ2c3_g53UGK23zzJU_JTZOZI,39
-kaqing-2.0.55.dist-info/top_level.txt,sha256=8_2PZkwBb-xDcnc8a2rAbQeJhXKXskc7zTP7pSPa1fw,5
-kaqing-2.0.55.dist-info/RECORD,,
+kaqing-2.0.56.dist-info/METADATA,sha256=50Y7sPo_HmrQyt5xK0N5IrJHVr-bikpprnZsuWjsLyo,132
+kaqing-2.0.56.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+kaqing-2.0.56.dist-info/entry_points.txt,sha256=SkzhuQJUWsXOzHeZ5TgQ2c3_g53UGK23zzJU_JTZOZI,39
+kaqing-2.0.56.dist-info/top_level.txt,sha256=8_2PZkwBb-xDcnc8a2rAbQeJhXKXskc7zTP7pSPa1fw,5
+kaqing-2.0.56.dist-info/RECORD,,

adam/sql/sql_utils.py

@@ -1,5 +0,0 @@
-import re
-
-def safe_terms(text: str):
-    tokens = re.findall(r'"[^"]+"|\b\w+\b|\S', text)
-    return tokens, len(tokens) > 1 or text.startswith(' ') or text.endswith(' ')