kaqing 2.0.177__tar.gz → 2.0.193__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of kaqing might be problematic. Click here for more details.

Files changed (277) hide show
  1. kaqing-2.0.193/MANIFEST.in +1 -0
  2. {kaqing-2.0.177 → kaqing-2.0.193}/PKG-INFO +1 -1
  3. {kaqing-2.0.177 → kaqing-2.0.193}/adam/app_session.py +1 -1
  4. {kaqing-2.0.177 → kaqing-2.0.193}/adam/batch.py +6 -6
  5. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/check_utils.py +3 -1
  6. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/__init__.py +4 -2
  7. kaqing-2.0.177/adam/commands/logs.py → kaqing-2.0.193/adam/commands/app/app.py +12 -11
  8. kaqing-2.0.193/adam/commands/app/app_ping.py +38 -0
  9. kaqing-2.0.193/adam/commands/app/show_app_actions.py +49 -0
  10. kaqing-2.0.193/adam/commands/app/show_app_id.py +44 -0
  11. kaqing-2.0.193/adam/commands/app/show_app_queues.py +38 -0
  12. kaqing-2.0.193/adam/commands/app/utils_app.py +106 -0
  13. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/audit/audit.py +6 -20
  14. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/audit/audit_repair_tables.py +1 -1
  15. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/audit/audit_run.py +1 -1
  16. kaqing-2.0.193/adam/commands/audit/completions_l.py +15 -0
  17. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/audit/show_last10.py +0 -1
  18. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/bash/bash.py +1 -1
  19. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/bash/utils_bash.py +1 -1
  20. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/cat.py +3 -7
  21. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/cd.py +2 -2
  22. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/cli_commands.py +6 -1
  23. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/clipboard_copy.py +3 -2
  24. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/command.py +53 -11
  25. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/commands_utils.py +41 -6
  26. kaqing-2.0.193/adam/commands/cql/completions_c.py +29 -0
  27. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/cql/cqlsh.py +3 -7
  28. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/cql/utils_cql.py +22 -58
  29. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/deploy_pg_agent.py +2 -2
  30. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/undeploy_pg_agent.py +2 -2
  31. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/devices/device.py +39 -8
  32. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/devices/device_app.py +15 -23
  33. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/devices/device_cass.py +13 -19
  34. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/devices/device_export.py +5 -2
  35. kaqing-2.0.193/adam/commands/devices/device_postgres.py +160 -0
  36. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/devices/devices.py +1 -1
  37. kaqing-2.0.193/adam/commands/download_cassandra_log.py +45 -0
  38. kaqing-2.0.193/adam/commands/download_file.py +47 -0
  39. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/clean_up_all_export_sessions.py +3 -3
  40. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/clean_up_export_sessions.py +5 -10
  41. kaqing-2.0.193/adam/commands/export/completions_x.py +11 -0
  42. kaqing-2.0.193/adam/commands/export/download_export_session.py +40 -0
  43. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/export.py +0 -16
  44. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/export_databases.py +13 -6
  45. kaqing-2.0.193/adam/commands/export/export_select.py +34 -0
  46. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/export_sessions.py +91 -6
  47. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/export_use.py +6 -5
  48. kaqing-2.0.193/adam/commands/export/export_x_select.py +48 -0
  49. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/exporter.py +60 -22
  50. kaqing-2.0.193/adam/commands/export/import_files.py +44 -0
  51. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/import_session.py +8 -4
  52. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/importer.py +7 -0
  53. kaqing-2.0.193/adam/commands/export/importer_athena.py +148 -0
  54. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/importer_sqlite.py +29 -4
  55. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/show_column_counts.py +10 -11
  56. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/show_export_databases.py +5 -3
  57. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/show_export_session.py +5 -6
  58. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/show_export_sessions.py +4 -10
  59. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/utils_export.py +42 -14
  60. kaqing-2.0.193/adam/commands/find_files.py +51 -0
  61. kaqing-2.0.193/adam/commands/find_processes.py +76 -0
  62. kaqing-2.0.193/adam/commands/generate_report.py +52 -0
  63. kaqing-2.0.193/adam/commands/head.py +36 -0
  64. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/intermediate_command.py +3 -0
  65. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/ls.py +2 -2
  66. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/medusa/medusa_restore.py +2 -16
  67. kaqing-2.0.193/adam/commands/medusa/utils_medusa.py +15 -0
  68. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/nodetool.py +3 -8
  69. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/param_set.py +1 -1
  70. kaqing-2.0.193/adam/commands/postgres/completions_p.py +22 -0
  71. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/postgres/postgres.py +14 -21
  72. kaqing-2.0.193/adam/commands/postgres/postgres_databases.py +269 -0
  73. kaqing-2.0.193/adam/commands/postgres/utils_postgres.py +79 -0
  74. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/preview_table.py +2 -2
  75. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_schedule_activate.py +6 -2
  76. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_schedule_start.py +1 -2
  77. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_schedule_stop.py +1 -2
  78. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/utils_reaper.py +10 -1
  79. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/repair/repair_scan.py +0 -2
  80. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/repair/repair_stop.py +0 -1
  81. kaqing-2.0.193/adam/commands/restart_cluster.py +47 -0
  82. kaqing-2.0.193/adam/commands/restart_node.py +51 -0
  83. kaqing-2.0.193/adam/commands/restart_nodes.py +47 -0
  84. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/shell.py +1 -1
  85. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show.py +1 -1
  86. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show_cassandra_repairs.py +5 -3
  87. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show_cassandra_status.py +29 -15
  88. kaqing-2.0.177/adam/commands/show/show_commands.py → kaqing-2.0.193/adam/commands/show/show_cli_commands.py +1 -1
  89. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show_params.py +1 -1
  90. kaqing-2.0.193/adam/commands/show/show_processes.py +50 -0
  91. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show_storage.py +9 -8
  92. {kaqing-2.0.177 → kaqing-2.0.193}/adam/config.py +1 -0
  93. kaqing-2.0.193/adam/embedded_params.py +2 -0
  94. {kaqing-2.0.177 → kaqing-2.0.193}/adam/repl.py +17 -9
  95. {kaqing-2.0.177 → kaqing-2.0.193}/adam/repl_commands.py +23 -13
  96. {kaqing-2.0.177 → kaqing-2.0.193}/adam/repl_session.py +7 -1
  97. {kaqing-2.0.177 → kaqing-2.0.193}/adam/repl_state.py +39 -10
  98. kaqing-2.0.193/adam/sql/async_executor.py +44 -0
  99. kaqing-2.0.193/adam/sql/lark_completer.py +286 -0
  100. kaqing-2.0.193/adam/sql/lark_parser.py +604 -0
  101. kaqing-2.0.193/adam/sql/qingl.lark +1076 -0
  102. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sql/sql_state_machine.py +8 -2
  103. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils.py +92 -21
  104. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/app_clusters.py +2 -2
  105. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/app_pods.py +7 -5
  106. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/cassandra_clusters.py +11 -3
  107. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/cassandra_nodes.py +4 -5
  108. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/k8s.py +9 -0
  109. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/pods.py +26 -5
  110. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/statefulsets.py +5 -2
  111. kaqing-2.0.193/adam/utils_local.py +4 -0
  112. kaqing-2.0.193/adam/utils_repl/__init__.py +0 -0
  113. kaqing-2.0.193/adam/utils_repl/appendable_completer.py +6 -0
  114. kaqing-2.0.193/adam/utils_repl/repl_completer.py +89 -0
  115. kaqing-2.0.193/adam/version.py +5 -0
  116. {kaqing-2.0.177 → kaqing-2.0.193}/kaqing.egg-info/PKG-INFO +1 -1
  117. {kaqing-2.0.177 → kaqing-2.0.193}/kaqing.egg-info/SOURCES.txt +37 -9
  118. kaqing-2.0.193/kaqing.egg-info/top_level.txt +2 -0
  119. {kaqing-2.0.177 → kaqing-2.0.193}/setup.py +2 -1
  120. kaqing-2.0.193/teddy/__init__.py +0 -0
  121. kaqing-2.0.193/teddy/lark_parser.py +436 -0
  122. kaqing-2.0.193/teddy/lark_parser2.py +618 -0
  123. kaqing-2.0.177/adam/commands/cql/cql_completions.py +0 -32
  124. kaqing-2.0.177/adam/commands/devices/device_postgres.py +0 -142
  125. kaqing-2.0.177/adam/commands/export/export_select.py +0 -76
  126. kaqing-2.0.177/adam/commands/export/export_select_x.py +0 -54
  127. kaqing-2.0.177/adam/commands/export/importer_athena.py +0 -81
  128. kaqing-2.0.177/adam/commands/postgres/postgres_context.py +0 -272
  129. kaqing-2.0.177/adam/commands/postgres/psql_completions.py +0 -10
  130. kaqing-2.0.177/adam/commands/postgres/utils_postgres.py +0 -70
  131. kaqing-2.0.177/adam/commands/report.py +0 -61
  132. kaqing-2.0.177/adam/commands/restart.py +0 -60
  133. kaqing-2.0.177/adam/commands/show/show_processes.py +0 -49
  134. kaqing-2.0.177/adam/embedded_params.py +0 -2
  135. kaqing-2.0.177/adam/utils_repl/repl_completer.py +0 -46
  136. kaqing-2.0.177/adam/version.py +0 -5
  137. kaqing-2.0.177/kaqing.egg-info/top_level.txt +0 -1
  138. {kaqing-2.0.177 → kaqing-2.0.193}/README +0 -0
  139. {kaqing-2.0.177 → kaqing-2.0.193}/adam/__init__.py +0 -0
  140. {kaqing-2.0.177 → kaqing-2.0.193}/adam/apps.py +0 -0
  141. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/__init__.py +0 -0
  142. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/check.py +0 -0
  143. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/check_context.py +0 -0
  144. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/check_result.py +0 -0
  145. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/compactionstats.py +0 -0
  146. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/cpu.py +0 -0
  147. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/cpu_metrics.py +0 -0
  148. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/disk.py +0 -0
  149. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/gossip.py +0 -0
  150. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/issue.py +0 -0
  151. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/memory.py +0 -0
  152. {kaqing-2.0.177 → kaqing-2.0.193}/adam/checks/status.py +0 -0
  153. {kaqing-2.0.177 → kaqing-2.0.193}/adam/cli.py +0 -0
  154. {kaqing-2.0.177 → kaqing-2.0.193}/adam/cli_group.py +0 -0
  155. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/__init__.py +0 -0
  156. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/column.py +0 -0
  157. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/columns.py +0 -0
  158. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/compactions.py +0 -0
  159. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/cpu.py +0 -0
  160. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/cpu_metrics.py +0 -0
  161. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/dir_data.py +0 -0
  162. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/dir_snapshots.py +0 -0
  163. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/gossip.py +0 -0
  164. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/host_id.py +0 -0
  165. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/memory.py +0 -0
  166. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/node_address.py +0 -0
  167. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/node_load.py +0 -0
  168. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/node_owns.py +0 -0
  169. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/node_status.py +0 -0
  170. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/node_tokens.py +0 -0
  171. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/node_utils.py +0 -0
  172. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/pod_name.py +0 -0
  173. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/volume_cassandra.py +0 -0
  174. {kaqing-2.0.177 → kaqing-2.0.193}/adam/columns/volume_root.py +0 -0
  175. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/alter_tables.py +0 -0
  176. {kaqing-2.0.177/adam/commands/audit → kaqing-2.0.193/adam/commands/app}/__init__.py +0 -0
  177. {kaqing-2.0.177/adam/commands/cql → kaqing-2.0.193/adam/commands/audit}/__init__.py +0 -0
  178. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/audit/show_slow10.py +0 -0
  179. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/audit/show_top10.py +0 -0
  180. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/audit/utils_show_top10.py +0 -0
  181. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/bash/__init__.py +0 -0
  182. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/bash/bash_completer.py +0 -0
  183. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/check.py +0 -0
  184. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/code.py +0 -0
  185. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/command_helpers.py +0 -0
  186. {kaqing-2.0.177/adam/commands/deploy → kaqing-2.0.193/adam/commands/cql}/__init__.py +0 -0
  187. {kaqing-2.0.177/adam/commands/devices → kaqing-2.0.193/adam/commands/deploy}/__init__.py +0 -0
  188. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/code_start.py +0 -0
  189. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/code_stop.py +0 -0
  190. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/code_utils.py +0 -0
  191. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/deploy.py +0 -0
  192. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/deploy_frontend.py +0 -0
  193. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/deploy_pod.py +0 -0
  194. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/deploy_utils.py +0 -0
  195. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/undeploy.py +0 -0
  196. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/undeploy_frontend.py +0 -0
  197. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/deploy/undeploy_pod.py +0 -0
  198. {kaqing-2.0.177/adam/commands/export → kaqing-2.0.193/adam/commands/devices}/__init__.py +0 -0
  199. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/devices/device_auit_log.py +0 -0
  200. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/exit.py +0 -0
  201. {kaqing-2.0.177/adam/commands/medusa → kaqing-2.0.193/adam/commands/export}/__init__.py +0 -0
  202. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/drop_export_database.py +0 -0
  203. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/export/drop_export_databases.py +0 -0
  204. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/help.py +0 -0
  205. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/issues.py +0 -0
  206. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/kubectl.py +0 -0
  207. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/login.py +0 -0
  208. {kaqing-2.0.177/adam/commands/postgres → kaqing-2.0.193/adam/commands/medusa}/__init__.py +0 -0
  209. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/medusa/medusa.py +0 -0
  210. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/medusa/medusa_backup.py +0 -0
  211. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/medusa/medusa_show_backupjobs.py +0 -0
  212. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/medusa/medusa_show_restorejobs.py +0 -0
  213. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/nodetool_commands.py +0 -0
  214. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/param_get.py +0 -0
  215. {kaqing-2.0.177/adam/commands/reaper → kaqing-2.0.193/adam/commands/postgres}/__init__.py +0 -0
  216. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/postgres/postgres_ls.py +0 -0
  217. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/postgres/postgres_preview.py +0 -0
  218. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/pwd.py +0 -0
  219. {kaqing-2.0.177/adam/commands/repair → kaqing-2.0.193/adam/commands/reaper}/__init__.py +0 -0
  220. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper.py +0 -0
  221. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_forward.py +0 -0
  222. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_forward_session.py +0 -0
  223. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_forward_stop.py +0 -0
  224. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_restart.py +0 -0
  225. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_run_abort.py +0 -0
  226. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_runs.py +0 -0
  227. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_runs_abort.py +0 -0
  228. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_schedules.py +0 -0
  229. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/reaper/reaper_status.py +0 -0
  230. {kaqing-2.0.177/adam/commands/show → kaqing-2.0.193/adam/commands/repair}/__init__.py +0 -0
  231. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/repair/repair.py +0 -0
  232. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/repair/repair_log.py +0 -0
  233. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/repair/repair_run.py +0 -0
  234. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/rollout.py +0 -0
  235. {kaqing-2.0.177/adam/sql → kaqing-2.0.193/adam/commands/show}/__init__.py +0 -0
  236. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show_adam.py +0 -0
  237. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show_cassandra_version.py +0 -0
  238. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show_host.py +0 -0
  239. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/show/show_login.py +0 -0
  240. {kaqing-2.0.177 → kaqing-2.0.193}/adam/commands/watch.py +0 -0
  241. {kaqing-2.0.177 → kaqing-2.0.193}/adam/embedded_apps.py +0 -0
  242. {kaqing-2.0.177 → kaqing-2.0.193}/adam/log.py +0 -0
  243. {kaqing-2.0.177 → kaqing-2.0.193}/adam/pod_exec_result.py +0 -0
  244. {kaqing-2.0.177/adam/sso → kaqing-2.0.193/adam/sql}/__init__.py +0 -0
  245. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sql/sql_completer.py +0 -0
  246. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sql/term_completer.py +0 -0
  247. {kaqing-2.0.177/adam/utils_k8s → kaqing-2.0.193/adam/sso}/__init__.py +0 -0
  248. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/authenticator.py +0 -0
  249. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/authn_ad.py +0 -0
  250. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/authn_okta.py +0 -0
  251. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/cred_cache.py +0 -0
  252. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/id_token.py +0 -0
  253. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/idp.py +0 -0
  254. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/idp_login.py +0 -0
  255. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/idp_session.py +0 -0
  256. {kaqing-2.0.177 → kaqing-2.0.193}/adam/sso/sso_config.py +0 -0
  257. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_athena.py +0 -0
  258. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_audits.py +0 -0
  259. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_issues.py +0 -0
  260. {kaqing-2.0.177/adam/utils_repl → kaqing-2.0.193/adam/utils_k8s}/__init__.py +0 -0
  261. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/config_maps.py +0 -0
  262. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/custom_resources.py +0 -0
  263. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/deployment.py +0 -0
  264. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/ingresses.py +0 -0
  265. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/jobs.py +0 -0
  266. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/kube_context.py +0 -0
  267. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/secrets.py +0 -0
  268. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/service_accounts.py +0 -0
  269. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/services.py +0 -0
  270. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_k8s/volumes.py +0 -0
  271. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_net.py +0 -0
  272. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_repl/automata_completer.py +0 -0
  273. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_repl/state_machine.py +0 -0
  274. {kaqing-2.0.177 → kaqing-2.0.193}/adam/utils_sqlite.py +0 -0
  275. {kaqing-2.0.177 → kaqing-2.0.193}/kaqing.egg-info/dependency_links.txt +0 -0
  276. {kaqing-2.0.177 → kaqing-2.0.193}/kaqing.egg-info/entry_points.txt +0 -0
  277. {kaqing-2.0.177 → kaqing-2.0.193}/setup.cfg +0 -0
