kaqing 2.0.184__tar.gz → 2.0.209__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 (280) hide show
  1. kaqing-2.0.209/MANIFEST.in +1 -0
  2. {kaqing-2.0.184 → kaqing-2.0.209}/PKG-INFO +1 -1
  3. {kaqing-2.0.184 → kaqing-2.0.209}/adam/app_session.py +1 -1
  4. {kaqing-2.0.184 → kaqing-2.0.209}/adam/batch.py +7 -7
  5. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/app/app.py +2 -2
  6. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/app/show_app_actions.py +1 -1
  7. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/app/utils_app.py +9 -1
  8. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/audit/audit.py +6 -20
  9. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/audit/audit_repair_tables.py +1 -1
  10. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/audit/audit_run.py +1 -1
  11. kaqing-2.0.209/adam/commands/audit/completions_l.py +15 -0
  12. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/audit/show_last10.py +0 -1
  13. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/bash/bash.py +1 -1
  14. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/bash/utils_bash.py +1 -1
  15. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/cli_commands.py +6 -1
  16. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/clipboard_copy.py +2 -1
  17. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/command.py +22 -9
  18. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/commands_utils.py +14 -6
  19. {kaqing-2.0.184/adam/commands → kaqing-2.0.209/adam/commands/cql}/alter_tables.py +1 -1
  20. kaqing-2.0.209/adam/commands/cql/completions_c.py +29 -0
  21. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/cql/cqlsh.py +2 -6
  22. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/cql/utils_cql.py +21 -13
  23. kaqing-2.0.209/adam/commands/debug/debug.py +22 -0
  24. kaqing-2.0.209/adam/commands/debug/debug_completes.py +35 -0
  25. kaqing-2.0.209/adam/commands/debug/debug_timings.py +35 -0
  26. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/devices/device.py +30 -4
  27. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/devices/device_app.py +1 -1
  28. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/devices/device_export.py +5 -2
  29. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/devices/device_postgres.py +13 -3
  30. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/devices/devices.py +1 -1
  31. kaqing-2.0.209/adam/commands/download_cassandra_log.py +45 -0
  32. kaqing-2.0.209/adam/commands/export/completions_x.py +11 -0
  33. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/download_export_session.py +2 -1
  34. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/export.py +0 -16
  35. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/export_databases.py +5 -4
  36. kaqing-2.0.209/adam/commands/export/export_select.py +34 -0
  37. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/export_sessions.py +12 -11
  38. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/export_use.py +3 -3
  39. kaqing-2.0.184/adam/commands/export/export_select.py → kaqing-2.0.209/adam/commands/export/export_x_select.py +12 -23
  40. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/exporter.py +140 -53
  41. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/import_files.py +2 -2
  42. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/import_session.py +0 -4
  43. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/importer.py +12 -11
  44. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/importer_athena.py +17 -36
  45. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/importer_sqlite.py +21 -9
  46. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/show_column_counts.py +10 -10
  47. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/show_export_databases.py +2 -1
  48. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/show_export_session.py +1 -1
  49. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/show_export_sessions.py +1 -1
  50. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/utils_export.py +36 -14
  51. kaqing-2.0.209/adam/commands/generate_report.py +52 -0
  52. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/intermediate_command.py +3 -0
  53. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/medusa/medusa_restore.py +2 -16
  54. kaqing-2.0.209/adam/commands/medusa/utils_medusa.py +15 -0
  55. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/nodetool.py +2 -7
  56. kaqing-2.0.209/adam/commands/postgres/completions_p.py +22 -0
  57. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/postgres/postgres.py +7 -14
  58. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/postgres/postgres_databases.py +3 -3
  59. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/postgres/postgres_ls.py +1 -1
  60. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/postgres/utils_postgres.py +12 -2
  61. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/preview_table.py +1 -1
  62. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_schedule_activate.py +6 -2
  63. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_schedule_start.py +1 -2
  64. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_schedule_stop.py +1 -2
  65. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/utils_reaper.py +10 -1
  66. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/repair/repair_scan.py +0 -2
  67. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/repair/repair_stop.py +0 -1
  68. kaqing-2.0.209/adam/commands/restart_cluster.py +47 -0
  69. kaqing-2.0.209/adam/commands/restart_node.py +51 -0
  70. kaqing-2.0.209/adam/commands/restart_nodes.py +47 -0
  71. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show.py +1 -1
  72. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show_cassandra_repairs.py +5 -3
  73. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show_cassandra_status.py +28 -15
  74. kaqing-2.0.184/adam/commands/show/show_commands.py → kaqing-2.0.209/adam/commands/show/show_cli_commands.py +1 -1
  75. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show_params.py +1 -1
  76. kaqing-2.0.209/adam/commands/show/show_processes.py +50 -0
  77. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show_storage.py +9 -8
  78. {kaqing-2.0.184 → kaqing-2.0.209}/adam/config.py +4 -5
  79. kaqing-2.0.209/adam/embedded_params.py +2 -0
  80. {kaqing-2.0.184 → kaqing-2.0.209}/adam/repl.py +19 -9
  81. {kaqing-2.0.184 → kaqing-2.0.209}/adam/repl_commands.py +24 -20
  82. {kaqing-2.0.184 → kaqing-2.0.209}/adam/repl_session.py +9 -1
  83. {kaqing-2.0.184 → kaqing-2.0.209}/adam/repl_state.py +16 -1
  84. kaqing-2.0.209/adam/sql/async_executor.py +44 -0
  85. kaqing-2.0.209/adam/sql/lark_completer.py +286 -0
  86. kaqing-2.0.209/adam/sql/lark_parser.py +604 -0
  87. kaqing-2.0.209/adam/sql/qingl.lark +1076 -0
  88. kaqing-2.0.209/adam/sso/__init__.py +0 -0
  89. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/cred_cache.py +2 -5
  90. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils.py +209 -79
  91. kaqing-2.0.209/adam/utils_k8s/__init__.py +0 -0
  92. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/app_clusters.py +11 -4
  93. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/app_pods.py +10 -5
  94. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/cassandra_clusters.py +8 -4
  95. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/cassandra_nodes.py +14 -5
  96. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/k8s.py +9 -0
  97. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/kube_context.py +1 -4
  98. {kaqing-2.0.184/adam → kaqing-2.0.209/adam/utils_k8s}/pod_exec_result.py +8 -2
  99. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/pods.py +83 -24
  100. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/statefulsets.py +5 -2
  101. kaqing-2.0.209/adam/utils_local.py +52 -0
  102. kaqing-2.0.209/adam/utils_repl/__init__.py +0 -0
  103. kaqing-2.0.209/adam/utils_repl/appendable_completer.py +6 -0
  104. kaqing-2.0.209/adam/utils_repl/repl_completer.py +89 -0
  105. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_sqlite.py +3 -8
  106. kaqing-2.0.209/adam/version.py +5 -0
  107. {kaqing-2.0.184 → kaqing-2.0.209}/kaqing.egg-info/PKG-INFO +1 -1
  108. {kaqing-2.0.184 → kaqing-2.0.209}/kaqing.egg-info/SOURCES.txt +30 -18
  109. kaqing-2.0.209/kaqing.egg-info/top_level.txt +2 -0
  110. {kaqing-2.0.184 → kaqing-2.0.209}/setup.py +2 -1
  111. kaqing-2.0.209/teddy/__init__.py +0 -0
  112. kaqing-2.0.209/teddy/lark_parser.py +436 -0
  113. kaqing-2.0.209/teddy/lark_parser2.py +618 -0
  114. kaqing-2.0.184/adam/commands/cat.py +0 -36
  115. kaqing-2.0.184/adam/commands/cd.py +0 -41
  116. kaqing-2.0.184/adam/commands/cql/cql_completions.py +0 -32
  117. kaqing-2.0.184/adam/commands/download_file.py +0 -47
  118. kaqing-2.0.184/adam/commands/export/export_select_x.py +0 -54
  119. kaqing-2.0.184/adam/commands/find_files.py +0 -51
  120. kaqing-2.0.184/adam/commands/find_processes.py +0 -76
  121. kaqing-2.0.184/adam/commands/head.py +0 -36
  122. kaqing-2.0.184/adam/commands/logs.py +0 -37
  123. kaqing-2.0.184/adam/commands/ls.py +0 -41
  124. kaqing-2.0.184/adam/commands/postgres/psql_completions.py +0 -11
  125. kaqing-2.0.184/adam/commands/report.py +0 -61
  126. kaqing-2.0.184/adam/commands/restart.py +0 -60
  127. kaqing-2.0.184/adam/commands/shell.py +0 -41
  128. kaqing-2.0.184/adam/commands/show/show_processes.py +0 -49
  129. kaqing-2.0.184/adam/embedded_params.py +0 -2
  130. kaqing-2.0.184/adam/utils_local.py +0 -4
  131. kaqing-2.0.184/adam/utils_repl/repl_completer.py +0 -46
  132. kaqing-2.0.184/adam/version.py +0 -5
  133. kaqing-2.0.184/kaqing.egg-info/top_level.txt +0 -1
  134. {kaqing-2.0.184 → kaqing-2.0.209}/README +0 -0
  135. {kaqing-2.0.184 → kaqing-2.0.209}/adam/__init__.py +0 -0
  136. {kaqing-2.0.184 → kaqing-2.0.209}/adam/apps.py +0 -0
  137. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/__init__.py +0 -0
  138. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/check.py +0 -0
  139. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/check_context.py +0 -0
  140. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/check_result.py +0 -0
  141. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/check_utils.py +0 -0
  142. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/compactionstats.py +0 -0
  143. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/cpu.py +0 -0
  144. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/cpu_metrics.py +0 -0
  145. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/disk.py +0 -0
  146. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/gossip.py +0 -0
  147. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/issue.py +0 -0
  148. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/memory.py +0 -0
  149. {kaqing-2.0.184 → kaqing-2.0.209}/adam/checks/status.py +0 -0
  150. {kaqing-2.0.184 → kaqing-2.0.209}/adam/cli.py +0 -0
  151. {kaqing-2.0.184 → kaqing-2.0.209}/adam/cli_group.py +0 -0
  152. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/__init__.py +0 -0
  153. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/column.py +0 -0
  154. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/columns.py +0 -0
  155. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/compactions.py +0 -0
  156. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/cpu.py +0 -0
  157. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/cpu_metrics.py +0 -0
  158. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/dir_data.py +0 -0
  159. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/dir_snapshots.py +0 -0
  160. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/gossip.py +0 -0
  161. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/host_id.py +0 -0
  162. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/memory.py +0 -0
  163. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/node_address.py +0 -0
  164. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/node_load.py +0 -0
  165. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/node_owns.py +0 -0
  166. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/node_status.py +0 -0
  167. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/node_tokens.py +0 -0
  168. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/node_utils.py +0 -0
  169. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/pod_name.py +0 -0
  170. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/volume_cassandra.py +0 -0
  171. {kaqing-2.0.184 → kaqing-2.0.209}/adam/columns/volume_root.py +0 -0
  172. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/__init__.py +0 -0
  173. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/app/__init__.py +0 -0
  174. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/app/app_ping.py +0 -0
  175. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/app/show_app_id.py +0 -0
  176. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/app/show_app_queues.py +0 -0
  177. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/audit/__init__.py +0 -0
  178. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/audit/show_slow10.py +0 -0
  179. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/audit/show_top10.py +0 -0
  180. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/audit/utils_show_top10.py +0 -0
  181. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/bash/__init__.py +0 -0
  182. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/bash/bash_completer.py +0 -0
  183. {kaqing-2.0.184/adam/commands/cql → kaqing-2.0.209/adam/commands/cassandra}/__init__.py +0 -0
  184. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/check.py +0 -0
  185. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/code.py +0 -0
  186. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/command_helpers.py +0 -0
  187. {kaqing-2.0.184/adam/commands/deploy → kaqing-2.0.209/adam/commands/cql}/__init__.py +0 -0
  188. {kaqing-2.0.184/adam/commands/devices → kaqing-2.0.209/adam/commands/debug}/__init__.py +0 -0
  189. {kaqing-2.0.184/adam/commands/export → kaqing-2.0.209/adam/commands/deploy}/__init__.py +0 -0
  190. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/code_start.py +0 -0
  191. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/code_stop.py +0 -0
  192. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/code_utils.py +0 -0
  193. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/deploy.py +0 -0
  194. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/deploy_frontend.py +0 -0
  195. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/deploy_pg_agent.py +0 -0
  196. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/deploy_pod.py +0 -0
  197. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/deploy_utils.py +0 -0
  198. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/undeploy.py +0 -0
  199. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/undeploy_frontend.py +0 -0
  200. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/undeploy_pg_agent.py +0 -0
  201. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/deploy/undeploy_pod.py +0 -0
  202. {kaqing-2.0.184/adam/commands/medusa → kaqing-2.0.209/adam/commands/devices}/__init__.py +0 -0
  203. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/devices/device_auit_log.py +0 -0
  204. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/devices/device_cass.py +0 -0
  205. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/exit.py +0 -0
  206. {kaqing-2.0.184/adam/commands/postgres → kaqing-2.0.209/adam/commands/export}/__init__.py +0 -0
  207. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/clean_up_all_export_sessions.py +0 -0
  208. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/clean_up_export_sessions.py +0 -0
  209. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/drop_export_database.py +0 -0
  210. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/export/drop_export_databases.py +0 -0
  211. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/help.py +0 -0
  212. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/issues.py +0 -0
  213. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/kubectl.py +0 -0
  214. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/login.py +0 -0
  215. {kaqing-2.0.184/adam/commands/reaper → kaqing-2.0.209/adam/commands/medusa}/__init__.py +0 -0
  216. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/medusa/medusa.py +0 -0
  217. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/medusa/medusa_backup.py +0 -0
  218. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/medusa/medusa_show_backupjobs.py +0 -0
  219. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/medusa/medusa_show_restorejobs.py +0 -0
  220. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/nodetool_commands.py +0 -0
  221. {kaqing-2.0.184/adam/commands/repair → kaqing-2.0.209/adam/commands/os}/__init__.py +0 -0
  222. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/param_get.py +0 -0
  223. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/param_set.py +0 -0
  224. {kaqing-2.0.184/adam/commands/show → kaqing-2.0.209/adam/commands/postgres}/__init__.py +0 -0
  225. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/postgres/postgres_preview.py +0 -0
  226. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/pwd.py +0 -0
  227. {kaqing-2.0.184/adam/sql → kaqing-2.0.209/adam/commands/reaper}/__init__.py +0 -0
  228. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper.py +0 -0
  229. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_forward.py +0 -0
  230. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_forward_session.py +0 -0
  231. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_forward_stop.py +0 -0
  232. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_restart.py +0 -0
  233. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_run_abort.py +0 -0
  234. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_runs.py +0 -0
  235. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_runs_abort.py +0 -0
  236. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_schedules.py +0 -0
  237. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/reaper/reaper_status.py +0 -0
  238. {kaqing-2.0.184/adam/sso → kaqing-2.0.209/adam/commands/repair}/__init__.py +0 -0
  239. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/repair/repair.py +0 -0
  240. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/repair/repair_log.py +0 -0
  241. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/repair/repair_run.py +0 -0
  242. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/rollout.py +0 -0
  243. {kaqing-2.0.184/adam/utils_k8s → kaqing-2.0.209/adam/commands/show}/__init__.py +0 -0
  244. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show_adam.py +0 -0
  245. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show_cassandra_version.py +0 -0
  246. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show_host.py +0 -0
  247. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/show/show_login.py +0 -0
  248. {kaqing-2.0.184 → kaqing-2.0.209}/adam/commands/watch.py +0 -0
  249. {kaqing-2.0.184 → kaqing-2.0.209}/adam/embedded_apps.py +0 -0
  250. {kaqing-2.0.184 → kaqing-2.0.209}/adam/log.py +0 -0
  251. {kaqing-2.0.184/adam/utils_repl → kaqing-2.0.209/adam/sql}/__init__.py +0 -0
  252. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sql/sql_completer.py +0 -0
  253. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sql/sql_state_machine.py +0 -0
  254. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sql/term_completer.py +0 -0
  255. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/authenticator.py +0 -0
  256. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/authn_ad.py +0 -0
  257. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/authn_okta.py +0 -0
  258. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/id_token.py +0 -0
  259. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/idp.py +0 -0
  260. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/idp_login.py +0 -0
  261. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/idp_session.py +0 -0
  262. {kaqing-2.0.184 → kaqing-2.0.209}/adam/sso/sso_config.py +0 -0
  263. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_athena.py +0 -0
  264. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_audits.py +0 -0
  265. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_issues.py +0 -0
  266. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/config_maps.py +0 -0
  267. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/custom_resources.py +0 -0
  268. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/deployment.py +0 -0
  269. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/ingresses.py +0 -0
  270. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/jobs.py +0 -0
  271. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/secrets.py +0 -0
  272. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/service_accounts.py +0 -0
  273. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/services.py +0 -0
  274. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_k8s/volumes.py +0 -0
  275. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_net.py +0 -0
  276. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_repl/automata_completer.py +0 -0
  277. {kaqing-2.0.184 → kaqing-2.0.209}/adam/utils_repl/state_machine.py +0 -0
  278. {kaqing-2.0.184 → kaqing-2.0.209}/kaqing.egg-info/dependency_links.txt +0 -0
  279. {kaqing-2.0.184 → kaqing-2.0.209}/kaqing.egg-info/entry_points.txt +0 -0
  280. {kaqing-2.0.184 → kaqing-2.0.209}/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.184
