kaqing 2.0.184__py3-none-any.whl → 2.0.227__py3-none-any.whl

adam/app_session.py

@@ -8,7 +8,7 @@ from adam.log import Log
 from adam.sso.idp import Idp
 from adam.sso.idp_login import IdpLogin
 from adam.config import Config
-from adam.utils import debug, debug_trace, json_to_csv, tabulize, log, log2, log_exc
+from adam.utils import debug, debug_trace, json_to_csv, tabulize, log2, log_exc
 from adam.apps import Apps
 
 class AppLogin:

adam/batch.py

@@ -2,28 +2,28 @@ import click
 
 from adam.commands.audit.audit import Audit, AuditCommandHelper
 from adam.commands.bash.bash import Bash
-from adam.commands.check import Check, CheckCommandHelper
-from adam.commands.clipboard_copy import ClipboardCopy, CopyCommandHelper
+from adam.commands.cassandra.restart_nodes import RestartNodes
+from adam.commands.cassandra.rollout import RollOut
+from adam.commands.cassandra.watch import Watch
+from adam.commands.cli.clipboard_copy import ClipboardCopy, CopyCommandHelper
 from adam.commands.command import Command
 from adam.commands.command_helpers import ClusterCommandHelper, ClusterOrPodCommandHelper, PodCommandHelper
 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.issues import Issues
-from adam.commands.login import Login
-from adam.commands.logs import Logs
-from adam.commands.ls import Ls
+from adam.commands.app.login import Login
+from adam.commands.cassandra.download_cassandra_log import DownloadCassandraLog
+from adam.commands.diag.check import Check, CheckCommandHelper
+from adam.commands.diag.generate_report import GenerateReport
+from adam.commands.diag.issues import Issues
+from adam.commands.fs.ls import Ls
 from adam.commands.medusa.medusa import Medusa
-from adam.commands.nodetool import NodeTool, NodeToolCommandHelper
+from adam.commands.nodetool.nodetool import NodeTool, NodeToolCommandHelper
 from adam.commands.postgres.postgres import Postgres, PostgresCommandHelper
 from adam.commands.preview_table import PreviewTable
 from adam.commands.reaper.reaper import Reaper, ReaperCommandHelper
 from adam.commands.repair.repair import Repair, RepairCommandHelper
-from adam.commands.report import Report
-from adam.commands.restart import Restart
-from adam.commands.rollout import RollOut
-from adam.commands.show.show import Show, ShowCommandHelper
-from adam.commands.watch import Watch
+from adam.commands.show import Show, ShowCommandHelper
 from adam.utils_k8s.kube_context import KubeContext
 from adam.repl import enter_repl
 from adam.repl_state import ReplState
