kaqing 2.0.82__tar.gz → 2.0.84__tar.gz

{kaqing-2.0.82 → kaqing-2.0.84}/PKG-INFO

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

{kaqing-2.0.82 → kaqing-2.0.84}/adam/batch.py

@@ -1,5 +1,6 @@
 import click
 
+from adam.commands.audit.audit import Audit, AuditCommandHelper
 from adam.commands.bash import Bash
 from adam.commands.check import Check, CheckCommandHelper
 from adam.commands.cp import ClipboardCopy, CopyCommandHelper
@@ -29,6 +30,15 @@ from adam.repl import enter_repl
 from adam.repl_state import ReplState
 from adam.cli_group import cli
 
+@cli.command(context_settings=dict(ignore_unknown_options=True, allow_extra_args=True), cls=AuditCommandHelper, help='Run audit functions.')
+@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.argument('extra_args', nargs=-1, metavar='repair', type=click.UNPROCESSED)
+def audit(kubeconfig: str, config: str, param: list[str], extra_args):
+    run_command(Audit(), kubeconfig, config, param, None, None, None, extra_args)
+
+
 @cli.command(context_settings=dict(ignore_unknown_options=True, allow_extra_args=True), cls=ClusterOrPodCommandHelper, help='Run a single bash command.')
 @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')

kaqing-2.0.84/adam/commands/audit/audit.py

@@ -0,0 +1,57 @@
+import click
+
+from adam.commands.audit.audit_repair_tables import AuditRepairTables
+from adam.commands.command import Command
+from adam.repl_state import ReplState
+from adam.sql.sql_completer import SqlCompleter
+from adam.utils import log2
+from adam.utils_athena import audit_table_names, run_audit_query
+
+class Audit(Command):
+    COMMAND = 'audit'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(Audit, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return Audit.COMMAND
+
+    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)
+
+        r = super().intermediate_run(cmd, state, args, Audit.cmd_list(), display_help=False)
+        if isinstance(r, str) and r == 'command-missing':
+            sql = 'select * from audit order by ts desc limit 10'
+            if args:
+                sql = ' '.join(args)
+            else:
+                log2(sql)
+
+            run_audit_query(sql)
+
+        return state
+
+    def completion(self, state: ReplState):
+        if state.device == ReplState.L:
+            return super().completion(state) | SqlCompleter.completions(lambda: audit_table_names())
+
+        return {}
+
+    def cmd_list():
+        return [AuditRepairTables()]
+
+    def help(self, _: ReplState):
+        return f"{Audit.COMMAND} \t run AWS Athena query on audit database"
+
+class AuditCommandHelper(click.Command):
+    def get_help(self, ctx: click.Context):
+        Command.intermediate_help(super().get_help(ctx), Audit.COMMAND, Audit.cmd_list(), show_cluster_help=False)

kaqing-2.0.84/adam/commands/audit/audit_repair_tables.py

@@ -0,0 +1,42 @@
+from adam.commands.command import Command
+from adam.repl_state import ReplState
+from adam.utils_athena import run_audit_query
+
+class AuditRepairTables(Command):
+    COMMAND = 'audit repair'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(AuditRepairTables, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return AuditRepairTables.COMMAND
+
+    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)
+
+        tables = ['audit']
+        if args:
+            tables = args
+
+        for table in tables:
+            run_audit_query(f'MSCK REPAIR TABLE {table}')
+
+        return state
+
+    def completion(self, state: ReplState):
+        if state.device == ReplState.L:
+            return super().completion(state)
+
+        return {}
+
+    def help(self, _: ReplState):
+        return f"{AuditRepairTables.COMMAND} \t run MSCK REPAIR command for new partition discovery"

{kaqing-2.0.82 → kaqing-2.0.84}/adam/commands/cd.py

@@ -59,6 +59,8 @@ class Cd(Command):
                         state.app_env = dir.split('@')[0]
                     else:
                         state.app_app = dir
+            elif state.device == ReplState.L:
+                pass
             else:
                 if dir == '':
                     state.sts = None
@@ -103,7 +105,7 @@ class Cd(Command):
                 return {Cd.COMMAND: {'..': None} | {app[0].split('-')[1]: None for app in Apps.apps(state.app_env)}}
             else:
                 return {Cd.COMMAND: {'..': None} | {env[0]: None for env in Apps.envs()}}
-        else:
+        elif state.device == ReplState.C:
             if state.pod:
                 return {Cd.COMMAND: {'..': None}}
             elif state.sts:

{kaqing-2.0.82 → kaqing-2.0.84}/adam/commands/command.py