3
+ Version: 2.0.209
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,16 +11,16 @@ 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
15
- from adam.commands.ls import Ls
14
+ from adam.commands.download_cassandra_log import DownloadCassandraLog
15
+ from adam.commands.fs.ls import Ls
16
16
  from adam.commands.medusa.medusa import Medusa
17
17
  from adam.commands.nodetool import NodeTool, NodeToolCommandHelper
18
18
  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.')
@@ -31,8 +31,8 @@ class App(Command):
31
31
 
32
32
  return state
33
33
 
34
- def completion(self, state: ReplState):
35
- return super().completion(state, {'--force': None})
34
+ def completion(self, _: ReplState):
35
+ return {}
36
36
 
37
37
  def help(self, _: ReplState):
38
38
  return f"<AppType>.<AppAction> <args> [--force]\t post app action; check with 'show app actions' command"
@@ -1,5 +1,5 @@
1
1
  from adam.app_session import AppSession
2
- from adam.apps import AppAction, Apps
2
+ from adam.apps import Apps
3
3
  from adam.commands.command import Command
4
4
  from adam.config import Config
5
5
  from adam.repl_state import ReplState
@@ -3,12 +3,20 @@ from typing import Union
3
3
 
4
4
  from adam.app_session import AppSession
5
5
  from adam.apps import Apps