@@ -0,0 +1 @@
1
+ recursive-include adam qingl.lark
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: kaqing
3
- Version: 2.0.177
3
+ Version: 2.0.193
4
4
  Summary: UNKNOWN
5
5
  Home-page: UNKNOWN
6
6
  License: UNKNOWN
@@ -8,7 +8,7 @@ from adam.log import Log
8
8
  from adam.sso.idp import Idp
9
9
  from adam.sso.idp_login import IdpLogin
10
10
  from adam.config import Config
11
- from adam.utils import debug, debug_trace, json_to_csv, tabulize, log, log2, log_exc
11
+ from adam.utils import debug, debug_trace, json_to_csv, tabulize, log2, log_exc
12
12
  from adam.apps import Apps
13
13
 
14
14
  class AppLogin:
@@ -11,7 +11,7 @@ from adam.commands.deploy.deploy import Deploy, DeployCommandHelper
11
11
  from adam.commands.deploy.undeploy import Undeploy, UndeployCommandHelper
12
12
  from adam.commands.issues import Issues
13
13
  from adam.commands.login import Login
14
- from adam.commands.logs import Logs
14
+ from adam.commands.download_cassandra_log import DownloadCassandraLog
15
15
  from adam.commands.ls import Ls
16
16
  from adam.commands.medusa.medusa import Medusa
