kaqing 1.77.0__py3-none-any.whl → 2.0.171__py3-none-any.whl

walker/commands/user_entry.py

@@ -1,69 +0,0 @@
-import os
-import signal
-import traceback
-
-from walker.sso.idp import Idp
-from walker.app_session import AppSession, IdpLogin
-from walker.apps import Apps
-from walker.commands.command import Command
-from walker.repl_state import ReplState
-from walker.utils import log2
-
-class UserEntry(Command):
-    COMMAND = 'entry'
-
-    # the singleton pattern
-    def __new__(cls, *args, **kwargs):
-        if not hasattr(cls, 'instance'): cls.instance = super(UserEntry, cls).__new__(cls)
-
-        return cls.instance
-
-    def __init__(self, successor: Command=None):
-        super().__init__(successor)
-
-    def command(self):
-        return UserEntry.COMMAND
-
-    def run(self, cmd: str, state: ReplState):
-        def custom_handler(signum, frame):
-            AppSession.ctrl_c_entered = True
-
-        signal.signal(signal.SIGINT, custom_handler)
-
-        if not(args := self.args(cmd)):
-            return super().run(cmd, state)
-
-        state, args = self.apply_state(args, state)
-
-        username: str = None
-        if len(args) > 0:
-            username = args[0]
-
-        login: IdpLogin = None
-        while not login:
-            try:
-                if not(host := Apps.app_host('c3', 'c3', state.namespace)):
-                    log2('Cannot locate ingress for app.')
-                    username = None
-                    continue
-
-                if not (login := Idp.login(host, username=username, use_cached=False)):
-                    log2('Invalid username/password. Please try again.')
-                    username = None
-            except:
-                # log2(traceback.format_exc())
-                pass
-
-        sh = f'{os.getcwd()}/login.sh'
-        if not os.path.exists(sh):
-            sh = f'{os.getcwd()}/docker/login.sh'
-
-        os.system(f'{sh} {login.shell_user()} {login.ser()}')
-
-        return state
-
-    def completion(self, _: ReplState):
-        return {}
-
-    def help(self, _: ReplState):
-        return f'{UserEntry.COMMAND}\t ttyd user entry'

walker/commands/watch.py

@@ -1,85 +0,0 @@
-import threading
-import time
-from kubernetes import client
-from typing import List
-
-from walker.commands.command import Command
-from walker.commands.commands_utils import show_pods, show_rollout
-from walker.config import Config
-from walker.k8s_utils.statefulsets import StatefulSets
-from walker.repl_state import ReplState, RequiredState
-from walker.utils import convert_seconds, log2
-
-class Watch(Command):
-    COMMAND = 'watch'
-
-    # the singleton pattern
-    def __new__(cls, *args, **kwargs):
-        if not hasattr(cls, 'instance'): cls.instance = super(Watch, cls).__new__(cls)
-
-        return cls.instance
-
-    def __init__(self, successor: Command=None):
-        super().__init__(successor)
-
-    def command(self):
-        return Watch.COMMAND
-
-    def required(self):
-        return RequiredState.CLUSTER_OR_POD
-
-    def run(self, cmd: str, state: ReplState):
-        if not(args := self.args(cmd)):
-            return super().run(cmd, state)
-
-        state, args = self.apply_state(args, state)
-        if not self.validate_state(state):
-            return state
-
-        pods = StatefulSets.pods(state.sts, state.namespace)
-        if not pods:
-            log2("No pods are found.")
-            return state
-
-        stop_event = threading.Event()
-        thread = threading.Thread(target=self.loop, args=(stop_event, state.sts, pods, state.namespace), daemon=True)
-        thread.start()
-
-        try:
-            log2(f"Press Ctrl+C to break.")
-
-            time.sleep(Config().get('watch.timeout', 3600 * 1))
-        except KeyboardInterrupt:
-            pass
-
-        log2("Stopping watch...")
-        stop_event.set()
-        thread.join()
-
-        return state
-
-    def loop(self, stop_flag: threading.Event, sts: str, pods: List[client.V1Pod], ns: str):
-        show_pods(pods, ns)
-        show_rollout(sts, ns)
-
-        cnt = Config().get('watch.interval', 10)
-        while not stop_flag.is_set():
-            time.sleep(1)
-            cnt -= 1
-
-            if not cnt:
-                show_pods(pods, ns)
-                show_rollout(sts, ns)
-                cnt = Config().get('watch.interval', 10)
-
-    def completion(self, state: ReplState):
-        if state.pod:
-            return {}
-
-        if not state.sts:
-            return {Watch.COMMAND: {n: None for n in StatefulSets.list_sts_names()}}
-
-        return {Watch.COMMAND: None}
-
-    def help(self, _: ReplState):
-        return f'{Watch.COMMAND}\t watch pod changes'