6
- from adam.pod_exec_result import PodExecResult
6
+ from adam.utils_k8s.pod_exec_result import PodExecResult
7
7
  from adam.repl_state import ReplState
8
8
  from adam.utils import log2
9
9
  from adam.utils_k8s.app_clusters import AppClusters
10
10
  from adam.utils_k8s.app_pods import AppPods
11
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
+
12
20
  class AppRestHandler:
13
21
  def __init__(self, state: ReplState, forced = False):
14
22
  self.state = state
@@ -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)
@@ -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,6 +6,7 @@ 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
 
@@ -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"
@@ -5,8 +5,10 @@ import subprocess
5
5
  import sys
6
6
  from typing import Union
7
7
 
8
+ from adam.config import Config
8
9
  from adam.repl_state import ReplState, RequiredState
9
- from adam.utils import is_lambda, log2
10
+ from adam.sql.lark_completer import LarkCompleter
11
+ from adam.utils import log2
10
12
 
11
13
  repl_cmds: list['Command'] = []
12
14
 
@@ -29,10 +31,10 @@ class Command:
29
31
 
30
32
  return None
31
33
 
32
- def completion(self, state: ReplState, leaf: dict[str, any] = None, pods: tuple[list[str], str] = None) -> dict[str, any]:
34
+ def completion(self, state: ReplState, leaf: dict[str, any] = None, pods: tuple[list[str], str] = None, auto: str = 'on', auto_key: str = None) -> dict[str, any]:
33
35
  # pods is a tuple of list of pod names and the current pod repl is on
