kaqing 2.0.188__py3-none-any.whl → 2.0.211__py3-none-any.whl

adam/commands/report.py

@@ -1,61 +0,0 @@
-import click
-import json
-
-from adam.checks.check_result import CheckResult
-from adam.checks.check_utils import run_checks
-from adam.commands.command import Command
-from adam.repl_state import ReplState
-from adam.utils import log2
-
-class Report(Command):
-    COMMAND = 'report'
-
-    # the singleton pattern
-    def __new__(cls, *args, **kwargs):
-        if not hasattr(cls, 'instance'): cls.instance = super(Report, cls).__new__(cls)
-
-        return cls.instance
-
-    def __init__(self, successor: Command=None):
-        super().__init__(successor)
-
-    def command(self):
-        return Report.COMMAND
-
-    def required(self):
-        return ReplState.NON_L
-
-    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):
-            output: dict[str, any] = {}
-
-            if state.in_repl:
-                args, show = Command.extract_options(args, ['-s', '--show'])
-
-                args, redirect = Command.extract_options(args, ['>'])
-                if not redirect or not args:
-                    log2('Please specify file name: e.g. report > /tmp/report.log')
-                    return 'no-report-destination'
-
-                results = run_checks(state.sts, state.namespace, state.pod, show_out=show)
-                output = CheckResult.report(results)
-                with open(args[0], "w") as json_file:
-                    json.dump(output, json_file, indent=2)
-                    log2(f'Report stored in {args[0]}.')
-            else:
-                args, show = Command.extract_options(args, ['-s', '--show'])
-
-                results = run_checks(state.sts, state.namespace, state.pod, show_out=show)
-                output = CheckResult.report(results)
-                click.echo(json.dumps(output, indent=2))
-
-            return output
-
-    def completion(self, state: ReplState):
-        return super().completion(state, {">": None})
-
-    def help(self, _: ReplState):
-        return f"{Report.COMMAND} > <file-name>\t generate report"

adam/commands/restart.py

@@ -1,60 +0,0 @@
-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 Restart(Command):
-    COMMAND = 'restart'
-
-    # the singleton pattern
-    def __new__(cls, *args, **kwargs):
-        if not hasattr(cls, 'instance'): cls.instance = super(Restart, cls).__new__(cls)
-
-        return cls.instance
-
-    def __init__(self, successor: Command=None):
-        super().__init__(successor)
-
-    def command(self):
-        return Restart.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_options(args, '--force') as (args, forced):
-                if not args:
-                    if state.pod:
-                        log2(f'Restarting {state.pod}...')
-                        Pods.delete(state.pod, state.namespace)
-                    else:
-                        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)
-                else:
-                    for arg in args:
-                        Pods.delete(arg, state.namespace)
-
-                return state
-
-    def completion(self, state: ReplState):
-        if super().completion(state):
-            if state.pod:
-                return {Restart.COMMAND: None}
-            elif state.sts:
-                return {Restart.COMMAND: {p: None for p in StatefulSets.pod_names(state.sts, state.namespace)}}
-
-        return {}
-
-    def help(self, _: ReplState):
-        return f"{Restart.COMMAND} [<host-id>] [--force]\t restart the node you're in or all the nodes in the cluster"

adam/commands/shell.py

@@ -1,41 +0,0 @@
-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'
-
-    # the singleton pattern
-    def __new__(cls, *args, **kwargs):
-        if not hasattr(cls, 'instance'): cls.instance = super(Shell, cls).__new__(cls)
-
-        return cls.instance
-
-    def __init__(self, successor: Command=None):
-        super().__init__(successor)
-
-    def command(self):
-        return Shell.COMMAND
-
-    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, _):
-            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')
-
-            return state
-
-    def completion(self, state: ReplState):
-        return super().completion(state)
-
-    def help(self, _: ReplState):
-        return f'{Shell.COMMAND}\t drop down to shell'