walker/embedded_params.py

@@ -1,2 +0,0 @@
-def params():
-    return {'app': {'console-endpoint': 'https://{host}/{env}/{app}/static/console/index.html', 'cr': {'cluster-regex': '(.*?-.*?)-.*', 'group': 'ops.c3.ai', 'v': 'v2', 'plural': 'c3cassandras'}, 'label': 'c3__app_id-0', 'login': {'admin-group': '{host}/C3.ClusterAdmin', 'ingress': '{app_id}-k8singr-appleader-001', 'timeout': 5, 'session-check-url': 'https://{host}/{env}/{app}/api/8/C3/userSessionToken', 'cache-creds': True, 'url': 'https://{host}/{env}/{app}'}, 'strip': '0'}, 'bash': {'workers': 32}, 'cassandra': {'service-name': 'all-pods-service'}, 'cql': {'workers': 32, 'samples': 3, 'secret': {'cluster-regex': '(.*?-.*?)-.*', 'name': '{cluster}-superuser', 'password-item': 'password'}}, 'checks': {'compactions-threshold': 250, 'cpu-busy-threshold': 98.0, 'cpu-threshold': 0.0, 'cassandra-data-path': '/c3/cassandra', 'root-disk-threshold': 50, 'cassandra-disk-threshold': 50, 'snapshot-size-cmd': "ls /c3/cassandra/data/data/*/*/snapshots | grep snapshots | sed 's/:$//g' | xargs -I {} du -sk {} | awk '{print $1}' | awk '{s+=$1} END {print s}'", 'snapshot-size-threshold': '40G', 'table-sizes-cmd': "ls -Al /c3/cassandra/data/data/ | awk '{print $9}' | sed 's/\\^r//g' | xargs -I {} du -sk /c3/cassandra/data/data/{}"}, 'get-host-id': {'workers': 32}, 'idps': {'ad': {'email-pattern': '.*@c3.ai', 'uri': 'https://login.microsoftonline.com/53ad779a-93e7-485c-ba20-ac8290d7252b/oauth2/v2.0/authorize?response_type=id_token&response_mode=form_post&client_id=00ff94a8-6b0a-4715-98e0-95490012d818&scope=openid+email+profile&redirect_uri=https%3A%2F%2Fplat.c3ci.cloud%2Fc3%2Fc3%2Foidc%2Flogin&nonce={nonce}&state=EMPTY'}, 'okta': {'email-pattern': '.*@c3iot.com', 'uri': 'https://c3energy.okta.com/oauth2/v1/authorize?response_type=id_token&response_mode=form_post&client_id={client_id}&scope=openid+email+profile+groups&redirect_uri=https%3A%2F%2F{host}%2Fc3%2Fc3%2Foidc%2Flogin&nonce={nonce}&state=EMPTY'}}, 'issues': {'workers': 32}, 'logs': {'path': '/c3/cassandra/logs/system.log'}, 'medusa': {'restore-auto-complete': False}, 'nodetool': {'workers': 32, 'samples': 3, 'commands_in_line': 40}, 'pg': {'name-pattern': '^{namespace}.*-k8spg-.*', 'excludes': '.helm., -admin-secret', 'agent': {'image': 'seanahnsf/kaqing', 'name': 'kaqing-agent', 'timeout': 86400}, 'default-db': 'postgres', 'default-schema': 'postgres', 'secret': {'endpoint-key': 'postgres-db-endpoint', 'port-key': 'postgres-db-port', 'username-key': 'postgres-admin-username', 'password-key': 'postgres-admin-password'}}, 'preview': {'rows': 10}, 'processes': {'columns': 'pod,cpu,mem', 'header': 'POD_NAME,CPU,MEM/LIMIT'}, 'reaper': {'service-name': 'reaper-service', 'port-forward': {'timeout': 86400, 'local-port': 9001}, 'abort-runs-batch': 10, 'show-runs-batch': 100, 'pod': {'cluster-regex': '(.*?-.*?-.*?-.*?)-.*', 'label-selector': 'k8ssandra.io/reaper={cluster}-reaper'}, 'secret': {'cluster-regex': '(.*?-.*?)-.*', 'name': '{cluster}-reaper-ui', 'password-item': 'password'}}, 'repair': {'log-path': '/home/cassrepair/logs/', 'image': 'ci-registry.c3iot.io/cloudops/cassrepair:2.0.13', 'secret': 'ciregistryc3iotio', 'env': {'interval': 24, 'timeout': 60, 'pr': False, 'runs': 1}}, 'repl': {'start-drive': 'a', 'auto-enter-app': 'c3/c3', 'auto-enter-only-cluster': True}, 'status': {'columns': 'status,address,load,tokens,owns,host_id,gossip,compactions', 'header': '--,Address,Load,Tokens,Owns,Host ID,GOSSIP,COMPACTIONS'}, 'storage': {'columns': 'pod,volume_root,volume_cassandra,snapshots,data,compactions', 'header': 'POD_NAME,VOLUME /,VOLUME CASS,SNAPSHOTS,DATA,COMPACTIONS'}, 'watch': {'auto': 'rollout', 'timeout': 3600, 'interval': 10}, 'debug': {'timings': False, 'exit-on-error': False, 'show-parallelism': False, 'show-out': False}}

