kaqing 2.0.172__py3-none-any.whl → 2.0.174__py3-none-any.whl

adam/utils_k8s/k8s.py

@@ -2,7 +2,7 @@ from collections.abc import Callable
 import re
 import portforward
 
-from adam.commands.command import InvalidState
+from adam.commands.command import InvalidStateException
 from adam.repl_state import ReplState
 from adam.utils import log2
 from adam.utils_k8s.kube_context import KubeContext
@@ -24,14 +24,14 @@ class PortForwardHandler:
         if not self.svc_or_pod:
             log2('No service or pod found.')
 
-            raise InvalidState(state)
+            raise InvalidStateException(state)
 
         if KubeContext.in_cluster():
             svc_name = self.svc_or_pod(True)
             if not svc_name:
                 log2('No service found.')
 
-                raise InvalidState(state)
+                raise InvalidStateException(state)
 
             # cs-a526330d23-cs-a526330d23-default-sts-0 ->
             # curl http://cs-a526330d23-cs-a526330d23-reaper-service.stgawsscpsr.svc.cluster.local:8080
@@ -40,13 +40,13 @@ class PortForwardHandler:
                 svc = f'{groups[1]}{svc_name}.{state.namespace}.svc.cluster.local:{self.target_port}'
                 return (svc, svc)
             else:
-                raise InvalidState(state)
+                raise InvalidStateException(state)
         else:
             pod = self.svc_or_pod(False)
             if not pod:
                 log2('No pod found.')
 
-                raise InvalidState(state)
+                raise InvalidStateException(state)
 
             self.pod = pod
             self.forward_connection = portforward.forward(state.namespace, pod, self.local_port, self.target_port)

adam/utils_sqlite.py

@@ -32,10 +32,7 @@ class SQLiteConnectionHandler:
         self.conn = None
 
     def __enter__(self) -> sqlite3.Connection:
-        if self.keyspace:
-            self.conn = sqlite3.connect(f'{SQLite.local_db_dir()}/{self.database}_{self.keyspace}.db')
-        else:
-            self.conn = SQLite.connect(self.database)
+        self.conn = SQLite.connect(self.database, self.keyspace)
 
         self.conn.__enter__()
 
@@ -97,17 +94,20 @@ class SQLite:
          if conn:
                conn.close()
 
-    def connect(database: str):
+    def connect(database: str, keyspace: str = None):
         os.makedirs(SQLite.local_db_dir(), exist_ok=True)
 
-        conn = sqlite3.connect(f'{SQLite.local_db_dir()}/{database}.db')
-        with SQLite.cursor(conn) as cursor:
-            for d in SQLite.database_names(database):
-                if d != f'{database}.db':
-                    q = f"ATTACH DATABASE '{SQLite.local_db_dir()}/{d}' AS {SQLite.keyspace(d)}"
-                    cursor.execute(q)
-
-        return conn
+        if keyspace:
+            return sqlite3.connect(f'{SQLite.local_db_dir()}/{database}_{keyspace}.db')
+        else:
+            conn = sqlite3.connect(f'{SQLite.local_db_dir()}/{database}_root.db')
+            with SQLite.cursor(conn) as cursor:
+                for d in SQLite.database_names(database):
+                    if d != f'{database}.db':
+                        q = f"ATTACH DATABASE '{SQLite.local_db_dir()}/{d}' AS {SQLite.keyspace(d)}"
+                        cursor.execute(q)
+
+            return conn
 
     @functools.lru_cache()
     def column_names(tables: list[str] = [], database: str = None, function: str = 'audit', partition_cols_only = False):

adam/version.py

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

{kaqing-2.0.172.dist-info → kaqing-2.0.174.dist-info}/METADATA

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

{kaqing-2.0.172.dist-info → kaqing-2.0.174.dist-info}/RECORD

@@ -9,7 +9,7 @@ adam/embedded_apps.py,sha256=lKPx63mKzJbNmwz0rgL4gF76M9fDGxraYTtNAIGnZ_s,419
 adam/embedded_params.py,sha256=4q3Sc8tmlWyY5Yyfpqatl9urorJ_rW6KGxr_LCiQ5RI,5724
 adam/log.py,sha256=vcJ1Q8LLnt3NSXqpVcKjAI2OZE6KaD3PEi1kfu_D8qs,1156
 adam/pod_exec_result.py,sha256=85jy_5dS6TEEk5ijU0B62YfUycxmD3dG8mhIHFPipc8,1260
