kaqing 2.0.98__py3-none-any.whl → 2.0.203__py3-none-any.whl

adam/commands/restart_cluster.py

@@ -0,0 +1,47 @@
+from adam.commands import extract_options
+from adam.commands.command import Command
+from adam.utils_k8s.pods import Pods
+from adam.utils_k8s.statefulsets import StatefulSets
+from adam.repl_state import ReplState, RequiredState
+from adam.utils import log2
+
+class RestartCluster(Command):
+    COMMAND = 'restart cluster'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(RestartCluster, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return RestartCluster.COMMAND
+
+    def required(self):
+        return RequiredState.CLUSTER
+
+    def run(self, cmd: str, state: ReplState):
+        if not(args := self.args(cmd)):
+            return super().run(cmd, state)
+
+        with self.validate(args, state) as (args, state):
+            with extract_options(args, '--force') as (args, forced):
+                if not forced:
+                    log2('Please add --force for restarting all nodes in a cluster.')
+
+                    return 'force-needed'
+
+                log2(f'Restarting all pods from {state.sts}...')
+                for pod_name in StatefulSets.pod_names(state.sts, state.namespace):
+                    Pods.delete(pod_name, state.namespace)
+
+                return state
+
+    def completion(self, state: ReplState):
+        return super().completion(state, {'--force': None})
+
+    def help(self, _: ReplState):
+        return f"{RestartCluster.COMMAND} --force\t restart all the nodes in the cluster"

adam/commands/restart_node.py

@@ -0,0 +1,51 @@
+from adam.commands import extract_options
+from adam.commands.command import Command
+from adam.commands.devices.devices import Devices
+from adam.utils_k8s.pods import Pods
+from adam.repl_state import ReplState, RequiredState
+from adam.utils import log2
+
+class RestartNode(Command):
+    COMMAND = 'restart node'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(RestartNode, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return RestartNode.COMMAND
+
+    def required(self):
+        return RequiredState.POD
+
+    def run(self, cmd: str, state: ReplState):
+        if not(args := self.args(cmd)):
+            return super().run(cmd, state)
+
+        with self.validate(args, state) as (args, state):
+            if not state.pod:
+                log2("'pod' is required")
+
+                return 'pod-needed'
+
+            with extract_options(args, '--force') as (args, forced):
+                if not forced:
+                    log2('Please add --force for restarting pod.')
+
+                    return 'force-needed'
+
+                log2(f'Restarting {state.pod}...')
+                Pods.delete(state.pod, state.namespace)
+
+                return state
+
+    def completion(self, state: ReplState):
+        return super().completion(state, {'--force': None}, pods=Devices.of(state).pods(state, '-'))
+
+    def help(self, _: ReplState):
+        return f"{RestartNode.COMMAND} --force\t restart the node"

adam/commands/restart_nodes.py

@@ -0,0 +1,47 @@
+from adam.commands import extract_options, validate_args
+from adam.commands.command import Command
+from adam.utils_k8s.pods import Pods
+from adam.utils_k8s.statefulsets import StatefulSets
+from adam.repl_state import ReplState, RequiredState
+from adam.utils import log2
+
+class RestartNodes(Command):
+    COMMAND = 'restart nodes'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(RestartNodes, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return RestartNodes.COMMAND
+
+    def required(self):
+        return RequiredState.CLUSTER
+
+    def run(self, cmd: str, state: ReplState):
+        if not(args := self.args(cmd)):
+            return super().run(cmd, state)
+
+        with self.validate(args, state, apply=False) as (args, state):
+            with extract_options(args, '--force') as (args, forced):
+                with validate_args(args, state, name='pod name'):
+                    if not forced:
+                        log2('Please add --force for restarting nodes.')
+
+                        return 'force-needed'
+
+                    for arg in args:
+                        Pods.delete(arg, state.namespace)
+
+                    return state
+
+    def completion(self, state: ReplState):
+        return super().completion(state, lambda: {p: {'--force': None} for p in StatefulSets.pod_names(state.sts, state.namespace)})
+
+    def help(self, _: ReplState):
+        return f"{RestartNodes.COMMAND} <pod-name>... --force\t restart Cassandra nodes"

adam/commands/rollout.py

@@ -2,6 +2,7 @@ import datetime
 from kubernetes import client
 from kubernetes.client.rest import ApiException
 
+from adam.commands import extract_options
 from adam.commands.command import Command
 from adam.commands.watch import Watch
 from adam.utils_k8s.statefulsets import StatefulSets
@@ -31,32 +32,28 @@ class RollOut(Command):
         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
+        with self.validate(args, state) as (args, state):
+            with extract_options(args, '--force') as (args, forced):
+                restarted, rollingout = StatefulSets.restarted_at(state.sts, state.namespace)
+                if rollingout and not forced:
+                    log2(f"* Cluster is being rolled out for {duration(restarted)}. Please wait until it's done or use --force.")
 
-        args, forced = Command.extract_options(args, '--force')
+                    return state
 
-        restarted, rollingout = StatefulSets.restarted_at(state.sts, state.namespace)
-        if rollingout and not forced:
-            log2(f"* Cluster is being rolled out for {duration(restarted)}. Please wait until it's done or use --force.")
+                self.rolling_restart(state.sts, state.namespace)
 
-            return state
+                auto_watch = False
+                if (auto_watch_cmds := Config().get('watch.auto', 'rollout')):
+                    cmds = [c.strip(' ') for c in auto_watch_cmds.split(',')]
+                    if self.command() in cmds:
+                        auto_watch = True
+                        log2('Rolling out cluster with auto watch...')
+                        Watch().run('watch', state)
 
-        self.rolling_restart(state.sts, state.namespace)
+                if not auto_watch:
+                    log2('Rolling out cluster...')
 
-        auto_watch = False
-        if (auto_watch_cmds := Config().get('watch.auto', 'rollout')):
-            cmds = [c.strip(' ') for c in auto_watch_cmds.split(',')]
-            if self.command() in cmds:
-                auto_watch = True
-                log2('Rolling out cluster with auto watch...')
-                Watch().run('watch', state)
-
-        if not auto_watch:
-            log2('Rolling out cluster...')
-
-        return state
+                return state
 
     def rolling_restart(self, statefulset, namespace):
         # kubectl rollout restart statefulset <statefulset-name>
@@ -82,9 +79,7 @@ class RollOut(Command):
             log2("Exception when calling AppsV1Api->read_namespaced_statefulset_status: %s\n" % e)
 
     def completion(self, state: ReplState):
-        if state.pod:
-            return {}
-        elif state.sts:
+        if super().completion(state):
             return {RollOut.COMMAND: None}
 
         return {}

adam/commands/shell.py

@@ -1,7 +1,9 @@
 import os
 
+from adam.commands import validate_args
 from adam.commands.command import Command
 from adam.repl_state import ReplState
+from adam.utils import log2
 
 class Shell(Command):
     COMMAND = ':sh'
@@ -18,13 +20,19 @@ class Shell(Command):
     def command(self):
         return Shell.COMMAND
 
-    def run(self, cmd: str, s0: ReplState):
+    def run(self, cmd: str, state: ReplState):
         if not(args := self.args(cmd)):
-            return super().run(cmd, s0)
+            return super().run(cmd, state)
 
-        _, args = self.apply_state(args, s0)
+        with self.validate(args, state) as (args, _):
+            with validate_args(args, state, at_least=0) as args_str:
+                if args_str:
+                    os.system(args_str)
+                    log2()
+                else:
+                    os.system('QING_DROPPED=true bash')
 
-        os.system('QING_DROPPED=true bash')
+            return state
 
     def completion(self, state: ReplState):
         return super().completion(state)

adam/commands/show/show.py

@@ -1,24 +1,23 @@
 import click
 
-from adam.commands.command import Command
+from adam.commands.app.show_app_actions import ShowAppActions
+from adam.commands.app.show_app_id import ShowAppId
+from adam.commands.app.show_app_queues import ShowAppQueues
+from adam.commands.intermediate_command import IntermediateCommand
 from adam.commands.medusa.medusa_show_backupjobs import MedusaShowBackupJobs
 from adam.commands.medusa.medusa_show_restorejobs import MedusaShowRestoreJobs
-from adam.commands.show.show_app_actions import ShowAppActions
-from adam.commands.show.show_app_queues import ShowAppQueues
 from adam.commands.show.show_host import ShowHost
 from adam.commands.show.show_login import ShowLogin
 from .show_params import ShowParams
-from .show_app_id import ShowAppId
 from .show_cassandra_status import ShowCassandraStatus
 from .show_cassandra_version import ShowCassandraVersion
-from .show_commands import ShowKubectlCommands
+from .show_cli_commands import ShowKubectlCommands
 from .show_processes import ShowProcesses
-from .show_repairs import ShowRepairs
+from .show_cassandra_repairs import ShowCassandraRepairs
 from .show_storage import ShowStorage
 from .show_adam import ShowAdam
-from adam.repl_state import ReplState
 
-class Show(Command):
+class Show(IntermediateCommand):
     COMMAND = 'show'
 
     # the singleton pattern
@@ -27,26 +26,14 @@ class Show(Command):
 
         return cls.instance
 
-    def __init__(self, successor: Command=None):
-        super().__init__(successor)
-
     def command(self):
         return Show.COMMAND
 
-    def run(self, cmd: str, state: ReplState):
-        if not(args := self.args(cmd)):
-            return super().run(cmd, state)
-
-        return super().intermediate_run(cmd, state, args, Show.cmd_list())
-
-    def cmd_list():
+    def cmd_list(self):
         return [ShowAppActions(), ShowAppId(), ShowAppQueues(), ShowHost(), ShowLogin(), ShowKubectlCommands(),
-                ShowParams(), ShowProcesses(), ShowRepairs(), ShowStorage(), ShowAdam(),
+                ShowParams(), ShowProcesses(), ShowCassandraRepairs(), ShowStorage(), ShowAdam(),
                 ShowCassandraStatus(), ShowCassandraVersion(), MedusaShowRestoreJobs(), MedusaShowBackupJobs()]
 
-    def completion(self, state: ReplState):
-        return super().completion(state)
-
 class ShowCommandHelper(click.Command):
     def get_help(self, ctx: click.Context):
-        Command.intermediate_help(super().get_help(ctx), Show.COMMAND, Show.cmd_list(), show_cluster_help=True)
+        IntermediateCommand.intermediate_help(super().get_help(ctx), Show.COMMAND, Show().cmd_list(), show_cluster_help=True)

adam/commands/show/show_adam.py

@@ -1,7 +1,7 @@
 import sys
 import os
 
-from adam.utils import lines_to_tabular, log2
+from adam.utils import tabulize, log2
 
 current_dir = os.path.dirname(os.path.abspath(__file__))
 
@@ -34,10 +34,10 @@ class ShowAdam(Command):
 
         package = os.path.dirname(os.path.abspath(__file__))
         package = package.split('/adam/')[0] + '/adam'
-        log2(lines_to_tabular([
+        tabulize([
             f'version\t{__version__}',
             f'source\t{package}'
-        ], separator='\t'))
+        ], separator='\t', to=2)
 
         return state
 

adam/commands/show/show_cassandra_repairs.py

@@ -0,0 +1,37 @@
+from adam.commands import extract_trailing_options
+from adam.commands.command import Command
+from adam.commands.cql.utils_cql import cassandra
+from adam.repl_state import ReplState, RequiredState
+
+class ShowCassandraRepairs(Command):
+    COMMAND = 'show cassandra repairs'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(ShowCassandraRepairs, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return ShowCassandraRepairs.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)
+
+        with self.validate(args, state) as (args, state):
+            with extract_trailing_options(args, '&') as (args, backgrounded):
+                with cassandra(state) as pods:
+                    return pods.nodetool('repair_admin list', backgrounded=backgrounded)
+
+    def completion(self, state: ReplState):
+        return super().completion(state, {'&': None})
+
+    def help(self, _: ReplState):
+        return f'{ShowCassandraRepairs.COMMAND}\t show Cassandra repairs'

adam/commands/show/show_cassandra_status.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 import sys
 
 from adam.checks.check_result import CheckResult
@@ -5,14 +6,15 @@ from adam.checks.check_utils import run_checks
 from adam.checks.compactionstats import CompactionStats
 from adam.checks.gossip import Gossip
 from adam.columns.columns import Columns
+from adam.commands import extract_options, extract_trailing_options
 from adam.commands.command import Command
-from adam.commands.issues import Issues
+from adam.commands.cql.utils_cql import cassandra
 from adam.config import Config
-from adam.utils_k8s.cassandra_nodes import CassandraNodes
-from adam.utils_k8s.secrets import Secrets
+from adam.repl_session import ReplSession
+from adam.utils_issues import IssuesUtils
 from adam.utils_k8s.statefulsets import StatefulSets
 from adam.repl_state import ReplState, RequiredState
-from adam.utils import lines_to_tabular, log, log2
+from adam.utils import SORT, tabulize, log2, log_exc
 from adam.checks.status import parse_nodetool_status
 
 class ShowCassandraStatus(Command):
@@ -37,51 +39,44 @@ class ShowCassandraStatus(Command):
         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
+        with self.validate(args, state) as (args, state):
+            with extract_trailing_options(args, '&') as (args, backgrounded):
+                with extract_options(args, ['-s', '--show']) as (args, show_out):
+                    if state.namespace and state.pod:
+                        self.show_single_pod(state, show_out=show_out, backgrounded=backgrounded)
+                    elif state.namespace and state.sts:
+                        self.merge(state, Config().get('nodetool.samples', sys.maxsize), show_output=show_out, backgrounded=backgrounded)
 
-        args, show_output = Command.extract_options(args, ['-s', '--show'])
+                    return state
 
-        if state.namespace and state.pod:
-            # self.show_table(state, [state.pod], state.namespace, show_output=show_output)
-            self.show_single_pod(state.sts, state.pod, state.namespace, show_output=show_output)
-        elif state.namespace and state.sts:
-            self.merge(state.sts, StatefulSets.pod_names(state.sts, state.namespace), state.namespace, Config().get('nodetool.samples', sys.maxsize), show_output=show_output)
-
-        return state
-
-    def show_single_pod(self, statefulset: str, pod_name: str, ns: str, show_output = False):
-        pod_name = pod_name.split('(')[0]
-        user, pw = Secrets.get_user_pass(pod_name, ns)
-        try:
-            result = CassandraNodes.exec(pod_name, ns, f"nodetool -u {user} -pw {pw} status", show_out=False)
-            status = parse_nodetool_status(result.stdout)
-            check_results = run_checks(cluster=statefulset, namespace=ns, checks=[CompactionStats(), Gossip()], show_output=show_output)
-            self.show_table(status, check_results)
-        except Exception as e:
-            log2(e)
+    def show_single_pod(self, state: ReplState, show_out = False, backgrounded = False):
+        with log_exc(True):
+            with cassandra(state) as pods:
+                result = pods.nodetool('status', show_out=False)
+                status = parse_nodetool_status(result.stdout)
+                check_results = run_checks(cluster=state.sts, namespace=state.namespace, checks=[CompactionStats(), Gossip()], show_out=show_out)
+                self.show_table(status, check_results, backgrounded=backgrounded)
 
-    def merge(self, statefulset: str, pod_names: list[str], ns: str, samples: int, show_output=False):
+    def merge(self, state: ReplState, samples: int, show_output=False, backgrounded = False):
         statuses: list[list[dict]] = []
+
+        pod_names = StatefulSets.pod_names(state.sts, state.namespace)
         for pod_name in pod_names:
             pod_name = pod_name.split('(')[0]
-            user, pw = Secrets.get_user_pass(pod_name, ns)
 
-            try:
-                result = CassandraNodes.exec(pod_name, ns, f"nodetool -u {user} -pw {pw} status", show_out=False)
-                status = parse_nodetool_status(result.stdout)
-                if status:
-                    statuses.append(status)
-                if samples <= len(statuses) and len(pod_names) != len(statuses):
-                    break
-            except Exception as e:
-                log2(e)
+            with log_exc(True):
+                with cassandra(state, pod=pod_name) as pods:
+                    result = pods.nodetool('status', show_out=False)
+                    status = parse_nodetool_status(result.stdout)
+                    if status:
+                        statuses.append(status)
+                    if samples <= len(statuses) and len(pod_names) != len(statuses):
+                        break
 
         combined_status = self.merge_status(statuses)
         log2(f'Showing merged status from {len(statuses)}/{len(pod_names)} nodes...')
-        check_results = run_checks(cluster=statefulset, namespace=ns, checks=[CompactionStats(), Gossip()], show_output=show_output)
-        self.show_table(combined_status, check_results)
+        check_results = run_checks(cluster=state.sts, namespace=state.namespace, checks=[CompactionStats(), Gossip()], show_out=show_output)
+        self.show_table(combined_status, check_results, backgrounded=backgrounded)
 
         return combined_status
 
@@ -102,27 +97,28 @@ class ShowCassandraStatus(Command):
 
         return combined
 
-    def show_table(self, status: list[dict[str, any]], check_results: list[CheckResult]):
+    def show_table(self, status: list[dict[str, any]], check_results: list[CheckResult], backgrounded=False):
         cols = Config().get('status.columns', 'status,address,load,tokens,owns,host_id,gossip,compactions')
         header = Config().get('status.header', '--,Address,Load,Tokens,Owns,Host ID,GOSSIP,COMPACTIONS')
         columns = Columns.create_columns(cols)
 
-        def line(status: dict):
-            cells = [c.host_value(check_results, status) for c in columns]
-            return ','.join(cells)
+        r = tabulize(status, lambda s: ','.join([c.host_value(check_results, s) for c in columns]), header=header, separator=',', sorted=SORT, to = 0 if backgrounded else 1)
 
-        lines = [line(d) for d in status]
-        lines.sort()
+        if backgrounded:
+            log_prefix = Config().get('log-prefix', '/tmp/qing')
+            log_file = f'{log_prefix}-{datetime.now().strftime("%d%H%M%S")}.log'
 
-        log(lines_to_tabular(lines, header, separator=','))
+            with open(log_file, 'w') as f:
+                f.write(r)
 
-        Issues.show(check_results)
+            ReplSession().append_history(f':sh cat {log_file}')
 
-    def completion(self, state: ReplState):
-        if state.sts:
-            return super().completion(state)
+            r = log_file
 
-        return {}
+        IssuesUtils.show(check_results)
+
+    def completion(self, state: ReplState):
+        return super().completion(state, {'-s': {'&': None}, '&': None})
 
     def help(self, _: ReplState):
         return f'{ShowCassandraStatus.COMMAND} [-s]\t show merged nodetool status  -s show commands on nodes'

adam/commands/show/show_cassandra_version.py

@@ -1,7 +1,5 @@
 from adam.commands.command import Command
-from adam.utils_k8s.cassandra_clusters import CassandraClusters
-from adam.utils_k8s.cassandra_nodes import CassandraNodes
-from adam.utils_k8s.secrets import Secrets
+from adam.commands.cql.utils_cql import cassandra
 from adam.repl_state import ReplState, RequiredState
 
 class ShowCassandraVersion(Command):
@@ -26,23 +24,12 @@ class ShowCassandraVersion(Command):
         if not(args := self.args(cmd)):
             return super().run(cmd, state)
 
-        state, _ = state.apply_args(args)
-        if not self.validate_state(state):
-            return state
-
-        user, pw = Secrets.get_user_pass(state.sts if state.sts else state.pod, state.namespace, secret_path='cql.secret')
-        command = f'cqlsh -u {user} -p {pw} -e "show version"'
-
-        if state.pod:
-            return CassandraNodes.exec(state.pod, state.namespace, command)
-        else:
-            return CassandraClusters.exec(state.sts, state.namespace, command, action='cql')
+        with self.validate(args, state) as (_, state):
+            with cassandra(state) as pods:
+                return pods.cql('show version', show_out=True, on_any=True)
 
     def completion(self, state: ReplState):
-        if state.sts:
-            return super().completion(state)
-
-        return {}
+        return super().completion(state)
 
     def help(self, _: ReplState):
         return f'{ShowCassandraVersion.COMMAND}\t show Cassandra version'

adam/commands/show/show_cli_commands.py

@@ -0,0 +1,56 @@
+from adam.commands.command import Command
+from adam.commands.cli_commands import CliCommands
+from adam.repl_state import ReplState, RequiredState
+from adam.utils import tabulize, log
+
+class ShowKubectlCommands(Command):
+    COMMAND = 'show cli-commands'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(ShowKubectlCommands, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return ShowKubectlCommands.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)
+
+        with self.validate(args, state) as (args, state):
+            v = CliCommands.values(state, collapse=True)
+            # node-exec-?, nodetool-?, cql-?, reaper-exec, reaper-forward, reaper-ui, reaper-username, reaper-password
+            cmds = [
+                f'bash,{v["node-exec-?"]}',
+                f'nodetool,{v["nodetool-?"]}',
+                f'cql,{v["cql-?"]}',
+            ]
+
+            if 'reaper-exec' in v:
+                cmds += [
+                    f'reaper,{v["reaper-exec"]}',
+                    f',{v["reaper-forward"]}  * should be run from your laptop',
+                    f',{v["reaper-ui"]}',
+                    f',{v["reaper-username"]}',
+                    f',{v["reaper-password"]}',
+                ]
+
+            cmds += [f'{k},{v0}' for k, v0 in v.items() if k.startswith('pg-')]
+
+            tabulize(cmds, separator=',')
+
+            return cmds
+
+    def completion(self, state: ReplState):
+        return super().completion(state)
+
+    def help(self, _: ReplState):
+        return f"{ShowKubectlCommands.COMMAND}\t show kubectl commands"

adam/commands/show/show_host.py

@@ -19,7 +19,7 @@ class ShowHost(Command):
         return ShowHost.COMMAND
 
     def run(self, cmd: str, state: ReplState):
-        if not(args := self.args(cmd)):
+        if not self.args(cmd):
             return super().run(cmd, state)
 
         log(get_my_host())