34
36
  if not pods:
35
- return self._completion(state, leaf)
37
+ return self._completion(state, leaf, auto=auto, auto_key=auto_key)
36
38
 
37
39
  c = {}
38
40
 
@@ -40,18 +42,27 @@ class Command:
40
42
  pod = pods[1]
41
43
 
42
44
  if pod:
43
- c |= self._completion(state, leaf)
45
+ c |= self._completion(state, leaf, auto=auto, auto_key=auto_key)
44
46
 
45
- c |= {f'@{p}': self._completion(state, leaf, to_validate=False) for p in pod_names if p != pod}
47
+ c |= {f'@{p}': self._completion(state.with_pod(p), leaf, to_validate=False, auto=auto, auto_key=auto_key) for p in pod_names if p != pod}
46
48
 
47
49
  return c
48
50
 
49
- def _completion(self, state: ReplState, leaf: dict[str, any] = None, to_validate = True) -> dict[str, any]:
51
+ def _completion(self, state: ReplState, leaf: dict[str, any] = None, to_validate = True, auto: str = 'on', auto_key: str = None) -> dict[str, any]:
50
52
  if to_validate and not self.validate_state(state, show_err=False):
51
53
  return {}
52
54
 
53
- if is_lambda(leaf):
54
- leaf = leaf()
55
+ if callable(leaf):
56
+ if auto_key:
57
+ auto_key = f'auto-complete.{auto_key}'
58
+ auto = Config().get(auto_key, 'off')
59
+
60
+ if auto == 'on':
61
+ leaf = leaf()
62
+ elif auto in ['lazy', 'jit']:
63
+ leaf = LarkCompleter.from_lambda(auto_key, leaf, auto=auto)
64
+ else:
65
+ leaf = None
55
66
 