-adam/repl.py,sha256=iPOR_m8yAHP0DtXRdpfPjt4i8VL7QgAUPxsnx8X6XU8,7346
+adam/repl.py,sha256=iQ16-x2kS-6Sak4CjqUE41-Af-MApGjPka4qPzKkhQc,7459
 adam/repl_commands.py,sha256=vCKee-0smNPTK_SMegU-6GR4ShTh_nY_Ac3fFIdCMqc,6440
 adam/repl_session.py,sha256=uIogcvWBh7wd8QQ-p_JgLsyJ8YJgINw5vOd6JIsd7Vo,472
 adam/repl_state.py,sha256=ckyoj6j17cxQy0h1jVyxAgDvRRAzJeOS0SV01gzbEs0,15532
@@ -18,8 +18,8 @@ adam/utils_athena.py,sha256=vIcOPMtDrgA2SM2a2VpOVOjTJ2_Hss63lwlSgG3kuLM,5409
 adam/utils_audits.py,sha256=lZ2U1AcROxLB4652auG091QFTPs_YNzbtdXBiK6bY_Y,4063
 adam/utils_issues.py,sha256=XDiHDxQcYEvIxwL-fOeIXDRPpKinnggB5sSM3Yo2P-I,1458
 adam/utils_net.py,sha256=byEtNVr8iG9UaD7dM77dN2WEBClB7YNKult7LKFTCOc,428
-adam/utils_sqlite.py,sha256=1OproUQZV-LhoOZQcXXbNlzgOPiol_TUxmDN-Plrul4,4294
-adam/version.py,sha256=cet-_oZdu-OKTz9h9kpdsHU5eD495ddFI4EMHtgb2r4,140
+adam/utils_sqlite.py,sha256=kF1zTCJE_aIEhgbCDOIUH23XvnLufKpHiiyhkKqH6is,4340
+adam/version.py,sha256=htfdPUqia9L7LzWep3oJtMmkyNtuf9mKyjtJsHQfT7g,140
 adam/checks/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 adam/checks/check.py,sha256=Qopr3huYcMu2bzQgb99dEUYjFzkjKHRI76S6KA9b9Rk,702
 adam/checks/check_context.py,sha256=FEHkQ32jY1EDopQ2uYWqy9v7aEEX1orLpJWhopwAlh4,402
@@ -53,17 +53,17 @@ adam/columns/node_utils.py,sha256=ql2StXmASC7equuabZVZEQ_BPTu2KkS99gY_3JXpIvU,77
 adam/columns/pod_name.py,sha256=IYw0ZKA7Fb9LaGXENqzZTiTgL98tahwFRtfy0KkKh2Q,280
 adam/columns/volume_cassandra.py,sha256=9KRNOzjNYganI9avN6zaA5_-7yxD4rV-KNxro9CSUg4,753
 adam/columns/volume_root.py,sha256=29ujLoCAf9LO75u62LxEaPD58s6ihV-tcK17OeLSOM0,556
-adam/commands/__init__.py,sha256=CY2fDCMwXj2z-SIwQ2OJzwO5Jmht1Ddl_ZdgNza4nvI,882
-adam/commands/alter_tables.py,sha256=9oj4wpy8sTEwqsFfSwOaUs3ISsmbtRBWqMjCvPflW68,3430
-adam/commands/cat.py,sha256=LAC5kv3-VxMBDB33N4VCi68Dj8nzO8AFRzZR2mI_7a8,1582
-adam/commands/cd.py,sha256=C0ikeXY3voFqfW0Fr2sc8FORFulBJ1q1nTD2UiRSLLs,1290
-adam/commands/check.py,sha256=poaOg2n3crI0o5AAbXfZfFKtALRkTQRBJDRsLBTHqPk,2467
+adam/commands/__init__.py,sha256=6MGdr7yYIzW-igcxq44jZRVuKaoIhbaOOqdEiSyElbM,1173
+adam/commands/alter_tables.py,sha256=yNqIYDoZd2xZEl3hc8E6V1Br6_BEM3X7DCnXFGuR1pQ,3216
+adam/commands/cat.py,sha256=-Xx04XCYDVszR68IrJRnGbeu6RC1YRc_bnrY87xqkr4,1295
+adam/commands/cd.py,sha256=g6GGHqg72pfxjVL_s5KFne2GL8taikWRWgCEU1nDRqg,1338
+adam/commands/check.py,sha256=Hf2ukd2DDuWkfCpGYQFENZwQuPfuMqA_silmTVEfaQ0,2334
 adam/commands/cli_commands.py,sha256=TGnm2uxtuDRmkNNMlPOpYAXJkgDQEuhDeimBcwsEtC8,3574
 adam/commands/code.py,sha256=8SwSYuMx2QNFB3RUfVPfYVUAf_unEjADx4xvrstepJc,1906