17
17
  from adam.commands.nodetool import NodeTool, NodeToolCommandHelper
@@ -19,8 +19,8 @@ from adam.commands.postgres.postgres import Postgres, PostgresCommandHelper
19
19
  from adam.commands.preview_table import PreviewTable
20
20
  from adam.commands.reaper.reaper import Reaper, ReaperCommandHelper
21
21
  from adam.commands.repair.repair import Repair, RepairCommandHelper
22
- from adam.commands.report import Report
23
- from adam.commands.restart import Restart
22
+ from adam.commands.generate_report import GenerateReport
23
+ from adam.commands.restart_nodes import RestartNodes
24
24
  from adam.commands.rollout import RollOut
25
25
  from adam.commands.show.show import Show, ShowCommandHelper
26
26
  from adam.commands.watch import Watch
@@ -128,7 +128,7 @@ def login(kubeconfig: str, config: str, param: list[str], namespace: str, pod: s
128
128
  @click.option('--pod', '-p', required=False, metavar='pod', help='Kubernetes pod name')
129
129
  @click.argument('extra_args', nargs=-1, metavar='<pod>', type=click.UNPROCESSED)
130
130
  def logs(kubeconfig: str, config: str, param: list[str], namespace: str, pod: str, extra_args):
131
- run_command(Logs(), kubeconfig, config, param, None, namespace, pod, extra_args)
131
+ run_command(DownloadCassandraLog(), kubeconfig, config, param, None, namespace, pod, extra_args)
132
132
 
133
133
 
134
134
  @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
221
221
  @click.option('--show', '-s', is_flag=True, help='show output from Cassandra nodes')
222
222
  @click.argument('extra_args', nargs=-1, metavar='[cluster|pod]', type=click.UNPROCESSED)
223
223
  def report(kubeconfig: str, config: str, param: list[str], cluster: str, namespace: str, pod: str, show: bool, extra_args):
224
- run_command(Report(), kubeconfig, config, param, cluster, namespace, pod, ('-s',) + extra_args if show else extra_args)
224
+ run_command(GenerateReport(), kubeconfig, config, param, cluster, namespace, pod, ('-s',) + extra_args if show else extra_args)
225
225
 
226
226
 
227
227
  @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
234
234
  @click.option('--force', is_flag=True, help='need for restarting the whole cluster')
235
235
  @click.argument('extra_args', nargs=-1, metavar='<cluster|pod>', type=click.UNPROCESSED)
236
236
  def restart(kubeconfig: str, config: str, param: list[str], cluster: str, namespace: str, pod: str, force: bool, extra_args):
237
- run_command(Restart(), kubeconfig, config, param, cluster, namespace, pod, ('--force',) + extra_args if force else extra_args)
237
+ run_command(RestartNodes(), kubeconfig, config, param, cluster, namespace, pod, ('--force',) + extra_args if force else extra_args)
238
238
 
239
239
 
240
240
  @cli.command(context_settings=dict(ignore_unknown_options=True, allow_extra_args=True), cls=ClusterOrPodCommandHelper, help='Rolling restart Cassandra Cluster.')
@@ -47,7 +47,9 @@ def run_checks(cluster: str = None, namespace: str = None, pod: str = None, chec
47
47
  if not pod or pod == pod_name:
48
48
  sts_ns_pods.append((sts, ns, pod_name))
49
49
 
50
- with parallelize(sts_ns_pods, Config().action_workers('issues', 30), msg='d`Running|Ran checks on {size} pods') as exec:
50
+ with parallelize(sts_ns_pods,
51
+ Config().action_workers('issues', 30),
52
+ msg='d`Running|Ran checks on {size} pods') as exec:
51
53
  return exec.map(lambda sts_ns_pod: run_checks_on_pod(checks, sts_ns_pod[0], sts_ns_pod[1], sts_ns_pod[2], show_out))
52
54
 
53
55
  def run_checks_on_pod(checks: list[Check], cluster: str = None, namespace: str = None, pod: str = None, show_out=True):
@@ -18,5 +18,7 @@ def extract_all_options(args: list[str], trailing = None, sequence = None, optio
18
18
  def extract_sequence(args: list[str], sequence: list[str]):
19
19
  return ExtractSequenceOptionsHandler(args, sequence = sequence)
20
20
 
21
- def validate_args(args: list[str], state: ReplState, count_at_least: int = 1, name: str = None, msg: Callable[[], None] = None, default: str = None):
22
- return ValidateArgCountHandler(args, state, count_at_least=count_at_least, name=name, msg=msg, default=default)
21
+ def validate_args(args: list[str], state: ReplState, at_least: int = 1, exactly: int = -1,
22
+ name: str = None, msg: Callable[[], None] = None, default: str = None,
23
+ separator=None):
24
+ return ValidateArgCountHandler(args, state, at_least=at_least, exactly=exactly, name=name, msg=msg, default=default, separator=separator)
@@ -1,14 +1,13 @@
1
+ from adam.commands import app, extract_options
1
2
  from adam.commands.command import Command
2
- from adam.config import Config
3
- from adam.utils_k8s.cassandra_nodes import CassandraNodes
4
3
  from adam.repl_state import ReplState, RequiredState
5
4
 
6
- class Logs(Command):
7
- COMMAND = 'logs'
5
+ class App(Command):
6
+ COMMAND = 'app'
8
7
 
9
8
  # the singleton pattern
10
9
  def __new__(cls, *args, **kwargs):
11
- if not hasattr(cls, 'instance'): cls.instance = super(Logs, cls).__new__(cls)
10
+ if not hasattr(cls, 'instance'): cls.instance = super(App, cls).__new__(cls)
12
11
 
13
12
  return cls.instance
14
13
 
@@ -16,22 +15,24 @@ class Logs(Command):
16
15
  super().__init__(successor)
17
16
 
18
17
  def command(self):
19
- return Logs.COMMAND
18
+ return App.COMMAND
20
19
 
21
20
  def required(self):
22
- return RequiredState.POD
21
+ return RequiredState.APP_APP
23
22
 
24
23
  def run(self, cmd: str, state: ReplState):
25
24
  if not(args := self.args(cmd)):
26
25
  return super().run(cmd, state)
27
26
 
28
27
  with self.validate(args, state) as (args, state):
29
- path = Config().get('logs.path', '/c3/cassandra/logs/system.log')
30
- return CassandraNodes.exec(state.pod, state.namespace, f'cat {path}')
28
+ with extract_options(args, '--force') as (args, forced):
29
+ with app(state) as http:
30
+ http.post(args, forced=forced)
31
+
32
+ return state
31
33
 
32
34
  def completion(self, _: ReplState):
33
- # available only on cli
34
35
  return {}
35
36
 
36
37
  def help(self, _: ReplState):
37
- return f'{Logs.COMMAND}\t show cassandra system log'
38
+ return f"<AppType>.<AppAction> <args> [--force]\t post app action; check with 'show app actions' command"
@@ -0,0 +1,38 @@
1
+ from adam.commands import app, extract_options
2
+ from adam.commands.command import Command
3
+ from adam.repl_state import ReplState, RequiredState
4
+
5
+ class AppPing(Command):
6
+ COMMAND = 'app ping'
7
+
8
+ # the singleton pattern
9
+ def __new__(cls, *args, **kwargs):
10
+ if not hasattr(cls, 'instance'): cls.instance = super(AppPing, cls).__new__(cls)
11
+
12
+ return cls.instance
13
+
14
+ def __init__(self, successor: Command=None):
15
+ super().__init__(successor)
16
+
17
+ def command(self):
18
+ return AppPing.COMMAND
19
+
20
+ def required(self):
21
+ return RequiredState.APP_APP
22
+
23
+ def run(self, cmd: str, state: ReplState):
24
+ if not(args := self.args(cmd)):
25
+ return super().run(cmd, state)
26
+
27
+ with self.validate(args, state) as (args, state):
28
+ with extract_options(args, '--force') as (args, forced):
29
+ with app(state) as http:
30
+ http.post(['Echo.echoStatic'], forced=forced)
31
+
32
+ return state
33
+
34
+ def completion(self, state: ReplState):
35
+ return super().completion(state, {'--force': None})
36
+
37
+ def help(self, _: ReplState):
38
+ return f"{AppPing.COMMAND} [--force]\t ping app server with Echo.echoStatic()"
@@ -0,0 +1,49 @@
1
+ from adam.app_session import AppSession
2
+ from adam.apps import Apps
3
+ from adam.commands.command import Command
4
+ from adam.config import Config
5
+ from adam.repl_state import ReplState
6
+ from adam.utils import tabulize, log
7
+
8
+ class ShowAppActions(Command):
9
+ COMMAND = 'show app actions'
10
+
11
+ # the singleton pattern
12
+ def __new__(cls, *args, **kwargs):
13
+ if not hasattr(cls, 'instance'): cls.instance = super(ShowAppActions, cls).__new__(cls)
14
+
15
+ return cls.instance
16
+
17
+ def __init__(self, successor: Command=None):
18
+ super().__init__(successor)
19
+
20
+ def command(self):
21
+ return ShowAppActions.COMMAND
22
+
23
+ def required(self):
24
+ return ReplState.A
25
+
26
+ def run(self, cmd: str, state: ReplState):
27
+ if not (args := self.args(cmd)):
28
+ return super().run(cmd, state)
29
+
30
+ with self.validate(args, state) as (args, state):
31
+ actions = []
32
+ for typ in Apps().app_types():
33
+ actions.extend(typ.actions)
34
+
35
+ lines = tabulize(actions, lambda a: str(a), header='ACTION,ARGS,DESCRIPTION', separator=',')
36
+ log()
37
+
38
+ app_session: AppSession = AppSession.create(state.app_env or 'c3', state.app_app or 'c3')
39
+ endpoint = Config().get('app.console-endpoint', 'https://{host}/{env}/{app}/static/console/index.html')
40
+ endpoint = endpoint.replace('{host}', app_session.host).replace('{env}', app_session.env).replace('{app}', state.app_app or 'c3')
41
+ tabulize([f'CONSOLE:,{endpoint}'], separator=',')
42
+
43
+ return lines
44
+
45
+ def completion(self, state: ReplState):
46
+ return super().completion(state)
47
+
48
+ def help(self, _: ReplState):
49
+ return f"{ShowAppActions.COMMAND}\t show app actions"
@@ -0,0 +1,44 @@
1
+ from adam.commands.command import Command
2
+ from adam.utils_k8s.custom_resources import CustomResources
3
+ from adam.repl_state import ReplState, RequiredState
4
+ from adam.utils import log
5
+
6
+ class ShowAppId(Command):
7
+ COMMAND = 'show app id'
8
+
9
+ # the singleton pattern
10
+ def __new__(cls, *args, **kwargs):
11
+ if not hasattr(cls, 'instance'): cls.instance = super(ShowAppId, cls).__new__(cls)
12
+
13
+ return cls.instance
14
+
15
+ def __init__(self, successor: Command=None):
16
+ super().__init__(successor)
17
+
18
+ def command(self):
19
+ return ShowAppId.COMMAND
20
+
21
+ def required(self):
22
+ return RequiredState.CLUSTER_OR_POD
23
+
24
+ def run(self, cmd: str, state: ReplState):
25
+ if not(args := self.args(cmd)):
26
+ return super().run(cmd, state)
27
+
28
+ with self.validate(args, state) as (args, state):
29
+ c3_app_id = 'Unknown'
30
+
31
+ apps = CustomResources.get_app_ids()
32
+ cr_name = CustomResources.get_cr_name(state.sts if state.sts else state.pod, namespace=state.namespace)
33
+ if cr_name in apps:
34
+ c3_app_id = (apps[cr_name])
35
+
36
+ log(c3_app_id)
37
+
38
+ return c3_app_id
39
+
40
+ def completion(self, state: ReplState):
41
+ return super().completion(state)
42
+
43
+ def help(self, _: ReplState):
44
+ return f'{ShowAppId.COMMAND}\t show app id for the Cassandra cluster'
@@ -0,0 +1,38 @@
1
+ from adam.commands import app, extract_options
2
+ from adam.commands.command import Command
3
+ from adam.repl_state import ReplState, RequiredState
4
+
5
+ class ShowAppQueues(Command):
6
+ COMMAND = 'show app queues'
7
+
8
+ # the singleton pattern
9
+ def __new__(cls, *args, **kwargs):
10
+ if not hasattr(cls, 'instance'): cls.instance = super(ShowAppQueues, cls).__new__(cls)
11
+
12
+ return cls.instance
13
+
14
+ def __init__(self, successor: Command=None):
15
+ super().__init__(successor)
16
+
17
+ def command(self):
18
+ return ShowAppQueues.COMMAND
19
+
20
+ def required(self):
21
+ return RequiredState.APP_APP
22
+
23
+ def run(self, cmd: str, state: ReplState):
24
+ if not(args := self.args(cmd)):
25
+ return super().run(cmd, state)
26
+
27
+ with self.validate(args, state) as (_, state):
28
+ with extract_options(args, '--force') as (args, forced):
29
+ with app(state) as http:
30
+ http.post(['InvalidationQueue.countAll'], forced=forced)
31
+
32
+ return state
33
+
34
+ def completion(self, state: ReplState):
35
+ return super().completion(state, {'--force': None})
36
+
37
+ def help(self, _: ReplState):
38
+ return f"{ShowAppQueues.COMMAND} [--force]\t show invalidation queue counts"
@@ -0,0 +1,106 @@
1
+ import json
2
+ from typing import Union
3
+
4
+ from adam.app_session import AppSession
5
+ from adam.apps import Apps
6
+ from adam.pod_exec_result import PodExecResult
7
+ from adam.repl_state import ReplState
8
+ from adam.utils import log2
9
+ from adam.utils_k8s.app_clusters import AppClusters
10
+ from adam.utils_k8s.app_pods import AppPods
11
+
12
+ def app_cd_dirs(state: ReplState):
13
+ if state.app_app:
14
+ return ['..'] + AppPods.pod_names(state.namespace, state.app_env, state.app_app)
15
+ elif state.app_env:
16
+ return ['..'] + [app[0].split('-')[1] for app in Apps.apps(state.app_env)]
17
+ else:
18
+ return [env[0] for env in Apps.envs()]
19
+
20
+ class AppRestHandler:
21
+ def __init__(self, state: ReplState, forced = False):
22
+ self.state = state
23
+ self.forced = forced
24
+
25
+ def __enter__(self):
26
+ return self.post
27
+
28
+ def __exit__(self, exc_type, exc_val, exc_tb):
29
+ return False
30
+
31
+ def post(self, args: list[str]) -> Union[ReplState, str]:
32
+ if not args:
33
+ return 'arg missing'
34
+
35
+ t_f = args[0].split('.')
36
+ if len(t_f) < 2:
37
+ return 'arg missing'
38
+
39
+ state = self.state
40
+
41
+ payload, valid = Apps().payload(t_f[0], t_f[1], args[1:] if len(args) > 1 else [])
42
+ if not valid:
43
+ log2('Missing one or more action arguments.')
44
+ return state
45
+
46
+ if payload:
47
+ try:
48
+ payload = json.loads(payload)
49
+ except json.decoder.JSONDecodeError as e:
50
+ log2(f'Invalid json argument: {e}')
51
+ return state
52
+
53
+ AppSession.run(state.app_env, state.app_app, state.namespace, t_f[0], t_f[1], payload=payload, forced=self.forced)
54
+
55
+ return state
56
+
57
+ class AppPodService:
58
+ def __init__(self, handler: 'AppHandler'):
59
+ self.handler = handler
60
+
61
+ def exec(self, command: str, show_out = True) -> Union[PodExecResult, list[PodExecResult]]:
62
+ state = self.handler.state
63
+
64
+ if state.app_pod:
65
+ return AppPods.exec(state.app_pod, state.namespace, command, show_out=show_out, shell='bash')
66
+ elif state.app_app:
67
+ pods = AppPods.pod_names(state.namespace, state.app_env, state.app_app)
68
+ return AppClusters.exec(pods, state.namespace, command, action='bash', show_out=show_out, shell='bash')
69
+
70
+ return []
71
+
72
+ def post(self, args: list[str], forced=False) -> Union[ReplState, str]:
73
+ state = self.handler.state
74
+
75
+ if not args:
76
+ return 'arg missing'
77
+
78
+ t_f = args[0].split('.')
79
+ if len(t_f) < 2:
80
+ return 'arg missing'
81
+
82
+ payload, valid = Apps().payload(t_f[0], t_f[1], args[1:] if len(args) > 1 else [])
83
+ if not valid:
84
+ log2('Missing one or more action arguments.')
85
+ return state
86
+
87
+ if payload:
88
+ try:
89
+ payload = json.loads(payload)
90
+ except json.decoder.JSONDecodeError as e:
91
+ log2(f'Invalid json argument: {e}')
92
+ return state
93
+
94
+ AppSession.run(state.app_env, state.app_app, state.namespace, t_f[0], t_f[1], payload=payload, forced=forced)
95
+
96
+ return state
97
+
98
+ class AppHandler:
99
+ def __init__(self, state: ReplState):
100
+ self.state = state
101
+
102
+ def __enter__(self):
103
+ return AppPodService(self)
104
+
105
+ def __exit__(self, exc_type, exc_val, exc_tb):
106
+ return False
@@ -1,6 +1,7 @@
1
1
  import click
2
2
 
3
3
  from adam.commands import validate_args
4
+ from adam.commands.audit.completions_l import completions_l
4
5
  from adam.commands.audit.audit_repair_tables import AuditRepairTables
5
6
  from adam.commands.audit.audit_run import AuditRun
6
7
  from adam.commands.audit.show_last10 import ShowLast10
@@ -10,7 +11,7 @@ from adam.commands.audit.utils_show_top10 import show_top10_completions_for_nest
10
11
  from adam.commands.command import Command
11
12
  from adam.commands.intermediate_command import IntermediateCommand
12
13
  from adam.repl_state import ReplState
13
- from adam.sql.sql_completer import SqlCompleter, SqlVariant
14
+ from adam.sql.lark_completer import LarkCompleter
14
15
  from adam.utils import log2, wait_log
15
16
  from adam.utils_athena import Athena
16
17
 
@@ -50,25 +51,10 @@ class Audit(IntermediateCommand):
50
51
  return state
51
52
 
52
53
  def completion(self, state: ReplState):
53
- if state.device == ReplState.L:
54
- if not self.schema_read:
55
- wait_log(f'Inspecting audit database schema...')
56
- self.schema_read = True
57
- # warm up the caches first time when l: drive is accessed
58
- Athena.table_names()
59
- Athena.column_names()
60
- Athena.column_names(partition_cols_only=True)
61
-
62
- return super().completion(state) | show_top10_completions_for_nesting() | SqlCompleter(
63
- lambda: Athena.table_names(),
64
- expandables={
65
- 'columns': lambda table: Athena.column_names(),
66
- 'partition_columns': lambda table: Athena.column_names(partition_cols_only=True)
67
- },
68
- variant=SqlVariant.ATHENA
69
- ).completions_for_nesting()
70
-
71
- return {}
54
+ if state.device != ReplState.L:
55
+ return {}
56
+
57
+ return completions_l()
72
58
 
73
59
  def cmd_list(self):
74
60
  return [AuditRepairTables(), AuditRun(), ShowLast10(), ShowSlow10(), ShowTop10()]
@@ -46,7 +46,7 @@ class AuditRepairTables(Command):
46
46
  with Audits.offload() as exec:
47
47
  exec.submit(lambda: self.auto_repair(hours))
48
48
 
49
- return super().completion(state)
49
+ # return super().completion(state)
50
50
 
51
51
  return {}
52
52
 
@@ -44,7 +44,7 @@ class AuditRun(Command):
44
44
  return state
45
45
 
46
46
  def completion(self, state: ReplState):
47
- return super().completion(state)
47
+ return {}
48
48
 
49
49
  def help(self, _: ReplState):
50
50
  return f"{AuditRun.COMMAND}\t run"
@@ -0,0 +1,15 @@
1
+ from adam.repl_state import ReplState
2
+ from adam.sql.lark_completer import LarkCompleter
3
+ from adam.utils_athena import Athena
4
+
5
+ def completions_l():
6
+ return LarkCompleter(
7
+ expandables={
8
+ 'tables': lambda x: Athena.table_names(),
9
+ 'columns': lambda table: Athena.column_names(),
10
+ 'partition-columns': lambda table: Athena.column_names(partition_cols_only=True),
11
+ 'topn-counts': lambda x: ['10'],
12
+ 'topn-types': lambda x: ['last', 'slow', 'top'],
13
+ 'topn-windows': lambda x: ['day', 'month'],
14
+ }, variant=ReplState.L
15
+ ).completions_for_nesting()
@@ -30,7 +30,6 @@ class ShowLast10(Command):
30
30
  return state
31
31
 
32
32
  def completion(self, _: ReplState):
33
-
34
33
  return {}
35
34
 
36
35
  def help(self, _: ReplState):
@@ -30,7 +30,7 @@ class Bash(Command):
30
30
  return exec(args)
31
31
 
32
32
  def completion(self, state: ReplState):
33
- return Devices.device(state).bash_completion(Bash.COMMAND, state, default = {})
33
+ return super().completion(state, {c : {'&': None} for c in ['ls', 'cat', 'head']}, pods=Devices.of(state).pods(state, '-'))
34
34
 
35
35
  def help(self, _: ReplState):
36
36
  return f'{Bash.COMMAND} [pod-name] [bash-commands] [&]\t run bash on the Cassandra nodes'
@@ -13,4 +13,4 @@ class BashHandler:
13
13
  return False
14
14
 
15
15
  def exec(self, args: list[str]):
16
- return Devices.device(self.s1).bash(self.s0, self.s1, args)
16
+ return Devices.of(self.s1).bash(self.s0, self.s1, args)
@@ -2,7 +2,6 @@ from adam.commands import validate_args
2
2
  from adam.commands.command import Command
3
3
  from adam.commands.devices.devices import Devices
4
4
  from adam.repl_state import ReplState, RequiredState
5
- from adam.utils import log2
6
5
 
7
6
  class Cat(Command):
8
7
  COMMAND = 'cat'
@@ -28,13 +27,10 @@ class Cat(Command):
28
27
 
29
28
  with self.validate(args, state) as (args, state):
30
29
  with validate_args(args, state, name='file'):
31
- return Devices.device(state).cat(cmd, state)
30
+ return Devices.of(state).bash(state, state, cmd.split(' '))
32
31
 
33
32
  def completion(self, state: ReplState):
34
- if super().completion(state):
35
- return Devices.device(state).cat_completion(Cat.COMMAND, state, default={})
36
-
37
- return {}
33
+ return super().completion(state, lambda: {f: None for f in Devices.of(state).files(state)}, pods=Devices.of(state).pods(state, '-'), auto='jit')
38
34
 
39
35
  def help(self, _: ReplState):
40
- return f'{Cat.COMMAND} file [&]\t run cat command on the Cassandra nodes'
36
+ return f'{Cat.COMMAND} file [&]\t run cat command on the pod'
@@ -28,14 +28,14 @@ class Cd(Command):
28
28
 
29
29
  with self.validate(args, state, apply=False) as (args, state):
30
30
  with validate_args(args, state, name='directory') as arg_str:
31
- device: Device = Devices.device(state)
31
+ device: Device = Devices.of(state)
32
32
  for dir in arg_str.split('/'):
33
33
  device.cd(dir, state)
34
34
 
35
35
  return state
36
36
 
37
37
  def completion(self, state: ReplState):
38
- return Devices.device(state).cd_completion(Cd.COMMAND, state, default = {})
38
+ return Devices.of(state).cd_completion(Cd.COMMAND, state, default = {})
39
39
 
40
40
  def help(self, _: ReplState):
41
41
  return f'{Cd.COMMAND} <path> | .. \t move around on the operational device hierarchy'
@@ -3,14 +3,19 @@ import re
3
3
 
4
4
  from adam.commands.reaper.utils_reaper import Reapers
5
5
  from adam.config import Config
6
+ from adam.utils import log_timing
6
7
  from adam.utils_k8s.kube_context import KubeContext
7
8
  from adam.utils_k8s.secrets import Secrets
8
9
  from adam.utils_k8s.statefulsets import StatefulSets
9
10
  from adam.repl_state import ReplState
10
11
 
11
12
  class CliCommands:
12
- @functools.lru_cache()
13
13
  def values(state: ReplState, collapse = False):
14
+ with log_timing('CliCommands.values'):
15
+ return CliCommands._values(state, collapse)
16
+
17
+ @functools.lru_cache()
18
+ def _values(state: ReplState, collapse = False):
14
19
  # node-exec-?, nodetool-?, cql-?, reaper-exec, reaper-forward, reaper-ui, reaper-usernae, reaper-password
15
20
  d = {}
16
21
 
@@ -6,11 +6,12 @@ from adam.commands import validate_args
6
6
  from adam.commands.command import Command, InvalidArgumentsException
7
7
  from adam.commands.command_helpers import ClusterOrPodCommandHelper
8
8
  from adam.commands.cli_commands import CliCommands
9
+ from adam.config import Config
9
10
  from adam.repl_state import ReplState, RequiredState
10
11
  from adam.utils import tabulize, log, log2
11
12
 
12
13
  class ClipboardCopy(Command):
13
- COMMAND = 'cp'
14
+ COMMAND = 'cli cp'
14
15
 
15
16
  # the singleton pattern
16
17
  def __new__(cls, *args, **kwargs):
@@ -62,7 +63,7 @@ class ClipboardCopy(Command):
62
63
  return state
63
64
 
64
65
  def completion(self, state: ReplState):
65
- return super().completion(state, lambda: {key: None for key in CliCommands.values(state).keys()})
66
+ return super().completion(state, lambda: {key: None for key in CliCommands.values(state).keys()}, auto_key='cli.cp-auto-complete')
66
67
 
67
68
  def help(self, _: ReplState):
68
69
  return f"{ClipboardCopy.COMMAND} <key>\t copy a value to clipboard for conveninence"