walker/k8s_utils/cassandra_clusters.py

@@ -1,48 +0,0 @@
-from collections.abc import Callable
-from concurrent.futures import ThreadPoolExecutor
-import sys
-from typing import TypeVar
-
-from walker.k8s_utils.cassandra_nodes import CassandraNodes
-from walker.pod_exec_result import PodExecResult
-from walker.utils import log2
-from .statefulsets import StatefulSets
-from .pods import Pods
-from .kube_context import KubeContext
-
-T = TypeVar('T')
-
-# utility collection on cassandra clusters; methods are all static
-class CassandraClusters:
-    def exec(statefulset: str, namespace: str, command: str, action: str = 'action', max_workers=0, show_out=True) -> list[PodExecResult]:
-        def body(executor: ThreadPoolExecutor, pod: str, namespace: str, show_out: bool):
-            if executor:
-                return executor.submit(CassandraNodes.exec, pod, namespace, command, False, False,)
-
-            return CassandraNodes.exec(pod, namespace, command, show_out=show_out)
-
-        def post(result, show_out: bool):
-            if KubeContext.show_out(show_out):
-                print(result.command)
-                if result.stdout:
-                    print(result.stdout)
-                if result.stderr:
-                    log2(result.stderr, file=sys.stderr)
-
-            return result
-
-        return StatefulSets.on_cluster(statefulset, namespace, body, post=post, action=action, max_workers=max_workers, show_out=show_out)
-
-    def on_cluster(statefulset: str,
-                namespace: str,
-                body: Callable[[ThreadPoolExecutor, str, str, bool], T],
-                post: Callable[[T], T] = None,
-                action: str = 'action', max_workers=0, show_out=True) -> list[T]:
-        pods = StatefulSets.pod_names(statefulset, namespace)
-
-        return Pods.on_pods(pods, namespace, body, post=post, action=action, max_workers=max_workers, show_out=show_out)
-
-    def pod_names_by_host_id(ss: str, ns: str):
-        pods = StatefulSets.pods(ss, ns)
-
-        return {CassandraNodes.get_host_id(pod.metadata.name, ns): pod.metadata.name for pod in pods}