@@ -104,19 +104,22 @@ class Command:
             cmd = s
         print()
 
-    def intermediate_run(self, cmd: str, state: ReplState, args: list[str], cmds: list['Command'], separator='\t'):
+    def intermediate_run(self, cmd: str, state: ReplState, args: list[str], cmds: list['Command'], separator='\t', display_help=True):
         state, _ = self.apply_state(args, state)
 
         if state.in_repl:
-            log(lines_to_tabular([c.help(state) for c in cmds], separator=separator))
+            if display_help:
+                log(lines_to_tabular([c.help(state) for c in cmds], separator=separator))
 
             return 'command-missing'
         else:
             # head with the Chain of Responsibility pattern
             cmds = Command.chain(cmds)
             if not cmds.run(cmd, state):
-                log2('* Command is missing.')
-                Command.display_help()
+                if display_help:
+                    log2('* Command is missing.')
+                    Command.display_help()
+                return 'command-missing'
 
         return state
 

{kaqing-2.0.82 → kaqing-2.0.84}/adam/commands/devices.py

@@ -1,6 +1,35 @@
 from adam.commands.command import Command
 from adam.repl_state import ReplState
 
+class DeviceAuditLog(Command):
+    COMMAND = f'{ReplState.L}:'
+
+    # the singleton pattern
+    def __new__(cls, *args, **kwargs):
+        if not hasattr(cls, 'instance'): cls.instance = super(DeviceAuditLog, cls).__new__(cls)
+
+        return cls.instance
+
+    def __init__(self, successor: Command=None):
+        super().__init__(successor)
+
+    def command(self):
+        return DeviceAuditLog.COMMAND
+
+    def run(self, cmd: str, state: ReplState):
+        if not self.args(cmd):
+            return super().run(cmd, state)
+
+        state.device = ReplState.L
+
+        return state
+
+    def completion(self, state: ReplState):
+        return super().completion(state)
+
+    def help(self, _: ReplState):
+        return f'{DeviceAuditLog.COMMAND}\t move to Audit Log Operations device'
+
 class DeviceCass(Command):
     COMMAND = f'{ReplState.C}:'
 

{kaqing-2.0.82 → kaqing-2.0.84}/adam/commands/ls.py

@@ -14,6 +14,7 @@ from adam.pod_exec_result import PodExecResult
 from adam.repl_state import ReplState
 from adam.utils import lines_to_tabular, log, log2
 from adam.apps import Apps
+from adam.utils_athena import audit_table_names
 
 class Ls(Command):
     COMMAND = 'ls'
@@ -71,6 +72,8 @@ class Ls(Command):
                 svcs = [n for n, ns in Apps.envs()]
 
                 log(lines_to_tabular(svcs, 'ENV', separator=','))
+        elif state.device == ReplState.L:
+            self.show_audit_log_tables()
         else:
             if state.pod:
                 r: PodExecResult = Cqlsh().run(f'cql describe tables', state)
@@ -127,6 +130,9 @@ class Ls(Command):
     def show_pg_tables(self, pg: PostgresSession):
         log(lines_to_tabular(pg_table_names(pg.namespace, pg.directory()), 'NAME', separator=','))
 
+    def show_audit_log_tables(self):
+        log(lines_to_tabular(audit_table_names(), 'NAME', separator=','))
+
     def completion(self, state: ReplState):
         if state.pod:
             return {}

{kaqing-2.0.82 → kaqing-2.0.84}/adam/commands/pwd.py

@@ -40,6 +40,8 @@ class Pwd(Command):
                     words.append(f'env/{state.app_env}')
                 if state.app_app:
                     words.append(f'app/{state.app_app}')
+            elif device == ReplState.L:
+                pass
             else:
                 if state.sts:
                     words.append(f'sts/{state.sts}')