56
67
  d = leaf
57
68
  for t in reversed(self.command().split(' ')):
@@ -130,7 +141,7 @@ class Command:
130
141
  options = [options]
131
142
 
132
143
  if args and trailing:
133
- while args[-1] in trailing:
144
+ while args and args[-1] in trailing:
134
145
  found_trailing = True
135
146
  args = args[:-1]
136
147
 
@@ -148,6 +159,8 @@ class Command:
148
159
  return new_args, found_trailing, found_sequence, found_options
149
160
 
150
161
  def extract_option_sequence(args, sequence):
162
+ new_args = args
163
+
151
164
  len_sub = len(sequence)
152
165
  len_main = len(args)
153
166
  for i in range(len_main - len_sub + 1):
@@ -8,7 +8,7 @@ from adam.utils_k8s.cassandra_nodes import CassandraNodes
8
8
  from adam.utils_k8s.pods import Pods
9
9
  from adam.utils_k8s.statefulsets import StatefulSets
10
10
  from adam.repl_state import ReplState
11
- from adam.utils import SORT, duration, tabulize, log, log2
11
+ from adam.utils import SORT, duration, kaqing_log_file, tabulize, log2
12
12
 
13
13
  def show_pods(pods: List[client.V1Pod], ns: str, show_namespace = True, show_host_id = True):
14
14
  if len(pods) == 0:
@@ -59,14 +59,22 @@ def show_rollout(sts: str, ns: str):
59
59
  else:
60
60
  log2(f'Cluster has completed rollout {d} ago.')
61
61
 
62
- def show_table(state: ReplState, pods: list[str], cols: str, header: str, show_out=False):
62
+ def show_table(state: ReplState, pods: list[str], cols: str, header: str, show_out=False, backgrounded = False):
63
63
  columns = Columns.create_columns(cols)
64
64
 
65
65
  results = run_checks(cluster=state.sts, pod=state.pod, namespace=state.namespace, checks=collect_checks(columns), show_out=show_out)
66
66
 
67
- def line(pod_name: str):
68
- return
67
+ r = tabulize(pods, lambda p: ','.join([c.pod_value(results, p) for c in columns]), header=header, separator=',', sorted=SORT, to = 0 if backgrounded else 1)
68
+
69
+ if backgrounded:
70
+ r = write_to_kaqing_log_file(r)
71
+
72
+ IssuesUtils.show(results, state.in_repl)
73
+
74
+ return r
69
75
 
70
- tabulize(pods, lambda p: ','.join([c.pod_value(results, p) for c in columns]), header=header, separator=',', sorted=SORT)
76
+ def write_to_kaqing_log_file(r: str):
77
+ with kaqing_log_file() as f:
78
+ f.write(r)
71
79
 
72
- IssuesUtils.show(results, state.in_repl)
80
+ return f.name
@@ -59,7 +59,7 @@ class AlterTables(Command):
59
59
  return state
60
60
 
61
61
  def completion(self, _: ReplState) -> dict[str, any]:
62
- # auto completion is taken care of by sql completer
62
+ # auto completion is taken care of by lark completer
63
63
  return {}
64
64
 
65
65
  def help(self, _: ReplState) -> str:
@@ -0,0 +1,29 @@
1
+ from adam.commands.cql.utils_cql import cassandra_keyspaces, cassandra_table_names
2
+ from adam.commands.export.export_sessions import ExportSessions
3
+ from adam.commands.export.export_databases import ExportDatabases
4
+ from adam.config import Config
5
+ from adam.repl_state import ReplState
6
+ from adam.sql.lark_completer import LarkCompleter
7
+ from adam.utils import log_timing
8
+ from adam.utils_k8s.statefulsets import StatefulSets
9
+
10
+ def completions_c(state: ReplState) -> dict[str, any]:
11
+ ps = Config().get('cql.alter-tables.gc-grace-periods', '3600,86400,864000,7776000').split(',')
12
+
13
+ with log_timing('lark.completions'):
14
+ return LarkCompleter(
15
+ expandables = {
16
+ 'tables': lambda x: cassandra_table_names(state),
17
+ 'keyspaces': lambda x: cassandra_keyspaces(state.with_no_pod()),
18
+ 'table-props': lambda x: {
19
+ 'GC_GRACE_SECONDS': ps
20
+ },
21
+ 'table-props-value': lambda x: {None: None, 'GC_GRACE_SECONDS': ps}[x],
22
+ 'export-database-types': lambda x: ['athena', 'sqlite', 'csv'],
23
+ 'export-databases': lambda x: ExportDatabases.database_names(),
24
+ 'export-sessions': lambda x: ExportSessions.export_session_names(state.sts, state.pod, state.namespace),
25
+ 'export-sessions-incomplete': lambda x: ExportSessions.export_session_names(state.sts, state.pod, state.namespace, export_state='pending_import'),
26
+ 'hosts': lambda x: [f'@{p}' for p in StatefulSets.pod_names(state.sts, state.namespace)],
27
+ },
28
+ variant=ReplState.C
29
+ ).completions_for_nesting()
@@ -3,9 +3,8 @@ import click
3
3
  from adam.commands import extract_trailing_options
4
4
  from adam.commands.command import Command
5
5
  from adam.commands.command_helpers import ClusterOrPodCommandHelper
6
- from adam.commands.cql.cql_completions import cql_completions
6
+ from adam.commands.cql.completions_c import completions_c
7
7
  from adam.commands.cql.utils_cql import cassandra
8
- from adam.utils_k8s.statefulsets import StatefulSets
9
8
  from adam.repl_state import ReplState, RequiredState
10
9
  from adam.utils import log
11
10
 
@@ -38,13 +37,10 @@ class Cqlsh(Command):
38
37
 
39
38
  def completion(self, state: ReplState) -> dict[str, any]:
40
39
  if state.device != state.C:
41
- # conflicts with psql completions
42
40
  return {}
43
41
 
44
42
  if state.sts or state.pod:
45
- c = cql_completions(state)
46
- return c | \
47
- {f'@{p}': c for p in StatefulSets.pod_names(state.sts, state.namespace)}
43
+ return completions_c(state)
48
44
 
49
45
  return {}