-adam/commands/command.py,sha256=mKG5MRAD1Abjw4ajDytGzsBeFVQBvauHEjQGU41KDFk,6800
+adam/commands/command.py,sha256=AO9y4D00Ho8ThC8PQV53wUHm01w3ox3d3UXrooySwNw,7931
 adam/commands/command_helpers.py,sha256=leOJJK1UXczNTJHN9TGMCbIpUpmpreULvQ-TvnsYS7w,1134
 adam/commands/commands_utils.py,sha256=PjhJqqGdKNsUYry4ZU53tlR_NVwc0Fb8J-ZarN0uiD0,2856
-adam/commands/cp.py,sha256=MLxhq4t7DDJGT20u3LVEL_f7Bu8Ch5SxJ3FsOJyftsc,3113
+adam/commands/cp.py,sha256=qYntAdgs5aUg9-_EXT6ngz4dxyzcsfGR8vkrFrs6hag,2879
 adam/commands/exit.py,sha256=T7bCEShd6MXQ8rXak7vGZTvqEOejPOUPtbdfy7d8WWA,753
 adam/commands/help.py,sha256=v8VPGfnMC_PY_oqJUlwdAqIeJ9-EicvMEVT1RDKSmyI,1907
 adam/commands/intermediate_command.py,sha256=HBSV0tdocHtpe4g3BDagEp55SpDdMAtOVb_ZWVh8v78,1755
@@ -74,9 +74,9 @@ adam/commands/logs.py,sha256=CS4eh5JWzoXboYSXXpiE6CDsky1Jsu61Q1MZAqGjOB8,1142
 adam/commands/ls.py,sha256=b6EBzjXdlagOkpD_OM77I_n18ikGub5obtuerruA9yA,1298
 adam/commands/nodetool.py,sha256=QUgKWjtPCuoA2vsip7naaf-5XGqHVinhWjsxmIxd8AA,2161
 adam/commands/nodetool_commands.py,sha256=5IgWC3rmeDD1cgwqQjiiWzi-wJpJ3n_8pAzz_9phXuk,2635
-adam/commands/param_get.py,sha256=ExSONU1fqi4kE3GGmWTBoeuLgWjOERWBURhZuwKhNYE,1350
-adam/commands/param_set.py,sha256=brmBcncmI4qUXSkmtf4FJ5JSClYvDa-OB0RQlECXuSw,1308
-adam/commands/preview_table.py,sha256=0vwrF3R_inbffQrf_fBqPs8EOaVQZLk57KszUNl6LpI,1140
+adam/commands/param_get.py,sha256=LOCjvqbfSYGOhX4p4I6kj2H6PFvmDyUiaRm28AOxH4w,1408
+adam/commands/param_set.py,sha256=GMvd38jGBcjH1ElcpHmdKwuTvvGanm_CNZ8hJJRFXss,1358
+adam/commands/preview_table.py,sha256=TfSipKOrd1avGQ4Au09wh0FkMKKRuHYqfShHHbTbe-w,1165
 adam/commands/pwd.py,sha256=cdnmJv3-9uRtN-GFlqlktdWNKZLdKCTD55DAkSa065I,1441
 adam/commands/report.py,sha256=IESGEq6akUY2IznKUZV90_5s2gRM4npexFQH7Pa3rAI,2086
 adam/commands/restart.py,sha256=CUjHxdWznqgY2BdYcHHKH2eolj3cPeE5xbMjkIiyJ2A,2202