@@ -58,6 +60,7 @@ class Pwd(Command):
         log(lines_to_tabular([
             device_line(state, ReplState.A),
             device_line(state, ReplState.C),
+            device_line(state, ReplState.L),
             device_line(state, ReplState.P),
             f'',
             f'HOST\t{host}',

kaqing-2.0.84/adam/embedded_params.py

@@ -0,0 +1,2 @@
+def config():
+    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, 'cache-username': True, 'url': 'https://{host}/{env}/{app}', 'another': "You're logged in to {has}. However, for this app, you need to log in to {need}.", 'token-server-url': 'http://localhost:{port}', 'password-max-length': 128}, 'strip': '0'}, 'audit': {'endpoint': 'https://4psvtaxlcb.execute-api.us-west-2.amazonaws.com/prod/', 'workers': 3, 'timeout': 10, 'athena': {'region': 'us-west-2', 'catalog': 'AwsDataCatalog', 'database': 'audit', 'output': 's3://s3.ops--audit/ddl/results'}}, 'bash': {'workers': 32}, 'cassandra': {'service-name': 'all-pods-service'}, 'cql': {'workers': 32, 'samples': 3, 'secret': {'cluster-regex': '(.*?-.*?)-.*', 'name': '{cluster}-superuser', 'password-item': 'password'}, 'alter-tables': {'excludes': 'system_auth,system_traces,reaper_db,system_distributed,system_views,system,system_schema,system_virtual_schema', 'gc-grace-periods': '3600,86400,864000,7776000', 'batching': True}}, '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', 'jwks-uri': 'https://login.microsoftonline.com/common/discovery/keys', 'contact': 'Please contact ted.tran@c3.ai.', 'whitelist-file': '/kaqing/members'}, 'okta': {'default': True, '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', 'jwks-uri': 'https://c3energy.okta.com/oauth2/v1/keys'}}, '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': {'name': 'ops-pg-agent', 'just-in-time': False, 'timeout': 86400, 'image': 'seanahnsf/kaqing'}, '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'}}, 'pod': {'name': 'ops', 'image': 'seanahnsf/kaqing-cloud', 'sa': {'name': 'ops', 'proto': 'c3', 'additional-cluster-roles': 'c3aiops-k8ssandra-operator'}, 'label-selector': 'run=ops'}, '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.14', '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': False, 'debugs': {'timings': False, 'exit-on-error': False, 'show-parallelism': False}}

{kaqing-2.0.82 → kaqing-2.0.84}/adam/repl.py

@@ -51,6 +51,8 @@ def enter_repl(state: ReplState):
                 msg += state.app_env
             if state.app_app:
                 msg += f'/{state.app_app}'
+        elif state.device == ReplState.L:
+            msg = f'{ReplState.L}:'
         else:
             msg = f'{ReplState.C}:'
             if state.pod:
@@ -97,6 +99,9 @@ def enter_repl(state: ReplState):
         event.app.current_buffer.text = ''
 
     with concurrent.futures.ThreadPoolExecutor(max_workers=Config().get('audit.workers', 3)) as executor:
+        # warm up AWS lambda - this log line may timeout and get lost, which is fine
+        executor.submit(audit_log, 'entering kaqing repl', state)
+
         # use sorted command list only for auto-completion
         sorted_cmds = sorted(cmd_list, key=lambda cmd: cmd.command())
         while True:
@@ -142,6 +147,10 @@ def enter_repl(state: ReplState):
                             c_sql_tried = True
                             cmd = f'app {cmd}'
                             cmds.run(cmd, state)
+                    elif state.device == ReplState.L:
+                        c_sql_tried = True
+                        cmd = f'audit {cmd}'
+                        cmds.run(cmd, state)
                     elif state.sts:
                         c_sql_tried = True
                         cmd = f'cql {cmd}'
@@ -178,11 +187,14 @@ def audit_log(cmd: str, state: ReplState):
         'line': cmd.replace('"', '""').replace('\n', ' '),
     }
     audit_endpoint = Config().get("audit.endpoint", "https://4psvtaxlcb.execute-api.us-west-2.amazonaws.com/prod/")
-    response = requests.post(audit_endpoint, json=payload, timeout=Config().get("audit.timeout", 10))
-    if response.status_code in [200, 201]:
-        Config().debug(response.text)
-    else:
-        log2(f"Error: {response.status_code} {response.text}")
+    try:
+        response = requests.post(audit_endpoint, json=payload, timeout=Config().get("audit.timeout", 10))
+        if response.status_code in [200, 201]:
+            Config().debug(response.text)
+        else:
+            log2(f"Error: {response.status_code} {response.text}")
+    except requests.exceptions.Timeout as e:
+        log2(f"Timeout occurred: {e}")
 
 @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')

{kaqing-2.0.82 → kaqing-2.0.84}/adam/repl_commands.py

@@ -1,6 +1,8 @@
 from adam.commands.alter_tables import AlterTables
 from adam.commands.app import App
 from adam.commands.app_ping import AppPing
+from adam.commands.audit.audit import Audit
+from adam.commands.audit.audit_repair_tables import AuditRepairTables
 from adam.commands.deploy.code_start import CodeStart
 from adam.commands.deploy.code_stop import CodeStop
 from adam.commands.deploy.deploy import Deploy
