kaqing 2.0.48__tar.gz → 2.0.49__tar.gz

{kaqing-2.0.48 → kaqing-2.0.49}/PKG-INFO

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

{kaqing-2.0.48 → kaqing-2.0.49}/adam/batch.py

@@ -5,7 +5,7 @@ from adam.commands.check import Check, CheckCommandHelper
 from adam.commands.cp import ClipboardCopy, CopyCommandHelper
 from adam.commands.command import Command
 from adam.commands.command_helpers import ClusterCommandHelper, ClusterOrPodCommandHelper, PodCommandHelper
-from adam.commands.cqlsh import CqlCommandHelper, Cqlsh
+from adam.commands.cql.cqlsh import CqlCommandHelper, Cqlsh
 from adam.commands.deploy.deploy import Deploy, DeployCommandHelper
 from adam.commands.deploy.undeploy import Undeploy, UndeployCommandHelper
 from adam.commands.describe.describe import Describe, DescribeCommandHelper

{kaqing-2.0.48 → kaqing-2.0.49}/adam/commands/alter_tables.py

@@ -1,5 +1,5 @@
 from adam.commands.command import Command
-from adam.commands.cql_utils import tables as get_tables, run_cql
+from adam.commands.cql.cql_utils import tables as get_tables, run_cql
 from adam.config import Config
 from adam.pod_exec_result import PodExecResult
 from adam.repl_state import ReplState, RequiredState

kaqing-2.0.49/adam/commands/cql/cql_completions.py

@@ -0,0 +1,12 @@
+from adam.commands.cql.cql_table_completer import CqlTableCompleter
+from adam.commands.cql.cql_utils import table_names
+from adam.repl_state import ReplState
+
+def cql_completions(state: ReplState) -> dict[str, any]:
+    completer = CqlTableCompleter(table_names(state))
+    return {
+        'describe': {
+            'keyspaces': None,
+            'table': completer.completions(),
+            'tables': None},
+    } | completer.completions()

kaqing-2.0.49/adam/commands/cql/cql_table_completer.py

@@ -0,0 +1,16 @@
+from adam.table_completer import NestedDict, TableCompleter
+
+class CqlTableCompleter(TableCompleter):
+    def __init__(self, tables: list[str], nested_dict: NestedDict = {}, ignore_case: bool = True):
+        self._tables = tables
+        self.ignore_case = ignore_case
+        self.append_nested_dict(nested_dict)
+
+    def __repr__(self) -> str:
+        return "CqlTableCompleter(%r)" % (len(self._tables))
+
+    def nested(self, data: NestedDict) -> 'TableCompleter':
+        return CqlTableCompleter(self._tables).append_nested_dict(data)
+
+    def tables(self) -> list[str]:
+        return self._tables

{kaqing-2.0.48/adam/commands → kaqing-2.0.49/adam/commands/cql}/cqlsh.py

@@ -2,7 +2,8 @@ import click
 
 from adam.commands.command import Command
 from adam.commands.command_helpers import ClusterOrPodCommandHelper
-from adam.commands.cql_utils import run_cql, table_names, tables
+from adam.commands.cql.cql_completions import cql_completions
+from .cql_utils import run_cql, table_names
 from adam.repl_state import ReplState, RequiredState
 from adam.utils import log, log2
 
@@ -53,21 +54,11 @@ class Cqlsh(Command):
         return run_cql(state, cql, opts, show_out=True)
 
     def completion(self, state: ReplState) -> dict[str, any]:
+        if state.device != state.C:
+            return {}
+
         if state.sts or state.pod:
-            ts = table_names(state)
-            return {Cqlsh.COMMAND: None} | {
-                'delete': {'from': {t: {'where': {'id': {'=': None}}} for t in ts}},
-                'insert': {'into': {t: {'values': None} for t in ts}},
-                'select': {'*': {'from': {t: {
-                    'limit': {'1': None},
-                    'where': {'id': {'=': None}}
-                } for t in ts}}},
-                'update': {t: {'set': None} for t in ts},
-                'describe': {
-                    'keyspaces': None,
-                    'table': {t: None for t in ts},
-                    'tables': None},
-            }
+            return {Cqlsh.COMMAND: None} | cql_completions(state)
 
         return {}
 

{kaqing-2.0.48 → kaqing-2.0.49}/adam/commands/describe/describe_keyspace.py

@@ -1,5 +1,5 @@
 from adam.commands.command import Command
-from adam.commands.cql_utils import keyspaces, run_cql
+from adam.commands.cql.cql_utils import keyspaces, run_cql
 from adam.pod_exec_result import PodExecResult
 from adam.repl_state import ReplState, RequiredState
 from adam.utils import log2

{kaqing-2.0.48 → kaqing-2.0.49}/adam/commands/describe/describe_keyspaces.py

@@ -1,5 +1,5 @@
 from adam.commands.command import Command
-from adam.commands.cql_utils import run_cql
+from adam.commands.cql.cql_utils import run_cql
 from adam.pod_exec_result import PodExecResult
 from adam.repl_state import ReplState, RequiredState
 from adam.utils import log2