@@ -96,9 +96,9 @@ adam/commands/bash/bash.py,sha256=AvXehWWltrT9sXttpeGFhRoNO0PAjyNEiqnbxSiNpZI,11
 adam/commands/bash/bash_completer.py,sha256=IStUg5LLTqc3vjupw33WtXUG9zRQ37BctjDYGDyYEXU,3883
 adam/commands/bash/utils_bash.py,sha256=LmpTVPaM0aEMDqKPFnyEp5EidAyQQiiUkcvRjvOu-20,433
 adam/commands/cql/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adam/commands/cql/cql_completions.py,sha256=EDuzxpbJcZTwt-OqVvNAfo7o_arL1Ni0EpTWvDv-KCg,1452
+adam/commands/cql/cql_completions.py,sha256=Wr37WHWO4-zAYMRlG38GpfxFAIFHlmBUKSAHHsx7ftU,1465
 adam/commands/cql/cqlsh.py,sha256=ddb1JBdCs8sMHt8L-1gDHIir9hbjxP0gJKpKCBFPFuM,2381
-adam/commands/cql/utils_cql.py,sha256=EBl11r_Y8nJWk-g7X2Aowe8vY6nsaxcMBYLZlszsXr0,12877
+adam/commands/cql/utils_cql.py,sha256=H3XmGATVUfANhejwgPU1BoAcvRbq1Z0zpDeZXSp9g7M,12776
 adam/commands/deploy/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 adam/commands/deploy/code_start.py,sha256=0h8AIPjThC7uxa4HVkYMdC-7qCNXk0QXO1zDrzIOA4s,1335
 adam/commands/deploy/code_stop.py,sha256=R9zvNVD4NLUD3J6ktZ53Rsalzn-t0ZxEpeyOoiizfgU,1153
@@ -114,58 +114,58 @@ adam/commands/deploy/undeploy_pg_agent.py,sha256=t9l9XJiChiH61Z7Zc967b3zH_Tsj7Gr
 adam/commands/deploy/undeploy_pod.py,sha256=Pc5xpU4z5_CHrtz3cAzReT4Qxyks2kQ-uWSRUnAYiJ4,1907
 adam/commands/devices/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 adam/commands/devices/device.py,sha256=UPqHLQXar5a3ENQDaqimcX2DrX3OPILrbBTTfIB39F4,3336
-adam/commands/devices/device_app.py,sha256=w7PTv1pYdHCFovxvHpOfO3p1Vki1VTYpC3VqFobx610,6794
+adam/commands/devices/device_app.py,sha256=Wik6s_RB6vtCsQQvIEK6PelnAvutzQrzWewLiMvqCiw,6811
 adam/commands/devices/device_auit_log.py,sha256=PaniVrADdqQM0xAkXtT3w5MFkERKWH0wdHl3mXnlMpo,1587
-adam/commands/devices/device_cass.py,sha256=66I7xoCUzaYv0la6itl8JFO579qe03HlBVQdUEm-MT4,7042
-adam/commands/devices/device_export.py,sha256=Fo5TCNaEgChrRR6nzoHmKCW7vDxjBofq-OnDDDNYfP8,3062
-adam/commands/devices/device_postgres.py,sha256=Km3O3t9ChyhUr7gwXgKUm1abuUHmkiSkCukx3e88GYM,5774
+adam/commands/devices/device_cass.py,sha256=GlbjnEu92hKFOls-jLqMJ0y6y5WPk5KV9CQqvf72hDE,7059
+adam/commands/devices/device_export.py,sha256=yROeaNC1e0HkUuXlPy_v1MUi9CmoDIta8KNVohzuVBI,2946
+adam/commands/devices/device_postgres.py,sha256=Iriz4m3fKSbyk2CuLYIxZuNE60fGnkSLhHtzrEx8jNQ,5791
 adam/commands/devices/devices.py,sha256=-VV8nU-Qcebx2g2ftOywmgrn8wW7-teCpxFQjU9wBR4,965
 adam/commands/export/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adam/commands/export/clean_up_all_export_sessions.py,sha256=VNe9pVcF8xrkFfQ_XO2qjueuJiJcLCkP1XV1kS7WYwg,1196