@@ -24,7 +26,7 @@ from adam.commands.cd import Cd
 from adam.commands.check import Check
 from adam.commands.command import Command
 from adam.commands.cql.cqlsh import Cqlsh
-from adam.commands.devices import DeviceApp, DeviceCass, DevicePostgres
+from adam.commands.devices import DeviceApp, DeviceAuditLog, DeviceCass, DevicePostgres
 from adam.commands.exit import Exit
 from adam.commands.medusa.medusa import Medusa
 from adam.commands.param_get import GetParam
@@ -77,7 +79,7 @@ class ReplCommands:
         return deduped
 
     def navigation() -> list[Command]:
-        return [Ls(), PreviewTable(), DeviceApp(), DevicePostgres(), DeviceCass(), Cd(), Pwd(), ClipboardCopy(),
+        return [Ls(), PreviewTable(), DeviceApp(), DevicePostgres(), DeviceCass(), DeviceAuditLog(), Cd(), Pwd(), ClipboardCopy(),
                 GetParam(), SetParam(), ShowParams(), ShowKubectlCommands(), ShowLogin(), ShowAdam(), ShowHost()]
 
     def cassandra_check() -> list[Command]:
@@ -89,7 +91,7 @@ class ReplCommands:
 
     def tools() -> list[Command]:
         return [Cqlsh(), Postgres(), Bash(), Shell(), CodeStart(), CodeStop(), DeployFrontend(), UndeployFrontend(),
-                DeployPod(), UndeployPod(), DeployPgAgent(), UndeployPgAgent()]
+                DeployPod(), UndeployPod(), DeployPgAgent(), UndeployPgAgent(), AuditRepairTables(), Audit()]
 
     def app() -> list[Command]:
         return [ShowAppActions(), ShowAppId(), ShowAppQueues(), AppPing(), App()]

{kaqing-2.0.82 → kaqing-2.0.84}/adam/repl_state.py

@@ -48,6 +48,7 @@ class RequiredState(Enum):
 class ReplState:
     A = 'a'
     C = 'c'