{kaqing-2.0.48 → kaqing-2.0.49}/adam/commands/describe/describe_table.py

@@ -1,5 +1,5 @@
 from adam.commands.command import Command
-from adam.commands.cql_utils import run_cql, table_names
+from adam.commands.cql.cql_utils import run_cql, table_names
 from adam.pod_exec_result import PodExecResult
 from adam.repl_state import ReplState, RequiredState
 from adam.utils import log2

{kaqing-2.0.48 → kaqing-2.0.49}/adam/commands/describe/describe_tables.py

@@ -1,5 +1,5 @@
 from adam.commands.command import Command
-from adam.commands.cql_utils import run_cql
+from adam.commands.cql.cql_utils import run_cql
 from adam.pod_exec_result import PodExecResult
 from adam.repl_state import ReplState, RequiredState
 from adam.utils import log2

{kaqing-2.0.48 → kaqing-2.0.49}/adam/commands/ls.py

@@ -2,7 +2,7 @@ import copy
 
 from adam.commands.command import Command
 from adam.commands.commands_utils import show_pods, show_rollout
-from adam.commands.cqlsh import Cqlsh
+from adam.commands.cql.cqlsh import Cqlsh
 from adam.commands.postgres.postgres_utils import pg_database_names, pg_table_names
 from adam.commands.postgres.postgres_session import PostgresSession
 from adam.config import Config

kaqing-2.0.49/adam/commands/postgres/pg_completions.py

@@ -0,0 +1,9 @@
+from adam.commands.postgres.pg_table_completer import PgTableCompleter
+
+def pg_completions(ns: str, pg_path: str):
+    return {
+        '\h': None,
+        '\d': None,
+        '\dt': None,
+        '\du': None
+    } | PgTableCompleter(ns, pg_path).completions()

kaqing-2.0.49/adam/commands/postgres/pg_table_completer.py

@@ -0,0 +1,18 @@
+from adam.commands.postgres.postgres_utils import pg_table_names
+from adam.table_completer import NestedDict, TableCompleter
+
+class PgTableCompleter(TableCompleter):
+    def __init__(self, namespace: str, pg_path: str, nested_dict: NestedDict = {}, ignore_case: bool = True):
+        self.namespace = namespace
+        self.pg_path = pg_path
+        self.ignore_case = ignore_case
+        self.append_nested_dict(nested_dict)
+
+    def __repr__(self) -> str:
+        return "PgTableCompleter(%r, pg_path=%r)" % (self.namespace, self.pg_path)
+
+    def nested(self, data: NestedDict) -> 'TableCompleter':
+        return PgTableCompleter(self.namespace, self.pg_path).append_nested_dict(data)
+
+    def tables(self) -> list[str]:
+        return pg_table_names(self.namespace, self.pg_path)

{kaqing-2.0.48 → kaqing-2.0.49}/adam/commands/postgres/postgres.py

@@ -65,6 +65,9 @@ class Postgres(Command):
         PostgresSession(state.namespace, state.pg_path).run_sql(' '.join(args))
 
     def completion(self, state: ReplState):
+        if state.device != state.P:
+            return {}
+
         leaf = {}
         session = PostgresSession(state.namespace, state.pg_path)
         if session.db:

{kaqing-2.0.48 → kaqing-2.0.49}/adam/commands/preview_table.py

@@ -1,7 +1,8 @@
 import functools
 
 from adam.commands.command import Command
-from adam.commands.cql_utils import run_cql, tables
+from adam.commands.cql.cql_table_completer import CqlTableCompleter
+from adam.commands.cql.cql_utils import run_cql, table_names, tables
 from adam.commands.postgres.pg_table_completer import PgTableCompleter
 from adam.commands.postgres.postgres_session import PostgresSession
 from adam.config import Config
@@ -72,11 +73,9 @@ class PreviewTable(Command):
 
     def completion(self, state: ReplState):
         if state.device == ReplState.P:
-            # return {PreviewTable.COMMAND: {f'<pg_tables:{state.namespace}:{state.pg_path}>': None}}
             return {PreviewTable.COMMAND: PgTableCompleter(state.namespace, state.pg_path)}
         elif state.sts:
-            tables = PreviewTable.cql_tables(state)
-            return {PreviewTable.COMMAND: {f'{k}.{t}': None for k, ts in tables.items() for t in ts}}
+            return {PreviewTable.COMMAND: CqlTableCompleter(table_names(state))}
 
         return {}
 

{kaqing-2.0.48 → kaqing-2.0.49}/adam/repl_commands.py

@@ -23,7 +23,7 @@ from adam.commands.bash import Bash
 from adam.commands.cd import Cd
 from adam.commands.check import Check
 from adam.commands.command import Command
-from adam.commands.cqlsh import Cqlsh
+from adam.commands.cql.cqlsh import Cqlsh
 from adam.commands.devices import DeviceApp, DeviceCass, DevicePostgres
 from adam.commands.exit import Exit
 from adam.commands.medusa.medusa import Medusa

kaqing-2.0.48/adam/commands/postgres/pg_table_completer.py → kaqing-2.0.49/adam/table_completer.py