-adam/commands/export/clean_up_export_sessions.py,sha256=eLnLUFs98HcdYu_LrU7-Mj0B64Mc79GaYhpcNB0vuXc,1672
-adam/commands/export/drop_export_database.py,sha256=iKXvP06dk7QP6sEE4WUyDoIkRuqm8nfEkPja6AP0_kE,1678
-adam/commands/export/drop_export_databases.py,sha256=sLNNygGw6Ltfe9t4i_DJEHckMgFy3qCqLYssA7aSdow,1270
-adam/commands/export/export.py,sha256=_nzqXntCZweQMoxeseB81Wpxsx-947W6DR1Fj7NB7WQ,2112
-adam/commands/export/export_databases.py,sha256=d4_9pR8Uo0N8qhK5FcqZ5K8dBjy-bu8th-w_D6T86Jc,5700
-adam/commands/export/export_handlers.py,sha256=1pBf-XxRyyYmGNrt_a-UaWvtPQ25MHwjP6sHLJC7dx4,2354
-adam/commands/export/export_select.py,sha256=CsFXJkeLMPgmmk09E4azgr2o5969atre8R78tWv64kQ,3291
-adam/commands/export/export_select_x.py,sha256=_wGW_PRBZMhFWGHGBXaakt8ERfsBRBpDGqLRBxbKxMY,1997
-adam/commands/export/export_use.py,sha256=1CR9L0-Ao7wlrXO_00sZKj9kKRq7j9Pudfdx-yOuUXo,1573
-adam/commands/export/exporter.py,sha256=-I1wjXHsT2dJh_AJqPQXXBASZxWQNTwq2hPVEFMddHg,16206
-adam/commands/export/import_session.py,sha256=8qZoOKHSETOXHFU_5se7h93QRWht9foxizymIhdnIFA,1375
-adam/commands/export/importer.py,sha256=TttTWE2VK4xjKFnsQyaJ0oRVsUQGdZW93kQmdOKKDI0,2292
-adam/commands/export/importer_athena.py,sha256=5T1BFzW_4Qk6MTZDTzdLPWonod_X7vCTNmSnXGiu7O8,3198
-adam/commands/export/importer_sqlite.py,sha256=4_JGcKAQ2k-5r8Zz_sLqMLEtmDiZogPlRg4vJAOG1_k,1667
-adam/commands/export/show_column_counts.py,sha256=nj77CLDm9qSvNiuMKsldR2faJPEy8WBL9wyghrC5hz8,1811
-adam/commands/export/show_export_databases.py,sha256=LrzSODX2G9KCu9BUMgpz353MaE7NuIreEmeZNTlW0Lk,1169
-adam/commands/export/show_export_session.py,sha256=ErYgzIXe0AKFKHX_XcgVVNURejGruYNHTIWOJB7QV-o,1634
-adam/commands/export/show_export_sessions.py,sha256=fW9vcnekTZJvPM4oNAiXXxoh9phGiSOUK6cDKM_Y_Ic,1580
-adam/commands/export/utils_export.py,sha256=lBcmqJJiWqf9BbeT6cKGu9qT3zPBuKIAWrqendZibEc,11247
+adam/commands/export/clean_up_all_export_sessions.py,sha256=yWDC9ts2HYachcfffqhbXwAIi5xwvsE3m7XNNFk-xLQ,1221
+adam/commands/export/clean_up_export_sessions.py,sha256=ERLcGgzY5j3GoqkGJy-THYDgJLANAlpWnCnZ4vkH-y0,1600
+adam/commands/export/drop_export_database.py,sha256=3S6yj0OuMknPZ1MpPad7eCm1V6BdDn6enHcjRC_fjek,1232
+adam/commands/export/drop_export_databases.py,sha256=WzOm4Ykf1O_ZRjF8oZn_PaKTuIzxFeRJhqWfsENCstg,1095
+adam/commands/export/export.py,sha256=l-3dSKXVdAteicfQvkLmELRJ42LBmCAqhOqL0uthrJA,2105
+adam/commands/export/export_databases.py,sha256=aZogr6UZsSjvqTkPemF-pdscMZMDkwx3JXOAspmlmS4,7626
+adam/commands/export/export_select.py,sha256=wVkw7Th3Abs-zRJO0Rk310GUl8nEOyWLbXNLaBZ3Mq8,3113
+adam/commands/export/export_select_x.py,sha256=WuSkuhB4yfz0IVt3-Dd42GG7FTWoNzBhVsfieEN51W0,2020
+adam/commands/export/export_sessions.py,sha256=vaT3o8K9uOXvJ5jzTxZONm6QcEAbhgCNqdK0-GXiY9s,4623
+adam/commands/export/export_use.py,sha256=j7eNmlnNtyTrk3AyAZzA8kkU2C7e1pmAmNFwIMdDSA0,1354
+adam/commands/export/exporter.py,sha256=qj0b6gXCmj7k0cCQFShHde54_l4n0zVhPCHkO559034,13858
+adam/commands/export/import_session.py,sha256=tb_ybcQbUN4kbc7MuUqsg75NY9tXzXdJSh1zcXs0ILQ,1393
+adam/commands/export/importer.py,sha256=V4EbanV-rEmbqP43u8_07YY98A7wdbwjo1j7BMNzfOU,2450
+adam/commands/export/importer_athena.py,sha256=zNKWFi90wACmOgVEfzMyEt6V4yF1Hl7uo6wYOwIh84k,3373
+adam/commands/export/importer_sqlite.py,sha256=0hvo3MoIs_J9vjsO0iCMRcYOzR7GpRx9Law7CVP5arQ,1720
+adam/commands/export/show_column_counts.py,sha256=zsECHmHgwFATmsyshLbDgG2uBgAosGEwPmwhB_RJF04,1638
+adam/commands/export/show_export_databases.py,sha256=G1Ue-c_fKs_xEesCZV7gVdPY6w-AbJ_M6jWSw-hf73Q,1229
+adam/commands/export/show_export_session.py,sha256=Gzrj87PZDwevACL17P7dH2a91QYkSwFezguAfMdtv0Q,1481
+adam/commands/export/show_export_sessions.py,sha256=xyuDu8oUkTdMKfBHhAUOcwIQdZWcl0ePkSM27wuomIA,1599
+adam/commands/export/utils_export.py,sha256=5TUHfkt4Lx5iO6OizSBxiJxyNgAVDAslUaVMX2dKkF0,11311
 adam/commands/medusa/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 adam/commands/medusa/medusa.py,sha256=Ey2kJpoKqPFP6m623B3ciMoLur5SVcHEBYsvW7G7aAY,958