walker/k8s_utils/cassandra_nodes.py

@@ -1,26 +0,0 @@
-from walker.config import Config
-from walker.k8s_utils.pods import Pods
-from walker.k8s_utils.secrets import Secrets
-from walker.pod_exec_result import PodExecResult
-
-# utility collection on cassandra nodes; methods are all static
-class CassandraNodes:
-    def exec(pod_name: str, namespace: str, command: str, show_out = True, throw_err = False) -> PodExecResult:
-        return Pods.exec(pod_name, "cassandra", namespace, command, show_out, throw_err)
-
-    def get_host_id(pod_name: str, ns: str):
-        try:
-            user, pw = Secrets.get_user_pass(pod_name, ns)
-            command = f'echo "SELECT host_id FROM system.local; exit" | cqlsh --no-color -u {user} -p {pw}'
-            result: PodExecResult = CassandraNodes.exec(pod_name, ns, command, show_out=Config().get('debug.trace', False))
-            next = False
-            for line in result.stdout.splitlines():
-                if next:
-                    return line.strip(' ')
-                if line.startswith('----------'):
-                    next = True
-                    continue
-        except Exception as e:
-            return str(e)
-
-        return 'Unknown'

walker/k8s_utils/pods.py

@@ -1,211 +0,0 @@
-from collections.abc import Callable
-from concurrent.futures import ThreadPoolExecutor, as_completed
-import sys
-import time
-from typing import TypeVar, cast
-from kubernetes import client
-from kubernetes.stream import stream
-from kubernetes.stream.ws_client import ERROR_CHANNEL
-
-from walker.config import Config
-from walker.pod_exec_result import PodExecResult
-from walker.utils import elapsed_time, log2
-from .kube_context import KubeContext
-
-T = TypeVar('T')
-_TEST_POD_EXEC_OUTS: PodExecResult = None
-
-# utility collection on pods; methods are all static
-class Pods:
-    def set_test_pod_exec_outs(outs: PodExecResult):
-        global _TEST_POD_EXEC_OUTS
-        _TEST_POD_EXEC_OUTS = outs
-
-        return _TEST_POD_EXEC_OUTS
-
-    def delete(pod_name: str, namespace: str):
-        try:
-            v1 = client.CoreV1Api()
-            api_response = v1.delete_namespaced_pod(pod_name, namespace)
-        except Exception as e:
-            log2("Exception when calling CoreV1Api->delete_namespaced_pod: %s\n" % e)
-
-    def on_pods(pods: list[str],
-                namespace: str,
-                body: Callable[[ThreadPoolExecutor, str, str, bool], T],
-                post: Callable[[T], T] = None,
-                action: str = 'action', max_workers=0, show_out=True) -> list[T]:
-        show_out = KubeContext.show_out(show_out)
-
-        if not max_workers:
-            max_workers = Config().action_workers(action, 0)
-        if max_workers > 0:
-            # if parallel, node sampling is suppressed
-            if KubeContext.show_parallelism():
-                log2(f'Executing on all nodes from statefulset in parallel...')
-            start_time = time.time()
-            try:
-                with ThreadPoolExecutor(max_workers=max_workers) as executor:
-                    # disable stdout from the pod_exec, then show the output in a for loop
-                    futures = [body(executor, pod, namespace, show_out) for pod in pods]
-                    if len(futures) == 0:
-                        return cast(list[T], [])
-
-                rs = [future.result() for future in as_completed(futures)]
-                if post:
-                    rs = [post(r, show_out=show_out) for r in rs]
-
-                return rs
-            finally:
-                if KubeContext.show_parallelism():
-                    log2(f"Parallel {action} elapsed time: {elapsed_time(start_time)} with {max_workers} workers")
-        else:
-            results: list[T] = []
-
-            samples = Config().action_node_samples(action, sys.maxsize)
-            l = min(len(pods), samples)
-            adj = 'all'
-            if l < len(pods):
-                adj = f'{l} sample'
-            if show_out:
-                log2(f'Executing on {adj} nodes from statefulset...')
-            for pod_name in pods:
-                try:
-                    result = body(None, pod_name, namespace, show_out)
-                    if post:
-                        result = post(result, show_out=show_out)
-                    results.append(result)
-                    if result:
-                        l -= 1
-                        if not l:
-                            break
-                except Exception as e:
-                    log2(e)
-
-            return results
-
-    def exec(pod_name: str, container: str, namespace: str, command: str, show_out = True, throw_err = False, interaction: Callable[[any, list[str]], any] = None):
-        if _TEST_POD_EXEC_OUTS:
-            return _TEST_POD_EXEC_OUTS
-
-        show_out = KubeContext.show_out(show_out)
-
-        api = client.CoreV1Api()
-
-        exec_command = ["/bin/sh", "-c", command]
-        k_command = f'kubectl exec {pod_name} -c {container} -n {namespace} -- {command}'
-        if show_out:
-            print(k_command)
-
-        resp = stream(
-            api.connect_get_namespaced_pod_exec,
-            pod_name,
-            namespace,
-            command=exec_command,
-            container=container,
-            stderr=True,
-            stdin=True,
-            stdout=True,
-            tty=True,
-            _preload_content=False,
-        )
-
-        stdout = []
-        stderr = []
-        error_output = None
-        try:
-            while resp.is_open():
-                resp.update(timeout=1)
-                if resp.peek_stdout():
-                    frag = resp.read_stdout()
-                    stdout.append(frag)
-                    if show_out: print(frag, end="")
-
-                    if interaction:
-                        interaction(resp, stdout)
-                if resp.peek_stderr():
-                    frag = resp.read_stderr()
-                    stderr.append(frag)
-                    if show_out: print(frag, end="")
-
-            try:
-                # get the exit code from server
-                error_output = resp.read_channel(ERROR_CHANNEL)
-            except Exception:
-                pass
-        except Exception as e:
-            if throw_err:
-                raise e
-            else:
-                log2(e)
-        finally:
-            resp.close()
-
-        return PodExecResult("".join(stdout), "".join(stderr), k_command, error_output)
-
-    def get_container(namespace: str, pod_name: str, container_name: str):
-        pod = Pods.get(namespace, pod_name)
-        if not pod:
-            return None
-
-        for container in pod.spec.containers:
-            if container_name == container.name:
-                return container
-
-        return None
-
-    def get(namespace: str, pod_name: str):
-        v1 = client.CoreV1Api()
-        return v1.read_namespaced_pod(name=pod_name, namespace=namespace)
-
-    def create_pod_spec(name: str, image: str, image_pull_secret: str, envs: list, volume_name: str, pvc_name:str, mount_path:str, command: list[str]=None, sa_name=None):
-        volume_mounts = []
-        if volume_name and pvc_name and mount_path:
-            volume_mounts=[client.V1VolumeMount(mount_path=mount_path, name=volume_name)]
-
-        container = client.V1Container(name=name, image=image, env=envs, command=command,
-                                    volume_mounts=volume_mounts)
-
-        volumes = []
-        if volume_name and pvc_name and mount_path:
-            volumes=[client.V1Volume(name=volume_name, persistent_volume_claim=client.V1PersistentVolumeClaimVolumeSource(claim_name=pvc_name))]
-
-        security_context = None
-        if not sa_name:
-            security_context=client.V1PodSecurityContext(run_as_user=1001, run_as_group=1001, fs_group=1001)
-
-        return client.V1PodSpec(
-            restart_policy="Never",
-            containers=[container],
-            image_pull_secrets=[client.V1LocalObjectReference(name=image_pull_secret)],
-            security_context=security_context,
-            service_account_name=sa_name,
-            volumes=volumes
-        )
-
-    def create(namespace: str, pod_name: str, image: str, command: list[str],
-               secret: str = None,
-               env: dict[str, any] = {},
-               volume_name: str = None,
-               pvc_name: str = None,
-               mount_path: str = None,
-               sa_name=None):
-        v1 = client.CoreV1Api()
-        envs = []
-        for k, v in env.items():
-            envs.append(client.V1EnvVar(name=str(k), value=str(v)))
-        pod = Pods.create_pod_spec(pod_name, image, secret, envs, volume_name, pvc_name, mount_path, command=command, sa_name=sa_name)
-        return v1.create_namespaced_pod(namespace=namespace,
-                                        body=client.V1Pod(spec=pod, metadata=client.V1ObjectMeta(name=pod_name)))
-
-    def wait_for_running(namespace: str, pod_name: str):
-        msged = False
-
-        while Pods.get(namespace, pod_name).status.phase != 'Running':
-            if not msged:
-                log2(f'Waiting for the {pod_name} pod to start up...')
-                msged = True
-            time.sleep(5)
-
-    def completed(namespace: str, pod_name: str):
-        return Pods.get(namespace, pod_name).status.phase in ['Succeeded', 'Failed']