@@ -128,7 +128,7 @@ def login(kubeconfig: str, config: str, param: list[str], namespace: str, pod: s
 @click.option('--pod', '-p', required=False, metavar='pod', help='Kubernetes pod name')
 @click.argument('extra_args', nargs=-1, metavar='<pod>', type=click.UNPROCESSED)
 def logs(kubeconfig: str, config: str, param: list[str], namespace: str, pod: str, extra_args):
-    run_command(Logs(), kubeconfig, config, param, None, namespace, pod, extra_args)
+    run_command(DownloadCassandraLog(), kubeconfig, config, param, None, namespace, pod, extra_args)
 
 
 @cli.command(context_settings=dict(ignore_unknown_options=True, allow_extra_args=True), cls=ClusterCommandHelper, help='List statefulset or pods.')
@@ -221,7 +221,7 @@ def repair(kubeconfig: str, config: str, param: list[str], cluster: str, namespa
 @click.option('--show', '-s', is_flag=True, help='show output from Cassandra nodes')
 @click.argument('extra_args', nargs=-1, metavar='[cluster|pod]', type=click.UNPROCESSED)
 def report(kubeconfig: str, config: str, param: list[str], cluster: str, namespace: str, pod: str, show: bool, extra_args):
-    run_command(Report(), kubeconfig, config, param, cluster, namespace, pod, ('-s',) + extra_args if show else extra_args)
+    run_command(GenerateReport(), kubeconfig, config, param, cluster, namespace, pod, ('-s',) + extra_args if show else extra_args)
 
 
 @cli.command(context_settings=dict(ignore_unknown_options=True, allow_extra_args=True), cls=ClusterOrPodCommandHelper, help='Restart Cassandra Cluster or Node')
@@ -234,7 +234,7 @@ def report(kubeconfig: str, config: str, param: list[str], cluster: str, namespa
 @click.option('--force', is_flag=True, help='need for restarting the whole cluster')
 @click.argument('extra_args', nargs=-1, metavar='<cluster|pod>', type=click.UNPROCESSED)
 def restart(kubeconfig: str, config: str, param: list[str], cluster: str, namespace: str, pod: str, force: bool, extra_args):
-    run_command(Restart(), kubeconfig, config, param, cluster, namespace, pod, ('--force',) + extra_args if force else extra_args)
+    run_command(RestartNodes(), kubeconfig, config, param, cluster, namespace, pod, ('--force',) + extra_args if force else extra_args)
 
 
 @cli.command(context_settings=dict(ignore_unknown_options=True, allow_extra_args=True), cls=ClusterOrPodCommandHelper, help='Rolling restart Cassandra Cluster.')

adam/checks/compactionstats.py

@@ -5,6 +5,7 @@ from adam.checks.check_context import CheckContext
 from adam.checks.check_result import CheckResult
 from adam.checks.issue import Issue
 from adam.config import Config
+from adam.utils import Color
 from adam.utils_k8s.cassandra_nodes import CassandraNodes
 
 class CompactionStats(Check):
@@ -15,7 +16,7 @@ class CompactionStats(Check):
         issues: list[Issue] = []
 
         try:
-            result = CassandraNodes.exec(ctx.pod, ctx.namespace, f"nodetool -u {ctx.user} -pw {ctx.pw} compactionstats", show_out=ctx.show_output)
+            result = CassandraNodes.exec(ctx.pod, ctx.namespace, f"nodetool -u {ctx.user} -pw {ctx.pw} compactionstats", show_out=ctx.show_output, text_color=Color.gray)
             compactions = parse_nodetool_compactionstats(result.stdout)
             pod_details = {
                 'name': ctx.pod,

adam/checks/cpu.py

@@ -5,6 +5,7 @@ from adam.checks.check_context import CheckContext
 from adam.checks.check_result import CheckResult
 from adam.checks.issue import Issue
 from adam.config import Config
+from adam.utils import Color
 from adam.utils_k8s.cassandra_nodes import CassandraNodes
 from adam.utils_k8s.custom_resources import CustomResources
 from adam.utils_k8s.pods import Pods
@@ -31,7 +32,7 @@ class Cpu(Check):
                 details['limit'] = container.resources.limits["cpu"]
 
             idle = 'Unknown'
-            result = CassandraNodes.exec(ctx.pod, ctx.namespace, "mpstat 5 2 | grep Average | awk '{print $NF}'", show_out=ctx.show_output)
+            result = CassandraNodes.exec(ctx.pod, ctx.namespace, "mpstat 5 2 | grep Average | awk '{print $NF}'", show_out=ctx.show_output, text_color=Color.gray)
             lines = result.stdout.strip(' \r\n').split('\n')
             line = lines[len(lines) - 1].strip(' \r')
             idle = details['idle'] = line

adam/checks/disk.py

@@ -1,12 +1,13 @@
 import re
 from pathlib import Path
+import traceback
 
 from adam.checks.check import Check
 from adam.checks.check_context import CheckContext
 from adam.checks.check_result import CheckResult
 from adam.checks.issue import Issue
 from adam.config import Config
-from adam.utils import log_exc
+from adam.utils import Color, log_exc
 from adam.utils_k8s.cassandra_nodes import CassandraNodes
 
 class Disk(Check):
@@ -19,16 +20,16 @@ class Disk(Check):
 
         try:
             cass_data_path = Config().get('checks.cassandra-data-path', '/c3/cassandra')
-            df_result = CassandraNodes.exec(ctx.pod, ctx.namespace, f"df -h | grep -e '{cass_data_path}' -e 'overlay'", show_out=ctx.show_output)
+            df_result = CassandraNodes.exec(ctx.pod, ctx.namespace, f"df -h | grep -e '{cass_data_path}' -e 'overlay'", show_out=ctx.show_output, text_color=Color.gray)
 
             snapshot_size = Config().get('checks.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}'")
-            ss_result = CassandraNodes.exec(ctx.pod, ctx.namespace, snapshot_size, show_out=ctx.show_output)
+            ss_result = CassandraNodes.exec(ctx.pod, ctx.namespace, snapshot_size, show_out=ctx.show_output, text_color=Color.gray)
 
             data_sizes = Config().get('checks.data-size-cmd', "du -sh /c3/cassandra/data/data")
-            ds_result = CassandraNodes.exec(ctx.pod, ctx.namespace, data_sizes, show_out=ctx.show_output)
+            ds_result = CassandraNodes.exec(ctx.pod, ctx.namespace, data_sizes, show_out=ctx.show_output, text_color=Color.gray)
 
             table_sizes = Config().get('checks.table-sizes-cmd', "ls -Al /c3/cassandra/data/data/ | awk '{print $9}' | sed 's/\^r//g' | xargs -I {} du -sk /c3/cassandra/data/data/{}")
-            ts_result = CassandraNodes.exec(ctx.pod, ctx.namespace, table_sizes, show_out=ctx.show_output)
+            ts_result = CassandraNodes.exec(ctx.pod, ctx.namespace, table_sizes, show_out=ctx.show_output, text_color=Color.gray)
 
             result = self.build_details(ctx, df_result.stdout, ss_result.stdout, ds_result.stdout, ts_result.stdout)
 

adam/checks/gossip.py

@@ -4,6 +4,7 @@ from adam.checks.check import Check
 from adam.checks.check_context import CheckContext
 from adam.checks.check_result import CheckResult
 from adam.checks.issue import Issue
+from adam.utils import Color
 from adam.utils_k8s.cassandra_nodes import CassandraNodes
 
 class Gossip(Check):
@@ -32,7 +33,7 @@ class Gossip(Check):
         #   TOKENS:21:<hidden>
         # STATUS line missing -> NORMAL
         # user, pw = get_user_pass(pod_name, ns)
-        result = CassandraNodes.exec(ctx.pod, ctx.namespace, f"nodetool -u {ctx.user} -pw {ctx.pw} gossipinfo", show_out=ctx.show_output)
+        result = CassandraNodes.exec(ctx.pod, ctx.namespace, f"nodetool -u {ctx.user} -pw {ctx.pw} gossipinfo", show_out=ctx.show_output, text_color=Color.gray)
 
         nodes = self.parse_gossipinfo(result.stdout)
         details = {

adam/checks/memory.py

@@ -2,6 +2,7 @@ from adam.checks.check import Check
 from adam.checks.check_context import CheckContext
 from adam.checks.check_result import CheckResult
 from adam.checks.issue import Issue
+from adam.utils import Color
 from adam.utils_k8s.cassandra_nodes import CassandraNodes
 from adam.utils_k8s.custom_resources import CustomResources
 from adam.utils_k8s.pods import Pods
@@ -47,7 +48,7 @@ class Memory(Check):
 
     def find_error(self, ctx: CheckContext, pattern: str, issue_desc: str):
         escaped = pattern.replace('"', '\"')
-        result = CassandraNodes.exec(ctx.pod, ctx.namespace, f'tac /c3/cassandra/logs/system.log | grep "{escaped}" | head -1', show_out=ctx.show_output)
+        result = CassandraNodes.exec(ctx.pod, ctx.namespace, f'tac /c3/cassandra/logs/system.log | grep "{escaped}" | head -1', show_out=ctx.show_output, text_color=Color.gray)
         if result.stdout.find(pattern) > 0:
             return Issue(
                 statefulset=ctx.statefulset,

adam/checks/status.py

@@ -4,6 +4,7 @@ from adam.checks.check import Check
 from adam.checks.check_context import CheckContext
 from adam.checks.check_result import CheckResult
 from adam.checks.issue import Issue
+from adam.utils import Color
 from adam.utils_k8s.cassandra_nodes import CassandraNodes
 
 class Status(Check):
@@ -14,7 +15,7 @@ class Status(Check):
         issues: list[Issue] = []
 
         try:
-            result = CassandraNodes.exec(ctx.pod, ctx.namespace, f"nodetool -u {ctx.user} -pw {ctx.pw} status", show_out=ctx.show_output)
+            result = CassandraNodes.exec(ctx.pod, ctx.namespace, f"nodetool -u {ctx.user} -pw {ctx.pw} status", show_out=ctx.show_output, text_color=Color.gray)
             status = parse_nodetool_status(result.stdout)
             pod_details = {
                 'name': ctx.pod,

adam/commands/app/app.py

@@ -31,8 +31,8 @@ class App(Command):
 
                 return state
 
-    def completion(self, state: ReplState):
-        return super().completion(state, {'--force': None})
+    def completion(self, _: ReplState):
+        return {}
 
-    def help(self, _: ReplState):
-        return f"<AppType>.<AppAction> <args> [--force]\t post app action; check with 'show app actions' command"
+    def help(self, state: ReplState):
+        return super().help(state, "post app action  use 'show app actions' for registered actions  --force refresh session", command='<AppType>.<AppAction>', args='<args> [--force]')

adam/commands/app/app_ping.py

@@ -34,5 +34,5 @@ class AppPing(Command):
     def completion(self, state: ReplState):
         return super().completion(state, {'--force': None})
 
-    def help(self, _: ReplState):
-        return f"{AppPing.COMMAND} [--force]\t ping app server with Echo.echoStatic()"
+    def help(self, state: ReplState):
+        return super().help(state, 'ping app server with Echo.echoStatic() --force refresh session', args='[--force]')

adam/commands/{login.py → app/login.py}

@@ -63,5 +63,5 @@ class Login(Command):
     def completion(self, state: ReplState):
         return super().completion(state)
 
-    def help(self, _: ReplState):
-        return f'{Login.COMMAND}\t SSO login'
+    def help(self, state: ReplState):
+        return super().help(state, 'SSO login')

adam/commands/app/show_app_actions.py

@@ -1,5 +1,5 @@
 from adam.app_session import AppSession
-from adam.apps import AppAction, Apps
+from adam.apps import Apps
 from adam.commands.command import Command
 from adam.config import Config
 from adam.repl_state import ReplState
@@ -45,5 +45,5 @@ class ShowAppActions(Command):
     def completion(self, state: ReplState):
         return super().completion(state)
 
-    def help(self, _: ReplState):
-        return f"{ShowAppActions.COMMAND}\t show app actions"
+    def help(self, state: ReplState):
+        return super().help(state, 'show registered app actions')

adam/commands/app/show_app_id.py

@@ -40,5 +40,5 @@ class ShowAppId(Command):
     def completion(self, state: ReplState):
         return super().completion(state)
 
-    def help(self, _: ReplState):
-        return f'{ShowAppId.COMMAND}\t show app id for the Cassandra cluster'
+    def help(self, state: ReplState):
+        return super().help(state, 'show app id for the Cassandra cluster')

adam/commands/app/show_app_queues.py

@@ -34,5 +34,5 @@ class ShowAppQueues(Command):
     def completion(self, state: ReplState):
         return super().completion(state, {'--force': None})
 
-    def help(self, _: ReplState):
-        return f"{ShowAppQueues.COMMAND} [--force]\t show invalidation queue counts"
+    def help(self, state: ReplState):
+        return super().help(state, 'show invalidation queue counts  --force refresh session', args='[--force]')

adam/commands/{show → app}/show_login.py

@@ -5,7 +5,7 @@ from adam.sso.idp import Idp
 from adam.sso.idp_login import IdpLogin
 from adam.commands.command import Command
 from adam.repl_state import ReplState
-from adam.utils import duration, tabulize, log, log2, log_exc
+from adam.utils import duration, tabulize, log2, log_exc
 
 class ShowLogin(Command):
     COMMAND = 'show login'
@@ -52,5 +52,5 @@ class ShowLogin(Command):
     def completion(self, state: ReplState):
         return super().completion(state)
 
-    def help(self, _: ReplState):
-        return f'{ShowLogin.COMMAND}\t show login details'
+    def help(self, state: ReplState):
+        return super().help(state, 'show SSO login details')

adam/commands/app/utils_app.py

@@ -3,12 +3,20 @@ from typing import Union
 
 from adam.app_session import AppSession
 from adam.apps import Apps
-from adam.pod_exec_result import PodExecResult
+from adam.utils_k8s.pod_exec_result import PodExecResult
 from adam.repl_state import ReplState
 from adam.utils import log2
 from adam.utils_k8s.app_clusters import AppClusters
 from adam.utils_k8s.app_pods import AppPods
 
+def app_cd_dirs(state: ReplState):
+    if state.app_app:
+        return ['..'] + AppPods.pod_names(state.namespace, state.app_env, state.app_app)
+    elif state.app_env:
+        return ['..'] + [app[0].split('-')[1] for app in Apps.apps(state.app_env)]
+    else:
+        return [env[0] for env in Apps.envs()]
+
 class AppRestHandler:
     def __init__(self, state: ReplState, forced = False):
         self.state = state

adam/commands/audit/audit.py

@@ -1,17 +1,16 @@
 import click
 
 from adam.commands import validate_args
+from adam.commands.audit.completions_l import completions_l
 from adam.commands.audit.audit_repair_tables import AuditRepairTables
 from adam.commands.audit.audit_run import AuditRun
 from adam.commands.audit.show_last10 import ShowLast10
 from adam.commands.audit.show_slow10 import ShowSlow10
 from adam.commands.audit.show_top10 import ShowTop10
-from adam.commands.audit.utils_show_top10 import show_top10_completions_for_nesting
 from adam.commands.command import Command
 from adam.commands.intermediate_command import IntermediateCommand
 from adam.repl_state import ReplState
-from adam.sql.sql_completer import SqlCompleter, SqlVariant
-from adam.utils import log2, wait_log
+from adam.utils import log2
 from adam.utils_athena import Athena
 
 class Audit(IntermediateCommand):
@@ -50,31 +49,16 @@ class Audit(IntermediateCommand):
             return state
 
     def completion(self, state: ReplState):
-        if state.device == ReplState.L:
-            if not self.schema_read:
-                wait_log(f'Inspecting audit database schema...')
-                self.schema_read = True
-                # warm up the caches first time when l: drive is accessed
-                Athena.table_names()
-                Athena.column_names()
-                Athena.column_names(partition_cols_only=True)
-
-            return super().completion(state) | show_top10_completions_for_nesting() | SqlCompleter(
-                lambda: Athena.table_names(),
-                expandables={
-                    'columns': lambda table: Athena.column_names(),
-                    'partition_columns': lambda table: Athena.column_names(partition_cols_only=True)
-                },
-                variant=SqlVariant.ATHENA
-            ).completions_for_nesting()
-
-        return {}
+        if state.device != ReplState.L:
+            return {}
+
+        return completions_l()
 
     def cmd_list(self):
         return [AuditRepairTables(), AuditRun(), ShowLast10(), ShowSlow10(), ShowTop10()]
 
-    def help(self, _: ReplState):
-        return f'[{Audit.COMMAND}] [<sql-statements>]\t run SQL queries on Authena audit database'
+    def help(self, state: ReplState):
+        return super().help(state, 'run SQL queries on Athena audit database', command='[audit] [<sql-statements>]')
 
 class AuditCommandHelper(click.Command):
     def get_help(self, ctx: click.Context):

adam/commands/audit/audit_repair_tables.py

@@ -46,7 +46,7 @@ class AuditRepairTables(Command):
                     with Audits.offload() as exec:
                         exec.submit(lambda: self.auto_repair(hours))
 
-            return super().completion(state)
+            # return super().completion(state)
 
         return {}
 
@@ -68,5 +68,5 @@ class AuditRepairTables(Command):
                 exec.submit(Athena.query, f'MSCK REPAIR TABLE {table}', None,)
             exec.submit(Audits.put_meta, Audits.PARTITIONS_ADDED, meta,)
 
-    def help(self, _: ReplState):
-        return f"{AuditRepairTables.COMMAND}\t run MSCK REPAIR command for new partition discovery"
+    def help(self, state: ReplState):
+        return super().help(state, 'run MSCK REPAIR to discover new partitions')

adam/commands/audit/audit_run.py

@@ -44,7 +44,7 @@ class AuditRun(Command):
             return state
 
     def completion(self, state: ReplState):
-        return super().completion(state)
+        return {}
 
-    def help(self, _: ReplState):
-        return f"{AuditRun.COMMAND}\t run"
+    def help(self, state: ReplState):
+        return super().help(state, 'run audit')

adam/commands/audit/completions_l.py

@@ -0,0 +1,15 @@
+from adam.repl_state import ReplState
+from adam.sql.lark_completer import LarkCompleter
+from adam.utils_athena import Athena
+
+def completions_l():
+    return LarkCompleter(
+        expandables={
+            'tables': lambda x: Athena.table_names(),
+            'columns': lambda table: Athena.column_names(),
+            'partition-columns': lambda table: Athena.column_names(partition_cols_only=True),
+            'topn-counts': lambda x: ['10'],
+            'topn-types': lambda x: ['last', 'slow', 'top'],
+            'topn-windows': lambda x: ['day', 'month'],
+        }, variant=ReplState.L
+    ).completions_for_nesting()

adam/commands/audit/show_last10.py

@@ -30,8 +30,7 @@ class ShowLast10(Command):
             return state
 
     def completion(self, _: ReplState):
-
         return {}
 
-    def help(self, _: ReplState):
-        return f'{ShowLast10.COMMAND} [limit]\t show last <limit> audit lines'
+    def help(self, state: ReplState):
+        return super().help(state, 'show last <limit> audit lines  <limit> defaults to 10', args='[limit]')

adam/commands/audit/show_slow10.py

@@ -32,5 +32,5 @@ class ShowSlow10(Command):
     def completion(self, _: ReplState):
         return {}
 
-    def help(self, _: ReplState):
-        return f'{ShowSlow10.COMMAND} [limit]\t show slow <limit> audit lines'
+    def help(self, state: ReplState):
+        return super().help(state, 'show slow <limit> audit lines  <limit> default to 10', args='[limit]')

adam/commands/audit/show_top10.py

@@ -32,5 +32,5 @@ class ShowTop10(Command):
     def completion(self, _: ReplState):
         return {}
 
-    def help(self, _: ReplState):
-        return f'{ShowTop10.COMMAND} [limit]\t show top <limit> audit lines'
+    def help(self, state: ReplState):
+        return super().help(state, 'show top <limit> audit lines  <limit> default to 10', args='[limit]')

adam/commands/bash/bash.py

@@ -30,7 +30,7 @@ class Bash(Command):
                 return exec(args)
 
     def completion(self, state: ReplState):
-        return Devices.device(state).bash_completion(Bash.COMMAND, state, default = {})
+        return super().completion(state, {c : {'&': None} for c in ['ls', 'cat', 'head']}, pods=Devices.of(state).pods(state, '-'))
 
-    def help(self, _: ReplState):
-        return f'{Bash.COMMAND} [pod-name] [bash-commands] [&]\t run bash on the Cassandra nodes'
+    def help(self, state: ReplState):
+        return super().help(state, 'run bash on Cassandra nodes', args='[bash-commands] [&]')

adam/commands/bash/utils_bash.py

@@ -13,4 +13,4 @@ class BashHandler:
         return False
 
     def exec(self, args: list[str]):
-        return Devices.device(self.s1).bash(self.s0, self.s1, args)
+        return Devices.of(self.s1).bash(self.s0, self.s1, args)

adam/commands/cassandra/download_cassandra_log.py

@@ -0,0 +1,45 @@
+from adam.commands.command import Command
+from adam.commands.devices.devices import Devices
+from adam.config import Config
+from adam.utils_k8s.pod_exec_result import PodExecResult
+from adam.utils import log2
+from adam.utils_k8s.cassandra_nodes import CassandraNodes
+from adam.repl_state import ReplState, RequiredState
+from adam.utils_k8s.pods import Pods
+
+class DownloadCassandraLog(Command):
+    COMMAND = 'download cassandra log'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(DownloadCassandraLog, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return DownloadCassandraLog.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):
+            path = Config().get('logs.path', '/c3/cassandra/logs/system.log')
+            r: PodExecResult = CassandraNodes.exec(state.pod, state.namespace, f'cat {path}', backgrounded=True, history=False)
+
+            to_file = Pods.download_file(state.pod, 'cassandra', state.namespace, path)
+            log2(f'Downloaded to {to_file}.')
+
+            return r
+
+    def completion(self, state: ReplState):
+        return super().completion(state, pods=Devices.of(state).pods(state, '-'), auto='jit')
+
+    def help(self, state: ReplState):
+        return super().help(state, 'download cassandra system log')

adam/commands/cassandra/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, state: ReplState):
+        return super().help(state, 'restart all the nodes in the cluster', args='--force')

adam/commands/cassandra/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, state: ReplState):
+        return super().help(state, 'restart the node', args='--force')