-adam/commands/medusa/medusa_backup.py,sha256=RmiND5P9l4aoDBur9kW2ICvFSA-R4pmL7kDg9E7iKo0,1821
-adam/commands/medusa/medusa_restore.py,sha256=-lXdklHSGeRNIHHr_EQ6T8NSOMTSKTT9vCvIr6H1yHQ,3301
+adam/commands/medusa/medusa_backup.py,sha256=1yWMGEipdZ_2vALAYdmyPjZWrX6kY7E0OuPxiBv0tfw,1759
+adam/commands/medusa/medusa_restore.py,sha256=_v5-d_yAjvpTivpROS51IV95Jj0WE1qZvDkY_kczQrw,3271
 adam/commands/medusa/medusa_show_backupjobs.py,sha256=rAlhpXH2mhY-1ScSGm32llLh5_fNbHS2wLR0pdi6LBw,1761
 adam/commands/medusa/medusa_show_restorejobs.py,sha256=KxPgUOHGqik4G7B4GnsX9Ogr1yDIR5k1lf4fkJXPjwM,1556
 adam/commands/postgres/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-adam/commands/postgres/postgres.py,sha256=pjnYxCHaYRNP7iPzKU_EGP4isjHFRie8_-tA1Y4H3dU,3661
+adam/commands/postgres/postgres.py,sha256=1QCgDe_8b6-z02XoUpk9wbQ2j5ZMi6rRiZ_bXElE6K8,3478
 adam/commands/postgres/postgres_context.py,sha256=y_3n5kWBT_XFFCdPS8qYm5_2v4bJ-PuwH2EO_-IZUiQ,10784
 adam/commands/postgres/postgres_ls.py,sha256=yT-V1mGnvImaIr4s75FUsV5klcrOB7HHJwWbY6h1Kpk,1111
 adam/commands/postgres/postgres_preview.py,sha256=Bmw3YC98tFdjUx2CRHOakcvLHCOLtEiKxvNDjlDoRo0,1174
 adam/commands/postgres/psql_completions.py,sha256=LVw7LvKSRh61Dfqxh_FZp7wmIeejHjK_V6QGwUrePfw,336