50
46
 
@@ -7,11 +7,19 @@ from adam.commands.commands_utils import show_table
7
7
  from adam.utils_k8s.cassandra_clusters import CassandraClusters
8
8
  from adam.utils_k8s.cassandra_nodes import CassandraNodes
9
9
  from adam.utils_k8s.secrets import Secrets
10
- from adam.pod_exec_result import PodExecResult
10
+ from adam.utils_k8s.pod_exec_result import PodExecResult
11
11
  from adam.repl_state import ReplState
12
12
  from adam.utils import log2, log_timing, wait_log
13
13
  from adam.utils_k8s.statefulsets import StatefulSets
14
14
 
15
+ def cd_dirs(state: ReplState) -> list[str]:
16
+ if state.pod:
17
+ return [".."]
18
+ elif state.sts:
19
+ return [".."] + StatefulSets.pod_names(state.sts, state.namespace)
20
+ else:
21
+ return StatefulSets.list_sts_names()
22
+
15
23
  @functools.lru_cache()
16
24
  def cassandra_keyspaces(state: ReplState, on_any=True):
17
25
  if state.pod:
@@ -47,7 +55,7 @@ def table_spec(state: ReplState, table: str, on_any=False) -> 'TableSpec':
47
55
 
48
56
  return parse_cql_desc_table(r.stdout if state.pod else r[0].stdout)
49
57
 
50
- def run_cql(state: ReplState, cql: str, opts: list = [], show_out = False, show_query = False, use_single_quotes = False, on_any = False, backgrounded=False, log_file=None) -> list[PodExecResult]:
58
+ def run_cql(state: ReplState, cql: str, opts: list = [], show_out = False, show_query = False, use_single_quotes = False, on_any = False, backgrounded=False, log_file=None, history=True) -> list[PodExecResult]:
51
59
  if show_query:
52
60
  log2(cql)
53
61
 
@@ -61,7 +69,7 @@ def run_cql(state: ReplState, cql: str, opts: list = [], show_out = False, show_
61
69
 
62
70
  with log_timing(cql):
63
71
  with cassandra(state) as pods:
64
- return pods.exec(command, action='cql', show_out=show_out, on_any=on_any, backgrounded=backgrounded, log_file=log_file)
72
+ return pods.exec(command, action='cql', show_out=show_out, on_any=on_any, backgrounded=backgrounded, log_file=log_file, history=history)
65
73
 
66
74
  def parse_cql_desc_tables(out: str):
67
75
  # Keyspace data_endpoint_auth
@@ -219,16 +227,16 @@ class CassandraPodService:
219
227
  def __init__(self, handler: 'CassandraExecHandler'):
220
228
  self.handler = handler
221
229
 
222
- def exec(self, command: str, action='bash', show_out = True, on_any = False, throw_err = False, shell = '/bin/sh', backgrounded = False, log_file = None) -> Union[PodExecResult, list[PodExecResult]]:
230
+ def exec(self, command: str, action='bash', show_out = True, on_any = False, throw_err = False, shell = '/bin/sh', backgrounded = False, log_file = None, history=True) -> Union[PodExecResult, list[PodExecResult]]:
223
231
  state = self.handler.state
224
232
  pod = self.handler.pod
225
233
 
226
234
  if pod:
227
235
  return CassandraNodes.exec(pod, state.namespace, command,
228
- show_out=show_out, throw_err=throw_err, shell=shell, backgrounded=backgrounded, log_file=log_file)
236
+ show_out=show_out, throw_err=throw_err, shell=shell, backgrounded=backgrounded, log_file=log_file, history=history)
229
237
  elif state.sts:
230
238
  return CassandraClusters.exec(state.sts, state.namespace, command, action=action,
231
- show_out=show_out, on_any=on_any, shell=shell, backgrounded=backgrounded, log_file=log_file)
239
+ show_out=show_out, on_any=on_any, shell=shell, backgrounded=backgrounded, log_file=log_file, history=history)
232
240
 
233
241
  return []
234
242
 
@@ -259,16 +267,16 @@ class CassandraPodService:
259
267
 
260
268
  return run_cql(state, query, opts=opts, show_out=show_out, show_query=show_query, use_single_quotes=use_single_quotes, on_any=on_any, backgrounded=backgrounded, log_file=log_file)
261
269
 
262
- def display_table(self, cols: str, header: str, show_out = True):
270
+ def display_table(self, cols: str, header: str, show_out = True, backgrounded = False):
263
271
  state = self.handler.state
264
272
 
265
273
  if state.pod:
266
- show_table(state, [state.pod], cols, header, show_out=show_out)
274
+ return show_table(state, [state.pod], cols, header, show_out=show_out, backgrounded = backgrounded)
267
275
  elif state.sts:
268
276
  pod_names = [pod.metadata.name for pod in StatefulSets.pods(state.sts, state.namespace)]
269
- show_table(state, pod_names, cols, header, show_out=show_out)
277
+ return show_table(state, pod_names, cols, header, show_out=show_out, backgrounded = backgrounded)
270
278
 
271
- def nodetool(self, args: str, status = False, show_out = True) -> Union[PodExecResult, list[PodExecResult]]:
279
+ def nodetool(self, args: str, status = False, show_out = True, backgrounded = False) -> Union[PodExecResult, list[PodExecResult]]:
272
280
  state = self.handler.state
273
281
  pod = self.handler.pod
274
282
 
@@ -276,9 +284,9 @@ class CassandraPodService:
276
284
  command = f"nodetool -u {user} -pw {pw} {args}"
277
285
 
278
286
  if pod:
279
- return CassandraNodes.exec(pod, state.namespace, command, show_out=show_out)
287
+ return CassandraNodes.exec(pod, state.namespace, command, show_out=show_out, backgrounded=backgrounded)
280
288
  else:
281
- return CassandraClusters.exec(state.sts, state.namespace, command, action='nodetool.status' if status else 'nodetool', show_out=show_out)
289
+ return CassandraClusters.exec(state.sts, state.namespace, command, action='nodetool.status' if status else 'nodetool', show_out=show_out, backgrounded=backgrounded)
282
290
 
283
291
  class CassandraExecHandler:
284
292
  def __init__(self, state: ReplState, pod: str = None):
@@ -294,4 +302,4 @@ class CassandraExecHandler:
294
302
  return False
295
303
 
296
304
  def cassandra(state: ReplState, pod: str=None):
297
- return CassandraExecHandler(state, pod=pod)
305
+ return CassandraExecHandler(state, pod=pod)
@@ -0,0 +1,22 @@
1
+ from adam.commands.command import Command
2
+ from adam.commands.debug.debug_completes import DebugCompletes
3
+ from adam.commands.debug.debug_timings import DebugTimings
4
+ from adam.commands.intermediate_command import IntermediateCommand
5
+
6
+ class Debug(IntermediateCommand):
7
+ COMMAND = 'debug'
8
+
9
+ # the singleton pattern
10
+ def __new__(cls, *args, **kwargs):
11
+ if not hasattr(cls, 'instance'): cls.instance = super(Debug, 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 Debug.COMMAND
20
+
21
+ def cmd_list(self):
22
+ return [DebugTimings(), DebugCompletes()]
@@ -0,0 +1,35 @@
1
+ from adam.commands import validate_args
2
+ from adam.commands.command import Command
3
+ from adam.config import Config
4
+ from adam.repl_state import ReplState
5
+
6
+ class DebugCompletes(Command):
7
+ COMMAND = 'debug completes'
8
+
9
+ # the singleton pattern
10
+ def __new__(cls, *args, **kwargs):
11
+ if not hasattr(cls, 'instance'): cls.instance = super(DebugCompletes, 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 DebugCompletes.COMMAND
20
+
21
+ def run(self, cmd: str, state: ReplState):
22
+ if not(args := self.args(cmd)):
23
+ return super().run(cmd, state)
24
+
25
+ with self.validate(args, state) as (args, state):
26
+ with validate_args(args, state, name='on, off or file') as args:
27
+ Config().set('debugs.complete', args)
28
+
29
+ return state
30
+
31
+ def completion(self, state: ReplState):
32
+ return super().completion(state, {f: None for f in ['on', 'off', 'file']})
33
+
34
+ def help(self, _: ReplState):
35
+ return f'{DebugCompletes.COMMAND} on|off|file\t turn auto complete debug on or off'
@@ -0,0 +1,35 @@
1
+ from adam.commands import validate_args
2
+ from adam.commands.command import Command
3
+ from adam.config import Config
4
+ from adam.repl_state import ReplState
5
+
6
+ class DebugTimings(Command):
7
+ COMMAND = 'debug timings'
8
+
9
+ # the singleton pattern
10
+ def __new__(cls, *args, **kwargs):
11
+ if not hasattr(cls, 'instance'): cls.instance = super(DebugTimings, 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 DebugTimings.COMMAND
20
+
21
+ def run(self, cmd: str, state: ReplState):
22
+ if not(args := self.args(cmd)):
23
+ return super().run(cmd, state)
24
+
25
+ with self.validate(args, state) as (args, state):
26
+ with validate_args(args, state, name='on, off or file') as args:
27
+ Config().set('debugs.timings', args)
28
+
29
+ return state
30
+
31
+ def completion(self, state: ReplState):
32
+ return super().completion(state, {f: None for f in ['on', 'off', 'file']})
33
+
34
+ def help(self, _: ReplState):
35
+ return f'{DebugTimings.COMMAND} on|off|file\t turn timing debug on or off'