@@ -1,25 +1,16 @@
+from abc import abstractmethod
 from typing import Any, Dict, Iterable, Mapping, Optional, Set, Union
 from prompt_toolkit.completion import CompleteEvent, Completer, Completion, NestedCompleter, WordCompleter
 from prompt_toolkit.document import Document
 
-from adam.commands.postgres.postgres_utils import pg_table_names
-
 NestedDict = Mapping[str, Union[Any, Set[str], None, Completer]]
 
-class PgTableCompleter(Completer):
-    def __init__(self, namespace: str, pg_path: str, nested_dict: NestedDict = {}, ignore_case: bool = True):
-        self.namespace = namespace
-        self.pg_path = pg_path
+class TableCompleter(Completer):
+    def __init__(self, nested_dict: NestedDict = {}, ignore_case: bool = True):
         self.ignore_case = ignore_case
         self.append_nested_dict(nested_dict)
 
-    def __repr__(self) -> str:
-        return "PgTableCompleter(%r, pg_path=%r)" % (self.namespace, self.pg_path)
-
-    def nested(self, data: NestedDict):
-        return PgTableCompleter(self.namespace, self.pg_path).append_nested_dict(data)
-
-    def append_nested_dict(self, data: NestedDict) -> "PgTableCompleter":
+    def append_nested_dict(self, data: NestedDict) -> "TableCompleter":
         options: Dict[str, Optional[Completer]] = {}
         for key, value in data.items():
             if isinstance(value, Completer):
@@ -77,7 +68,26 @@ class PgTableCompleter(Completer):
                     yield c
         else:
             completer = WordCompleter(
-                list(pg_table_names(self.namespace, self.pg_path)), ignore_case=self.ignore_case
+                self.tables(), ignore_case=self.ignore_case
             )
             for c in completer.get_completions(document, complete_event):
-                yield c
+                yield c
+
+    @abstractmethod
+    def nested(self, data: NestedDict) -> 'TableCompleter':
+        pass
+
+    @abstractmethod
+    def tables(self) -> list[str]:
+        pass
+
+    def completions(self):
+        return {
+            'delete': {'from': self.nested({'where': {'id': {'=': {"'id'": None}}}})},
+            'insert': {'into': self.nested({'values(': None})},
+            'select': {'*': {'from': self.nested({
+                'limit': {'1': {'where': {'id': {'=': {"'id'": None}}}}},
+                'where': {'id': {'=': {"'id'": {'limit': {'1': None}}}}}
+            })}},
+            'update': self.nested({'set': {'column': {'=': None}}}),
+        }

kaqing-2.0.49/adam/version.py

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

{kaqing-2.0.48 → kaqing-2.0.49}/kaqing.egg-info/PKG-INFO

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

{kaqing-2.0.48 → kaqing-2.0.49}/kaqing.egg-info/SOURCES.txt

@@ -15,6 +15,7 @@ adam/repl.py
 adam/repl_commands.py
 adam/repl_session.py
 adam/repl_state.py
+adam/table_completer.py
 adam/utils.py
 adam/version.py
 adam/checks/__init__.py
@@ -60,8 +61,6 @@ adam/commands/command.py
 adam/commands/command_helpers.py
 adam/commands/commands_utils.py
 adam/commands/cp.py
-adam/commands/cql_utils.py
-adam/commands/cqlsh.py
 adam/commands/devices.py
 adam/commands/exit.py
 adam/commands/help.py
@@ -80,6 +79,11 @@ adam/commands/restart.py
 adam/commands/rollout.py
 adam/commands/shell.py
 adam/commands/watch.py
+adam/commands/cql/__init__.py
+adam/commands/cql/cql_completions.py
+adam/commands/cql/cql_table_completer.py
+adam/commands/cql/cql_utils.py
+adam/commands/cql/cqlsh.py
 adam/commands/deploy/__init__.py
 adam/commands/deploy/code_start.py
 adam/commands/deploy/code_stop.py

{kaqing-2.0.48 → kaqing-2.0.49}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name='kaqing',
-    version='2.0.48',
+    version='2.0.49',
     packages=find_packages(),
     entry_points={
         'console_scripts': [

kaqing-2.0.48/adam/commands/postgres/pg_completions.py

@@ -1,17 +0,0 @@
-from adam.commands.postgres.pg_table_completer import PgTableCompleter
-
-def pg_completions(ns: str, pg_path: str):
-    table = PgTableCompleter(ns, pg_path)
-    return {
-        '\h': None,
-        '\d': None,
-        '\dt': None,
-        '\du': None,
-        'delete': {'from': table.nested({'where': {'id': {'=': {"'id'": None}}}})},
-        'insert': {'into': table.nested({'values(': None})},
-        'select': {'*': {'from': table.nested({
-            'limit': {'1': {'where': {'id': {'=': {"'id'": None}}}}},
-            'where': {'id': {'=': {"'id'": {'limit': {'1': None}}}}}
-        })}},
-        'update': table.nested({'set': {'column': {'=': None}}}),
-    }

kaqing-2.0.48/adam/version.py

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