-adam/commands/postgres/utils_postgres.py,sha256=SeAtjxF2dBiu9G4nTjMcgDXIMFrFafvj48jbG1Lkg3s,2025
+adam/commands/postgres/utils_postgres.py,sha256=omriBzuW0NQ16IF1Hy6FeVLvfGSWJeTMk34jwigDxrw,2108
 adam/commands/reaper/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 adam/commands/reaper/reaper.py,sha256=Xrfjvqtn5txnaMwofBb77yEsmSYLO_-Bi43ABq2ZmaQ,1398
 adam/commands/reaper/reaper_forward.py,sha256=Su3tFrx2MTTXvaPC0EsmCnTKDuUjziyED583IX2UJ84,3326
 adam/commands/reaper/reaper_forward_session.py,sha256=CfKMYQPDZYQBtYQjxcPs7ns5e8AGZj9Q0LD8dOxvLl0,160
 adam/commands/reaper/reaper_forward_stop.py,sha256=a8i6Zp4rYKFt6qC7iqpf57gDZTMAtZy1H1PXq5tTOHM,1375
 adam/commands/reaper/reaper_restart.py,sha256=Ms3OWhV6pqEtmQiupvKO9ArTVBPlniPtg4BwPCgaFq8,1173
-adam/commands/reaper/reaper_run_abort.py,sha256=rqargemlEXoeL6e1OYyroxGanGopi0rgk-RYTSFejJU,1363
+adam/commands/reaper/reaper_run_abort.py,sha256=f3D2EhtiPiLXBOx0vY-XmBivDrgLbppb4dgnStucNVk,1276
 adam/commands/reaper/reaper_runs.py,sha256=hXfGdtlRyf6jOyoILGFEec_lL4RWn_Eqiw69Zge9SLo,3065
 adam/commands/reaper/reaper_runs_abort.py,sha256=ykY8D5fcZwqMRebM2JxFClVksjeDIBTIQe5nSk5oyp0,2058
-adam/commands/reaper/reaper_schedule_activate.py,sha256=WHr1avsnK9_0YUA42jMF6gsc-d9UWszAVurCdbVotXY,1496
-adam/commands/reaper/reaper_schedule_start.py,sha256=9SQoQRPWUIZQZlfDy08LOwHj0N-jXoLFWwEynI9QTfc,1483
-adam/commands/reaper/reaper_schedule_stop.py,sha256=R_oktRyc2oqZg7-yvYLgshdSRyY1rCQn40yS7-z83UI,1479
+adam/commands/reaper/reaper_schedule_activate.py,sha256=PELW78_kblgGegyRFXtIOMf3eIHxJt2D3jNEYITwHFw,1485
+adam/commands/reaper/reaper_schedule_start.py,sha256=VjZ2E-pm8QpnVuMYw53VhfXUgfORC-DjJoDiERKqxYM,1472
+adam/commands/reaper/reaper_schedule_stop.py,sha256=FWaUF3j8R6v57bNQm2np96QwsvBUS3Ee7TDDl5avbRM,1464
 adam/commands/reaper/reaper_schedules.py,sha256=vbvt0JojQecWCo5i4M-3HenWSnJU68F1MAoWcPhp8bo,1071
 adam/commands/reaper/reaper_status.py,sha256=IuRbQwlZB4pDZu4iuPk5TXdTqvzBTMtAl68tHiiPZM0,1801
 adam/commands/reaper/utils_reaper.py,sha256=jQIjpBfQxCPfD_63RpbRZ_tbW2335mBaYiBHalhijSc,7541
@@ -211,7 +211,7 @@ adam/utils_k8s/custom_resources.py,sha256=95zVZchFtFWGZ_P-jLaSMl6WYTObn7PqeTLxZ_
 adam/utils_k8s/deployment.py,sha256=SLhnMm5GMXwEldj2OupSFBUsvNjynwSNrv5tIDvLMrc,2921
 adam/utils_k8s/ingresses.py,sha256=q8nJTIoxa_dVrGMl1HS5yTS_11jLNBAnWlc2ZishMEE,3458
 adam/utils_k8s/jobs.py,sha256=HbUBNXcKcuXaNzc8NmqR_Sq35yHQ_qeQtiQFFM-J2QY,2534
-adam/utils_k8s/k8s.py,sha256=eodkX7ZnqArniiAOHAQsqgBPYCHZj-Silv48g1rcGg4,3088
+adam/utils_k8s/k8s.py,sha256=hq3kPTdDB9Fyny64mFLepOgtqpEEVw6wm6IveWzBRfY,3133
 adam/utils_k8s/kube_context.py,sha256=xJF_72vUJu-X9MpIYzOIfnj7KEWU7a_sLBR-H3994Y0,3311
 adam/utils_k8s/pods.py,sha256=Y3FnAxnmNJwpvUaIkXf2Wkdq2iSpw5AU2aJMKRLf9AQ,10865
 adam/utils_k8s/secrets.py,sha256=xnqRSumCIcPaASDcErzxjx1BCE-qJkPoG5wIP7YHzIo,2430