+    L = 'l'
     P = 'p'
 
     def __init__(self, device: str = None,
@@ -129,7 +130,7 @@ class ReplState:
             if index < 6:
                 state = copy.copy(state)
 
-                groups = re.match(r'^([a|c|p]):(.*)$', arg)
+                groups = re.match(r'^([a|c|l|p]):(.*)$', arg)
                 if groups:
                     if groups[1] == 'p':
                         state.device = 'p'
@@ -146,6 +147,8 @@ class ReplState:
                                 state.namespace = p_and_ns[1]
                             elif ns := KubeContext.in_cluster_namespace():
                                 state.namespace = ns
+                    elif groups[1] == 'l':
+                        state.device = 'l'
                     else:
                         state.device = 'a'
                         if path := groups[2]:

{kaqing-2.0.82 → kaqing-2.0.84}/adam/sql/state_machine.py

@@ -134,7 +134,7 @@ SPEC = [
     'select_a                        > ,                > select_a_comma_',
     'select_a_comma_                 > name|*           > select_a                                ^ *',
     'select_a_                       > from             > select_from                             ^ from',
-    'select_from_                    > name             > select_from_x                           ^ (select,tables',
+    'select_from_                    > name|audit       > select_from_x                           ^ (select,tables',
     '-                               > (                > select_from_lp_',
     '-                               < )                > select_from_sq',
     'select_from_lp_                 > select           > select',
@@ -269,7 +269,8 @@ KEYWORDS = [
     'left join', 'right outer join', 'right join', 'full join', 'full outer join',
     'insert', 'into', 'values',
     'update', 'where', 'set',
-    'delete'
+    'delete',
+    'audit'
 ]
 
 class StateTo:

kaqing-2.0.84/adam/utils_athena.py

@@ -0,0 +1,61 @@
+import functools
+import time
+import boto3
+
+from adam.config import Config
+from adam.utils import lines_to_tabular, log, log2
+
+@functools.lru_cache()
+def audit_table_names():
+   region_name = Config().get('audit.athena.region', 'us-west-2')
+   database_name = Config().get('audit.athena.database', 'audit')
+   catalog_name = Config().get('audit.athena.catalog', 'AwsDataCatalog')
+
+   athena_client = boto3.client('athena', region_name=region_name)
+   paginator = athena_client.get_paginator('list_table_metadata')
+
+   table_names = []
+   for page in paginator.paginate(CatalogName=catalog_name, DatabaseName=database_name):
+      for table_metadata in page.get('TableMetadataList', []):
+         table_names.append(table_metadata['Name'])
+
+   return table_names
+
+def run_audit_query(sql: str):
+   athena_client = boto3.client('athena')
+
+   database_name = Config().get('audit.athena.database', 'audit')
+   s3_output_location = Config().get('audit.athena.output', 's3://s3.ops--audit/ddl/results')
+
+   response = athena_client.start_query_execution(
+      QueryString=sql,
+      QueryExecutionContext={
+            'Database': database_name
+      },
+      ResultConfiguration={
+            'OutputLocation': s3_output_location
+      }
+   )
+
+   query_execution_id = response['QueryExecutionId']
+
+   while True:
+      query_status = athena_client.get_query_execution(QueryExecutionId=query_execution_id)
+      state = query_status['QueryExecution']['Status']['State']
+      if state in ['SUCCEEDED', 'FAILED', 'CANCELLED']:
+            break
+      time.sleep(1)
+
+   if state == 'SUCCEEDED':
+      results_response = athena_client.get_query_results(QueryExecutionId=query_execution_id)
+      column_info = results_response['ResultSet']['Rows'][0]['Data']
+      columns = [col.get('VarCharValue') for col in column_info]
+      lines = []
+      for row in results_response['ResultSet']['Rows'][1:]:
+            row_data = [col.get('VarCharValue') for col in row['Data']]
+            lines.append('\t'.join(row_data))
+
+      log(lines_to_tabular(lines, header='\t'.join(columns), separator='\t'))
+   else:
+      log2(f"Query failed or was cancelled. State: {state}")
+      log2(f"Reason: {query_status['QueryExecution']['Status'].get('StateChangeReason')}")

kaqing-2.0.84/adam/version.py

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

{kaqing-2.0.82 → kaqing-2.0.84}/kaqing.egg-info/PKG-INFO

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

{kaqing-2.0.82 → kaqing-2.0.84}/kaqing.egg-info/SOURCES.txt

@@ -16,6 +16,7 @@ adam/repl_commands.py
 adam/repl_session.py
 adam/repl_state.py
 adam/utils.py
+adam/utils_athena.py
 adam/utils_net.py
 adam/version.py
 adam/checks/__init__.py
@@ -79,6 +80,9 @@ adam/commands/restart.py
 adam/commands/rollout.py
 adam/commands/shell.py
 adam/commands/watch.py
+adam/commands/audit/__init__.py
+adam/commands/audit/audit.py
+adam/commands/audit/audit_repair_tables.py
 adam/commands/cql/__init__.py
 adam/commands/cql/cql_completions.py
 adam/commands/cql/cql_table_completer.py

{kaqing-2.0.82 → kaqing-2.0.84}/setup.py

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

kaqing-2.0.82/adam/embedded_params.py

@@ -1,2 +0,0 @@
-def config():
-    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, 'cache-username': True, 'url': 'https://{host}/{env}/{app}', 'another': "You're logged in to {has}. However, for this app, you need to log in to {need}.", 'token-server-url': 'http://localhost:{port}', 'password-max-length': 128}, 'strip': '0'}, 'audit': {'endpoint': 'https://4psvtaxlcb.execute-api.us-west-2.amazonaws.com/prod/', 'workers': 3, 'timeout': 10}, 'bash': {'workers': 32}, 'cassandra': {'service-name': 'all-pods-service'}, 'cql': {'workers': 32, 'samples': 3, 'secret': {'cluster-regex': '(.*?-.*?)-.*', 'name': '{cluster}-superuser', 'password-item': 'password'}, 'alter-tables': {'excludes': 'system_auth,system_traces,reaper_db,system_distributed,system_views,system,system_schema,system_virtual_schema', 'gc-grace-periods': '3600,86400,864000,7776000', 'batching': True}}, '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', 'jwks-uri': 'https://login.microsoftonline.com/common/discovery/keys', 'contact': 'Please contact ted.tran@c3.ai.', 'whitelist-file': '/kaqing/members'}, 'okta': {'default': True, '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', 'jwks-uri': 'https://c3energy.okta.com/oauth2/v1/keys'}}, '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': {'name': 'ops-pg-agent', 'just-in-time': False, 'timeout': 86400, 'image': 'seanahnsf/kaqing'}, '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'}}, 'pod': {'name': 'ops', 'image': 'seanahnsf/kaqing-cloud', 'sa': {'name': 'ops', 'proto': 'c3', 'additional-cluster-roles': 'c3aiops-k8ssandra-operator'}, 'label-selector': 'run=ops'}, '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.14', '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': False, 'debugs': {'timings': False, 'exit-on-error': False, 'show-parallelism': False}}

kaqing-2.0.82/adam/version.py

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