walker/repl.py

@@ -1,165 +0,0 @@
-import re
-import time
-import traceback
-import click
-from prompt_toolkit.completion import NestedCompleter
-from prompt_toolkit.key_binding import KeyBindings
-
-from walker.cli_group import cli
-from walker.commands.command import Command
-from walker.commands.command_helpers import ClusterCommandHelper
-from walker.commands.help import Help
-from walker.commands.postgres.postgres_session import PostgresSession
-from walker.config import Config
-from walker.k8s_utils.kube_context import KubeContext
-from walker.k8s_utils.statefulsets import StatefulSets
-from walker.repl_commands import ReplCommands
-from walker.repl_session import ReplSession
-from walker.repl_state import ReplState
-from walker.utils import deep_merge_dicts, lines_to_tabular, log2
-from walker.apps import Apps
-from . import __version__
-
-def enter_repl(state: ReplState):
-    cmd_list = ReplCommands.repl_cmd_list() + [Help()]
-    # head with the Chain of Responsibility pattern
-    cmds: Command = Command.chain(cmd_list)
-    session = ReplSession().prompt_session
-
-    def prompt_msg():
-        msg = ''
-        if state.device == ReplState.P:
-            msg = f'{ReplState.P}:'
-            pg = PostgresSession(state.namespace, state.pg_path) if state.pg_path else None
-            if pg and pg.db:
-                msg += pg.db
-            elif pg and pg.host:
-                msg += pg.host
-        elif state.device == ReplState.A:
-            msg = f'{ReplState.A}:'
-            if state.app_env:
-                msg += state.app_env
-            if state.app_app:
-                msg += f'/{state.app_app}'
-        else:
-            msg = f'{ReplState.C}:'
-            if state.pod:
-                # cs-d0767a536f-cs-d0767a536f-default-sts-0
-                group = re.match(r".*?-.*?-(.*)", state.pod)
-                msg += group[1]
-            elif state.sts:
-                # cs-d0767a536f-cs-d0767a536f-default-sts
-                group = re.match(r".*?-.*?-(.*)", state.sts)
-                msg += group[1]
-
-        return f"{msg}$ " if state.bash_session else f"{msg}> "
-
-    log2(f'kaqing {__version__}')
-    ss = StatefulSets.list_sts_name_and_ns()
-
-    if state.device == ReplState.C:
-        if not ss:
-            raise Exception("no Cassandra clusters found")
-        elif len(ss) == 1 and Config().get('repl.auto-enter-only-cluster', True):
-            cluster = ss[0]
-            state.sts = cluster[0]
-            state.namespace = cluster[1]
-            state.wait_log(f'Moving to the only Cassandra cluster: {state.sts}@{state.namespace}...')
-    elif state.device == ReplState.A:
-        if app := Config().get('repl.auto-enter-app', 'c3/c3'):
-            if app != 'no':
-                ea = app.split('/')
-                state.app_env = ea[0]
-                if len(ea) > 1:
-                    state.app_app = ea[1]
-                    state.wait_log(f'Moving to {state.app_env}/{state.app_app}...')
-                else:
-                    state.wait_log(f'Moving to {state.app_env}...')
-
-    kb = KeyBindings()
-
-    @kb.add('c-c')
-    def _(event):
-        event.app.current_buffer.text = ''
-
-    while True:
-        try:
-            completer = NestedCompleter.from_nested_dict({})
-            if not state.bash_session:
-                completions = {}
-                if state.app_app:
-                    completions = Apps(path='apps.yaml').commands()
-                    # completions = {k: None for k in Apps(path='apps.yaml').commands()}
-
-                for cmd in cmd_list:
-                    s1 = time.time()
-                    try:
-                        completions = deep_merge_dicts(completions, cmd.completion(state))
-                    finally:
-                        if Config().get('debug.timings', False):
-                            print('Timing completion calc', cmd.command(), f'{time.time() - s1:.2f}')
-                        pass
-
-                completer = NestedCompleter.from_nested_dict(completions)
-
-            cmd = session.prompt(prompt_msg(), completer=completer, key_bindings=kb)
-            s0 = time.time()
-
-            if state.bash_session:
-                if cmd.strip(' ') == 'exit':
-                    state.exit_bash()
-                    continue
-
-                cmd = f'bash {cmd}'
-
-            if cmd and cmd.strip(' ') and not cmds.run(cmd, state):
-                c_sql_tried = False
-                if state.device == ReplState.P:
-                    pg = PostgresSession(state.namespace, state.pg_path)
-                    if pg.db:
-                        c_sql_tried = True
-                        cmd = f'pg {cmd}'
-                        cmds.run(cmd, state)
-                elif state.device == ReplState.A:
-                    if state.app_app:
-                        c_sql_tried = True
-                        cmd = f'app {cmd}'
-                        cmds.run(cmd, state)
-                elif state.sts:
-                    c_sql_tried = True
-                    cmd = f'cql {cmd}'
-                    cmds.run(cmd, state)
-
-                if not c_sql_tried:
-                    log2(f'* Invalid command: {cmd}')
-                    log2()
-                    lines = [c.help(state) for c in cmd_list if c.help(state)]
-                    log2(lines_to_tabular(lines, separator='\t'))
-        except EOFError:  # Handle Ctrl+D (EOF) for graceful exit
-            break
-        except Exception as e:
-            if Config().get('debug.exit-on-error', False):
-                raise e
-            else:
-                # log2(e)
-                traceback.print_exc()
-        finally:
-            state.clear_wait_log_flag()
-            if Config().get('debug.timings', False) and 'cmd' in locals() and 's0' in locals():
-                print('Timing command', cmd, f'{time.time() - s0:.2f}')
-
-@cli.command(context_settings=dict(ignore_unknown_options=True, allow_extra_args=True), cls=ClusterCommandHelper, help="Enter interactive shell.")
-@click.option('--kubeconfig', '-k', required=False, metavar='path', help='path to kubeconfig file')
-@click.option('--config', default='params.yaml', metavar='path', help='path to kaqing parameters file')
-@click.option('--param', '-v', multiple=True, metavar='<key>=<value>', help='parameter override')
-@click.option('--cluster', '-c', required=False, metavar='statefulset', help='Kubernetes statefulset name')
-@click.option('--namespace', '-n', required=False, metavar='namespace', help='Kubernetes namespace')
-@click.argument('extra_args', nargs=-1, metavar='[cluster]', type=click.UNPROCESSED)
-def repl(kubeconfig: str, config: str, param: list[str], cluster:str, namespace: str, extra_args):
-    KubeContext.init_config(kubeconfig)
-    if not KubeContext.init_params(config, param):
-        return
-
-    state = ReplState(device=Config().get('repl.start-drive', 'a'), ns_sts=cluster, namespace=namespace, in_repl=True)
-    state, _ = state.apply_args(extra_args)
-    enter_repl(state)