@@ -223,8 +223,8 @@ adam/utils_repl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 adam/utils_repl/automata_completer.py,sha256=LJP_2WHHR7AtjX00MJ59VGQEL3t0XS-qYnDmMaZe-Tk,1641
 adam/utils_repl/repl_completer.py,sha256=lsu0vWFdTs8SSSY9gxHN-fQSap-Wqz1QhMF4FqGf7qA,1722
 adam/utils_repl/state_machine.py,sha256=kO4_oSi_M53f3QQjINzzb2VFptjbnqX3KRC0G8LqqeA,5426
-kaqing-2.0.172.dist-info/METADATA,sha256=Kh823njRV9ghwLLznWxlQQ4zdUf4pyNHpsike8ZSjmo,133
-kaqing-2.0.172.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-kaqing-2.0.172.dist-info/entry_points.txt,sha256=SkzhuQJUWsXOzHeZ5TgQ2c3_g53UGK23zzJU_JTZOZI,39
-kaqing-2.0.172.dist-info/top_level.txt,sha256=8_2PZkwBb-xDcnc8a2rAbQeJhXKXskc7zTP7pSPa1fw,5
-kaqing-2.0.172.dist-info/RECORD,,
+kaqing-2.0.174.dist-info/METADATA,sha256=RW9iMrX4oWDyFahuaYLnN--UVcZfqK3SP-x_1xgVR60,133
+kaqing-2.0.174.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+kaqing-2.0.174.dist-info/entry_points.txt,sha256=SkzhuQJUWsXOzHeZ5TgQ2c3_g53UGK23zzJU_JTZOZI,39
+kaqing-2.0.174.dist-info/top_level.txt,sha256=8_2PZkwBb-xDcnc8a2rAbQeJhXKXskc7zTP7pSPa1fw,5
+kaqing-2.0.174.dist-info/RECORD,,

adam/commands/export/export_handlers.py

@@ -1,71 +0,0 @@
-from adam.commands.command import Command
-from adam.commands.export.export_databases import ExportDatabases
-from adam.commands.export.exporter import Exporter
-from adam.commands.export.utils_export import ExportSpec, state_with_pod
-from adam.repl_state import ReplState
-from adam.utils import log, log2
-
-class ExportService:
-    def __init__(self, handler: 'ExporterHandler'):
-        self.handler = handler
-
-    def export(self, args: list[str], export_only=False):
-        state = self.handler.state
-        export_session = state.export_session
-        spec: ExportSpec = None
-        try:
-            with state_with_pod(state) as state:
-                # --export-only for testing only
-                statuses, spec = Exporter.export_tables(args, state, export_only=export_only)
-                if not statuses:
-                    return state
-
-                Exporter.clear_export_session_cache()
-
-                if spec.importer == 'csv' or export_only:
-                    ExportDatabases.disply_export_session(state.sts, state.pod, state.namespace, spec.session)
-                else:
-                    log()
-                    ExportDatabases.display_export_db(state.export_session)
-        finally:
-            # if exporting to csv, do not bind the new session id to repl state
-            if spec and spec.importer == 'csv':
-                state.export_session = export_session
-
-        return state
-
-    def import_sesion(self, args: list[str]):
-        state = self.handler.state
-
-        if not args:
-            if state.in_repl:
-                log2('Specify export session name.')
-            else:
-                log2('* Export session name is missing.')
-
-                Command.display_help()
-
-            return 'command-missing'
-
-        with state_with_pod(state) as state:
-            tables, _ = Exporter.import_session(args, state)
-            if tables:
-                Exporter.clear_export_session_cache()
-
-                log()
-                ExportDatabases.display_export_db(state.export_session)
-
-        return state
-
-class ExporterHandler:
-    def __init__(self, state: ReplState):
-        self.state = state
-
-    def __enter__(self):
-        return ExportService(self)
-
-    def __exit__(self, exc_type, exc_val, exc_tb):
-        return False
-
-def export(state: ReplState):
-    return ExporterHandler(state)