walker/repl_commands.py

@@ -1,58 +0,0 @@
-from walker.commands.app import App
-from walker.commands.app_ping import AppPing
-from walker.commands.frontend.code_start import CodeStart
-from walker.commands.frontend.code_stop import CodeStop
-from walker.commands.show.show_app_queues import ShowAppQueues
-from walker.commands.cp import ClipboardCopy
-from walker.commands.bash import Bash
-from walker.commands.cd import Cd
-from walker.commands.check import Check
-from walker.commands.command import Command
-from walker.commands.cqlsh import Cqlsh
-from walker.commands.devices import DeviceApp, DeviceCass, DevicePostgres
-from walker.commands.exit import Exit
-from walker.commands.medusa.medusa import Medusa
-from walker.commands.param_get import GetParam
-from walker.commands.issues import Issues
-from walker.commands.ls import Ls
-from walker.commands.nodetool import NodeTool
-from walker.commands.postgres.postgres import Postgres
-from walker.commands.preview_table import PreviewTable
-from walker.commands.processes import Processes
-from walker.commands.pwd import Pwd
-from walker.commands.reaper.reaper import Reaper
-from walker.commands.repair.repair import Repair
-from walker.commands.report import Report
-from walker.commands.restart import Restart
-from walker.commands.rollout import RollOut
-from walker.commands.param_set import SetParam
-from walker.commands.show.show import Show
-from walker.commands.show.show_app_actions import ShowAppActions
-from walker.commands.show.show_app_id import ShowAppId
-from walker.commands.status import Status
-from walker.commands.storage import Storage
-from walker.commands.watch import Watch
-
-class ReplCommands:
-    def repl_cmd_list() -> list[Command]:
-        return ReplCommands.app() + [CodeStart(), CodeStop()] + [DeviceApp(), DevicePostgres(), DeviceCass()] + ReplCommands.navigation() + \
-               ReplCommands.cassandra_check() + ReplCommands.cassandra_ops() + ReplCommands.tools() + ReplCommands.exit()
-
-    def navigation() -> list[Command]:
-        return [Ls(), PreviewTable(), Cd(), Pwd(), ClipboardCopy(), GetParam(), SetParam()] + Show.cmd_list()
-
-    def cassandra_check() -> list[Command]:
-        return [Check(), Issues(), NodeTool(), Processes(), Report(), Status(), Storage()]
-
-    def cassandra_ops() -> list[Command]:
-        return Medusa.cmd_list() + [Restart(), RollOut(), Watch()] + Reaper.cmd_list() + Repair.cmd_list()
-
-    def tools() -> list[Command]:
-        return [Cqlsh(), Postgres(), Bash()]
-
-    def app() -> list[Command]:
-        return [ShowAppActions(), ShowAppId(), ShowAppQueues(), AppPing(), App(), ]
-        # return [C3Echo(), C3IqCount(), C3()]
-
-    def exit() -> list[Command]:
-        return [Exit()]