cgcsdk 1.0.7__py3-none-any.whl → 1.0.10__py3-none-any.whl

{cgcsdk-1.0.7.dist-info → cgcsdk-1.0.10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cgcsdk
-Version: 1.0.7
+Version: 1.0.10
 Summary: Comtegra GPU Cloud REST API client
 Author: Comtegra AI Team
 Author-email: ai@comtegra.pl
@@ -30,8 +30,6 @@ Requires-Dist: statsd
 Requires-Dist: requests
 Requires-Dist: setuptools
 Requires-Dist: colorama
-Requires-Dist: redis
-Requires-Dist: pymongo
 Requires-Dist: psycopg2-binary
 
 # Comtegra GPU Cloud CLI Client
@@ -54,10 +52,8 @@ For now, we provide the ability to spawn compute resources like:
 Notebooks are equiped with all CUDA libraries and GPU drivers which enables the usage of GPU for accelerated computations.
 Apart from compute resources, we provide the database engines accessible from within your namespace:
 
-1. [MongoDB](https://www.mongodb.org/)
-2. [PostgreSQL](https://www.postgresql.org/)
-3. [Redis](https://redis.io/)
-4. [Weaviate](https://weaviate.io/)
+1. [PostgreSQL](https://www.postgresql.org/)
+2. [Weaviate](https://weaviate.io/)
 
 More are coming!
 Please follow instructions to get started.

{cgcsdk-1.0.7.dist-info → cgcsdk-1.0.10.dist-info}/RECORD

@@ -1,54 +1,62 @@
-cgc/.env,sha256=0RLfGys6OqABCJ5Ak-tj_AUbvaLN1A6Erd4O-fnAUXE,209
-cgc/CHANGELOG.md,sha256=unRU_finGpvnpbv2U4hFa1E_IJ_g0X6uoGW4jTmZ3P8,7808
+cgc/.env,sha256=I-5qrmy3GL9dhkX_k4UN59twXdPL_Tu_Emwf-ZU6-hg,210
+cgc/CHANGELOG.md,sha256=3eTeQjVZkrfGb5Ld-YUIfM9dios7vOZp3cc2xVILfFQ,8532
 cgc/__init__.py,sha256=d03Xv8Pw4ktNyUHfmicP6XfxYPXnVYLaCZPyUlg_RNQ,326
-cgc/cgc.py,sha256=wcngqkKT1AxkKcc-F63cBrfx9mxLKfDM8H8ciuIFnyw,1493
+cgc/cgc.py,sha256=SaZIGvZpA25OBx2KnQlZVjWNY2ejCyPU5KEkwqh7sh0,1655
 cgc/config.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cgc/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cgc/commands/cgc_cmd.py,sha256=QwqjZl9FmAMOvuUQMJm-wOV88OxbdwK5UIS0zkg-mAo,4956
 cgc/commands/cgc_cmd_responses.py,sha256=wO9Hf5_4jaw1ZRhQvh9Za0S_vhc22t-0_CoEGz5ndNE,2245
 cgc/commands/cgc_helpers.py,sha256=ngArFjVw-8P_2g7J8k3b9xgDxfJw7JeaOtkhTySMSGU,1174
-cgc/commands/exceptions.py,sha256=l3Sms3D2fxSpLQQQEYeLWxO3to82myTQ0VFgFYdQdLU,45
-cgc/commands/auth/__init__.py,sha256=K8HkHHotMnK7SQRAst5rx_wprHEphPo_w2KToEymjAY,399
-cgc/commands/auth/auth_cmd.py,sha256=jfijV-C0uOnpj5LcOgsW7Sn5JiemZ-nuQRQUEjEvaEk,3856
-cgc/commands/auth/auth_responses.py,sha256=kf5Btt7-9oKH8AGL7_zcjDL4sXWmqmt3pV37S1GeqBQ,1839
-cgc/commands/auth/auth_utils.py,sha256=R-iwoZbMWTpPJ3O7tp3cU6l-csl-ULWoazE7t3L5hmU,5359
+cgc/commands/cgc_models.py,sha256=T4eCMO5AO2kZpIRqpZ7ZfgO9gxCo_MLlY33msCzrLvA,182
+cgc/commands/exceptions.py,sha256=kVCWEdcqmkLO5QVIINdEXQw29_glyX2C11ZMFgT7b8E,155
+cgc/commands/auth/__init__.py,sha256=JnqNezSEUEVVPQIymya4llXr8LIug2vymQSAGBr4Ovg,397
+cgc/commands/auth/auth_cmd.py,sha256=FiRM16RGtP1zoJ9InbFAa-3W-EkJQbzBDxF0AmZa1JA,4196
+cgc/commands/auth/auth_responses.py,sha256=oZ1aC7tyrT43imF2iMaL0u1jTJzuGLEmCN8RBeMxpf4,1959
+cgc/commands/auth/auth_utils.py,sha256=-diCsKuF5UcQZoq962RuLktrG6sRmipJrDkm9Sf3wbg,5507
 cgc/commands/billing/__init__.py,sha256=0arQm0R3Ouw7tXPooJsvWd_pGeHhzaVwQCbWMKQPT9A,753
 cgc/commands/billing/billing_cmd.py,sha256=T1E5WW5Z-RlzddaotimTuvLvIbGihNpStIBETnlUznM,4140
 cgc/commands/billing/billing_responses.py,sha256=HAD5N-Odx3Jz1OmhO4v66rHoXpTYIOGlXDsrs0da9dk,1949
 cgc/commands/billing/billing_utils.py,sha256=zXLbBBcWeOgur-r0OKiIjaKeaxMNxASXWzCTeTsyC6o,4711
 cgc/commands/compute/__init__.py,sha256=lCdLfZ0ECSHtXEUSwq5YRHH85yXHchSsz8ZJvmClPtI,239
-cgc/commands/compute/compute_cmd.py,sha256=jgO61ULXIAuRn1LcFn0VU0X2-oLvBQB-0_OYMzXau0w,15490
-cgc/commands/compute/compute_models.py,sha256=Am9UdlTGZRGIEOw4oHvFHwedM1IMunOTCVuhZ8nMiRc,1173
-cgc/commands/compute/compute_responses.py,sha256=eOmcllyOqPYqN0kSUzSpuC2S1rFmkkawgc_F-0-LSIQ,5807
-cgc/commands/compute/compute_utills.py,sha256=XXnd_EuF9vCmW14r0ThlN98ScKSx3KAeqFbWy2g2mlk,8812
+cgc/commands/compute/compute_cmd.py,sha256=eXKeoOV3DsOypuxzKMa3MpITOKKbF2cZjy30iofHHG8,15489
+cgc/commands/compute/compute_models.py,sha256=yse6mCEhhE2RdWIO7EHx4r5Y-350gWWy4UWFsdltNNg,778
+cgc/commands/compute/compute_responses.py,sha256=DnRIZC_F1ZE887fMOXXYGYuYmB9n5x63sf8Ga_TL4Ww,5806
+cgc/commands/compute/compute_utils.py,sha256=XXnd_EuF9vCmW14r0ThlN98ScKSx3KAeqFbWy2g2mlk,8812
 cgc/commands/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cgc/commands/db/db_cmd.py,sha256=r0iwQ9oIQLcozshVyaHBxQMijYLrcfKvm0bXr9jGhek,3309
+cgc/commands/db/db_cmd.py,sha256=4iamsHTy68YjrKztyRGz2PqQgo54Q_rURgm2zpp_hL8,3512
+cgc/commands/db/db_models.py,sha256=oJUAF23nb_r8fSyhLHMJrRbRzLY_vpJQu6Vy4WRa0cg,1036
 cgc/commands/debug/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cgc/commands/debug/debug_cmd.py,sha256=kuAuh5YOqzGFjoiYZwfM9FJ1z5OeSpC0JAIUEzS83lM,412
 cgc/commands/jobs/__init__.py,sha256=E-438wgIzlnGmXs5jgmWAhJ1KNV6UXF2gz8SXu3UxA0,231
-cgc/commands/jobs/job_utils.py,sha256=BMwd7GA-WxC-NfpiziCQkG1TCNP98S3DWNpYmJbFCLY,5952
+cgc/commands/jobs/job_utils.py,sha256=_WffuTWYZamaXmrkRQIRmQzFsrvNbWzSwXlVLqLbmis,6194
 cgc/commands/jobs/jobs_cmd.py,sha256=Q-orK6B9Zk1zAf8sOM6QqF9Eeu092P-UEg4GRA-zX-s,6555
-cgc/commands/jobs/jobs_responses.py,sha256=XYFTWHKvHKeZ-aOWlO0MI0iZnVm77SKqg9Y3N2CJaZU,1771
+cgc/commands/jobs/jobs_responses.py,sha256=QXFXA4zwQOo5Gvq5rEc7J_cxxsYqkdU19X9MCcZetUM,1771
 cgc/commands/resource/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cgc/commands/resource/resource_cmd.py,sha256=ALROaJPRK5BB8R_oEC7JzOl3J8umMmhC_HJuwGFaw_M,4053
+cgc/commands/resource/resource_cmd.py,sha256=y2R6ZF0jFhTppD9JUpwLIcEdFRXg82Xw_5yrNIfG4mI,4089
 cgc/commands/resource/resource_responses.py,sha256=sES7mAi_Cv5B6Z3I_6eUOqVwOr2HgMO45cz8MiNZetQ,197
+cgc/commands/user/__init__.py,sha256=Rhaa2SDXQsJVJnTX0oFyZ90tAuiAUBeRy3Bd415S4os,382
+cgc/commands/user/keys_cmd.py,sha256=dfHnhHmEwRH1xl_wRNQcpCT-THJTBYPHMTJSI2AsDXE,4509
+cgc/commands/user/keys_models.py,sha256=IYojnAu3aTGJ7qHIYvZy-0YFzXRmueZNplS4RqdfJfw,378
+cgc/commands/user/keys_responses.py,sha256=TBHFWAVN1kEgP6sudbMwfpiZOcfTr0a0SD-f-Wn-ed8,1389
+cgc/commands/user/keys_utils.py,sha256=w1TcMTlKJNxXjnj-XsiUU8N__-l1TsWqfPPjGpRqtI0,2285
+cgc/commands/user/secret_cmd.py,sha256=_b9ewJlEE4gMCrgdyycpO5MOdJvibeHObAiPE-mWIgA,4165
+cgc/commands/user/secret_models.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cgc/commands/user/secret_responses.py,sha256=sg7HP5a0LnUvlo9w8oOcYASwI2cPeRnecjTVMa5yk7Y,1600
+cgc/commands/user/secret_utils.py,sha256=70dn937rzecQ0eE-8zb9fLnBDWtjWjdQY1LJvHv3NzM,1623
 cgc/commands/volume/__init__.py,sha256=Ou3kyb72aaXkrVCfQCVdniA65R2xHsRFgebooG1gflA,384
 cgc/commands/volume/data_model.py,sha256=meprXdaXLo3mMTZta1ks1-BJ7G0rO16qi_ycH-sXJpY,1295
 cgc/commands/volume/volume_cmd.py,sha256=5bd0vq5--Y7IfniO33tGBYH0Z67Le2s_AMTiwJcPBA4,7673
 cgc/commands/volume/volume_responses.py,sha256=WefUohXxXZ9Znfc6P2XjoAM2RlA19hMKcVaW-xG9HWs,3199
 cgc/commands/volume/volume_utils.py,sha256=n6s0FgpsYyxFMp_JdMRCzRi5Ar3_Svg9JDvWSdX4lhk,1919
-cgc/sdk/__init__.py,sha256=qR4-ySTsunbZBRqtUGa6DAzW-E8fXf1wlDSwIifLOeU,297
+cgc/sdk/__init__.py,sha256=m8uAD2e_ADbHC4_kaOpLrUk_bHy7wC56rPjhcttclCs,177
 cgc/sdk/exceptions.py,sha256=99XIzDO6LYKjex715troH-MkGUN7hi2Bit4KHfSHDis,214
-cgc/sdk/handlers.py,sha256=ECCHNe1pErsXFlmwHewsWRvYqzAZ5j5TrSqwernpLJk,868
-cgc/sdk/job.py,sha256=SclaqhY0rDACT_yg9e9AiYTsmXglIt6iTharwUPSAa4,5332
-cgc/sdk/mongodb.py,sha256=TJ2XU7nilNRXLOIpQQPrRiVxHN2TaVM5QOSuMRtNDVs,7221
+cgc/sdk/job.py,sha256=q9Vsarc3rKzurM4AOmbQDsQUVdyRqx0UzJVe_uO8xCU,5318
 cgc/sdk/postgresql.py,sha256=ziXaMMwjSF3k1OAID3F9npqWVxreQaoZ8wn7X8x1FZw,1637
-cgc/sdk/redis.py,sha256=W5wS9Sgyv4098yzWAwG7qEk4HEDwscE3JmWgPC3NCzc,2844
-cgc/sdk/resource.py,sha256=Wodt8pe15zg1uuclfV4Qxntffph70ULhNig_gOgMKz8,13544
+cgc/sdk/resource.py,sha256=w8SVRqrx0Mj1FS91Bt0oaMAfC4CDEcomgqzFCNgGaPc,13537
 cgc/telemetry/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cgc/telemetry/basic.py,sha256=XagCcyH4QSEPmfiQ1WCjqXslnJO6IaJCY0AMPySd5rc,3175
 cgc/tests/__init__.py,sha256=8aI3MVpkzaj0_UX02kZCtY5vmGO0rnq0mw2H04-OHf8,102743
-cgc/tests/responses_tests.py,sha256=siKJkxt1C-x2wzc1iOrgxjoBIvUY_n989HjxjYNr-fM,9181
+cgc/tests/responses_tests.py,sha256=TtTwMZFo-nqJ-gHRmObYZsVKFdaEpT0JovMynmcxxfQ,9180
 cgc/tests/desired_responses/test_billing_invoice.txt,sha256=KR5m2gamn_bgfBdBmWDH2sPRJIPOw1u8kdH-gYE8jow,1579
 cgc/tests/desired_responses/test_billing_status.txt,sha256=2KSUixFOrhXI5ON6qtsIUmzFye5LBZB1xneY8ST0MqE,2381
 cgc/tests/desired_responses/test_billing_stop_events_compute.txt,sha256=nHdixgLhAXDMeoDvjk9Mwv6b0JIAsW8j7Z6SS5scjSs,234
@@ -57,26 +65,26 @@ cgc/tests/desired_responses/test_compute_list.txt,sha256=3CDAuILJdwBmX6-GLj3zDbn
 cgc/tests/desired_responses/test_compute_list_no_labels.txt,sha256=-OeQIaEHHsHZ81tCOI5j6VQYUMlj8VYcyyOU_DhPOpU,155
 cgc/tests/desired_responses/test_tabulate_response.txt,sha256=beNyCTS9fwrHn4ueEOVk2BpOeSYZWumIa3H5EUGnW1I,789
 cgc/tests/desired_responses/test_volume_list.txt,sha256=vYB1p50BBHD801q7LUdDc_aca4ezQ8CFLWw7I-b4Uao,309
-cgc/utils/__init__.py,sha256=l9JF-WnvmhlolxbDKlJPsxquZ-fjtvv7wKvn2zpu5IM,3466
+cgc/utils/__init__.py,sha256=o3YYCh02WiAvmHw-8Hy3FJA4LjGWwPkaLTur6tEvsbE,3669
 cgc/utils/click_group.py,sha256=Scfw8eMIyt2dE1ezUq2JuiI-E_LklqXQXJEr7L-EG6A,633
-cgc/utils/config_utils.py,sha256=VFTrcN9QeOByIobhh48TfAm-BWON4PN8zX0H8PdUNTU,2729
-cgc/utils/custom_exceptions.py,sha256=eVduUpAwUqSCi1hyqal36xj-kWVcy9AeGdhbs1praYI,2432
-cgc/utils/message_utils.py,sha256=jdUHtR2-gEvyxYu1T0OY-a6cBz5N8TatRaxKNuNcJtU,1766
-cgc/utils/prepare_headers.py,sha256=xNDins83jdMre80s3orsR3Xu0jPUd82CHppvLq2tbeA,2561
-cgc/utils/requests_helper.py,sha256=ghn8LTxWqfRvy7BXQdxD4VHX8b-ypHkbnFXY05ig7_A,2050
+cgc/utils/config_utils.py,sha256=NlPs334y6BSNGoYbFjrB5cKc9trTWIXOn8UN6O4ztNU,2884
+cgc/utils/custom_exceptions.py,sha256=qvHdzaunZswZgN96iOHZIfLjehlJ79mcjqoMoW-tqEM,2628
+cgc/utils/message_utils.py,sha256=FAiUC-0zJiMhfPQAQC0ki1ZUs1vI_QqHwLmfoCDbLeU,1790
+cgc/utils/prepare_headers.py,sha256=67YMPogNoGbKnHQmGFc3iJCbsDZbRVjAIe3kBY-Dq98,2838
+cgc/utils/requests_helper.py,sha256=qhSTfOb0o-luHLAeIuiHTOl0R_ykkdQldQZz3D6Sr6s,2062
 cgc/utils/response_utils.py,sha256=9vJqAt2UFJ1n-oesFPe6CB_ooGoStjl-twY_31Jt4_I,7374
 cgc/utils/update.py,sha256=AsQwhcBqsjgNPKn6AN6ojt0Ew5otvJXyshys6bjr7DQ,413
 cgc/utils/version_control.py,sha256=VsqNzNYWDvU3VsoPuYIaZV7Img-K2_DmEOHd7rZlRxk,3267
 cgc/utils/consts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cgc/utils/consts/env_consts.py,sha256=sHO1vzWZALiXB4Ej2E7S5v0i-DfTwHBIIC1fIP_MocI,1286
+cgc/utils/consts/env_consts.py,sha256=8huPDBkbNEXoPHskFhYwiRi6TTw0_e9ylClpgh1NF2E,1290
 cgc/utils/consts/message_consts.py,sha256=8CIe3N_HL6Pj-gSArkPkpegsvm-QMWxqqnSgtzG08Qw,1218
 cgc/utils/cryptography/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cgc/utils/cryptography/aes_crypto.py,sha256=S0rKg38oy7rM5lTrP6DDpjLA-XRxuZggAXyxMFHtyzY,3333
 cgc/utils/cryptography/encryption_module.py,sha256=rbblBBorHYPGl-iKblyZX3_NuPEvUTpnH1l_RgNGCbA,1958
 cgc/utils/cryptography/rsa_crypto.py,sha256=h3jU5qPpj9uVjP1rTqZJTdYB5yjhD9HZpr_nD439h9Q,4180
-cgcsdk-1.0.7.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cgcsdk-1.0.7.dist-info/METADATA,sha256=lJihZEso4JIomyx6n8X1CA5GHAQT9B3XCWrFR1dANjw,3174
-cgcsdk-1.0.7.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-cgcsdk-1.0.7.dist-info/entry_points.txt,sha256=bdfIHeJ6Y-BBr5yupCVoK7SUrJj1yNdew8OtIOg_3No,36
-cgcsdk-1.0.7.dist-info/top_level.txt,sha256=nqW9tqcIcCXFigQT69AuOk7XHKc4pCuv4HGJQGXb6iA,12
-cgcsdk-1.0.7.dist-info/RECORD,,
+cgcsdk-1.0.10.dist-info/LICENSE,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cgcsdk-1.0.10.dist-info/METADATA,sha256=R_I_GsefZji2cCg2zbrk90fXOyWg5MDE0rWJmwiFjQU,3058
+cgcsdk-1.0.10.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+cgcsdk-1.0.10.dist-info/entry_points.txt,sha256=bdfIHeJ6Y-BBr5yupCVoK7SUrJj1yNdew8OtIOg_3No,36
+cgcsdk-1.0.10.dist-info/top_level.txt,sha256=nqW9tqcIcCXFigQT69AuOk7XHKc4pCuv4HGJQGXb6iA,12
+cgcsdk-1.0.10.dist-info/RECORD,,

cgc/sdk/handlers.py

@@ -1,24 +0,0 @@
-from pymongo.errors import OperationFailure, ConnectionFailure
-# TODO: redis, postgres
-# TODO: print -> click.echo ?
-
-def exception_handler(func):
-    def inner_function(*args, **kwargs):
-        err_count = 0
-        err_limit = 3
-        while True and err_count < err_limit:
-            try:
-                err_count += 1
-                return func(*args, **kwargs)
-            except (ConnectionFailure,) as e:
-                print(f"MongoDB connection error: {e}")
-                print(f"retrying {err_count}/{err_limit} ...")
-                # args[0]._reset_connection()
-            except OperationFailure as e:
-                print(f"MongoDB OperationFailure: {e}")
-                raise e
-        else:
-            print(f"MongoDB exception for customer")
-            # TODO: HTTP EXCEPTION FOR CUSTOMER, but keep thread alive
-
-    return inner_function

cgc/sdk/mongodb.py

@@ -1,204 +0,0 @@
-from pymongo.errors import ConnectionFailure
-from pymongo import MongoClient, DESCENDING, ASCENDING
-from cgc.sdk.handlers import exception_handler
-
-
-class MongoConnector:
-    """
-    :param database_name: Database name
-    :type database_name: str
-    :param hosts: Host address defined as IP or DNS Name with port ex. db.example.com:27017
-    :type hosts: list
-    :param username: Username used for connection, not required, defaults to None
-    :type username: str | None
-    :param password: Password for Username, used with Username, defaults to None
-    :type password: str | None
-    :param authsource: Database to authenticate user with, defaults to admin
-    :type authsource: str
-    """
-
-    def __init__(
-        self,
-        database_name: str,
-        hosts: str,
-        username: str = None,
-        password: str = None,
-        authsource: str = "admin",
-        replica_set: str = None,
-    ) -> None:
-        self._database_name = database_name
-        self._hosts = hosts
-        assert type(hosts) is str
-        'hosts must be a str of host addresses ex. "db.example.com:27017,db2.example.com:27017"'
-        self._username = username
-        self._password = password
-        self._authsource = authsource
-        _extra_args = []
-        if replica_set:
-            _extra_args.append(f"replicaSet={replica_set}")
-        if authsource:
-            _extra_args.append(f"authSource={authsource}")
-        _extra_args.append("readPreference=primaryPreferred")
-        _extra_args.append("ssl=false")
-        self._extra_args = "&".join(_extra_args)
-        print(self._extra_args)
-        self.connect()
-
-    def disconnect(self):
-        print(f"Disconnecting from MongoDB: {self._hosts}")
-        self._mongo_client.close()
-        print(f"Disconnected from MongoDB")
-
-    def connect(self):
-        _credentials = (
-            ":".join([self._username, self._password]) if self._username else ""
-        )
-        _host_connection_string = (
-            "@".join([_credentials, self._hosts]) if _credentials else self._hosts
-        )
-        while True:
-            try:
-                self._mongo_client = MongoClient(
-                    "mongodb://{}/{}?{}".format(
-                        _host_connection_string, self._database_name, self._extra_args
-                    )
-                )
-                self._mongo_client_server_info = self._mongo_client.server_info()
-                print(f"Connected to MongoDB ({self._database_name}): {self._hosts}")
-                self._db = self._mongo_client[self._database_name]
-                break
-            except (ConnectionFailure,) as e:
-                print(f"MongoDB connection error: {e}")
-                print(f"retrying to connect...")
-
-    def _select_collection(self, collection_name: str):
-        return self._db[collection_name]
-
-    def get_pymongo_client(self):
-        return self._mongo_client
-
-    @staticmethod
-    def _ascending():
-        return ASCENDING
-
-    @staticmethod
-    def _descending():
-        return DESCENDING
-
-    @exception_handler
-    def find(self, collection_name: str, query: dict, session=None):
-        collection = self._select_collection(collection_name)
-        return collection.find(query, session=session)
-
-    @exception_handler
-    def find_one(self, collection_name: str, query: dict, session=None):
-        collection = self._select_collection(collection_name)
-        return collection.find_one(query, session=session)
-
-    @exception_handler
-    def insert_one(self, collection_name: str, query: dict, session=None):
-        collection = self._select_collection(collection_name)
-        return collection.insert_one(query, session=session)
-
-    @exception_handler
-    def delete_one(self, collection_name: str, query: dict, session=None):
-        collection = self._select_collection(collection_name)
-        return collection.delete_one(query, session=session)
-
-    @exception_handler
-    def aggregate(self, collection_name: str, pipeline, session=None):
-        collection = self._select_collection(collection_name)
-        return collection.aggregate(pipeline=pipeline, session=session)
-
-    @exception_handler
-    def count_documents(self, collection_name: str, query: dict, session=None):
-        collection = self._select_collection(collection_name)
-        return collection.count_documents(query, session=session)
-
-    @exception_handler
-    def update_one(
-        self,
-        collection_name: str,
-        query: dict,
-        update: dict,
-        upsert: bool = False,
-        session=None,
-    ):
-        collection = self._select_collection(collection_name)
-        return collection.update_one(query, update, upsert=upsert, session=session)
-
-    @exception_handler
-    def watch(
-        self,
-        collection_name: str,
-        operation_before_watch,
-        operation_on_event,
-        first_operation_kwargs={},
-        second_operation_kwargs={},
-    ):
-        """Function to watch for changes in a collection. Preferred to run over threading as daemon.
-
-        :param collection_name: Name of the collection to watch
-        :type collection_name: str
-        :param operation_on_event: Function to run on event
-        :type operation_on_event: function
-        """
-        operation_before_watch(**first_operation_kwargs)
-        while True:
-            print("Creating new watch cursor")
-            watch_cursor = self._select_collection(
-                collection_name=collection_name
-            ).watch()
-            print("Watching: {}.{}\n".format(self._database_name, collection_name))
-            for d in watch_cursor:
-                if d["operationType"] == "invalidate":
-                    print(
-                        "Watch cursor invalidated (deleted collection: {}?)".format(
-                            collection_name
-                        )
-                    )
-                    print("Closing watch cursor")
-                    watch_cursor.close()
-                    break
-                else:
-                    operation_on_event(d, **second_operation_kwargs)
-                    watch_cursor.close()
-                    return d
-            else:
-                continue
-            break
-
-    @staticmethod
-    def example_operation_for_watch(d):
-        print(d)
-        # del d["fullDocument"]["_id"]
-        # print("local time   : {}".format(datetime.utcnow()))
-        # print("cluster time : {}".format(d["clusterTime"].as_datetime()))
-        # print("collection   : {}.{}".format(d["ns"]["db"], d["ns"]["coll"]))
-        # try:
-        #     print("doc          : {}".format(d["fullDocument"]))
-        # except KeyError:
-        #     pass
-
-
-def get_mongo_access(
-    app_name: str, password: str, database: str = "db", restart: bool = False
-):
-    global _mongo_access
-
-    def init_access():
-        global _mongo_access
-        _mongo_access = MongoConnector(
-            database_name=database,
-            hosts=f"{app_name}:27017",
-            username="admin",
-            password=password,
-        )
-
-    try:
-        if not isinstance(_mongo_access, MongoConnector) or restart:
-            init_access()
-    except NameError:
-        init_access()
-        pass
-    return _mongo_access

cgc/sdk/redis.py

@@ -1,91 +0,0 @@
-import redis.asyncio as redis_async
-import redis
-
-
-class RedisConnector:
-    redis_client_async = None
-    redis_client = None
-
-    def __init__(
-        self, host: str, password: str = None, decode_responses: bool = False
-    ) -> None:
-        self._host = host
-        assert type(host) is str
-        "host must be a str containing redis app name"
-        self._password = password
-        self._decode_responses = decode_responses
-
-    def connect(self, async_client: bool = False):
-        while True:
-            try:
-                if not async_client:
-                    self.redis_client = redis.Redis(
-                        host=self._host,
-                        port=6379,
-                        password=self._password,
-                        decode_responses=self._decode_responses,
-                    )
-                else:
-                    self.redis_client_async = redis_async.Redis(
-                        host=self._host,
-                        port=6379,
-                        password=self._password,
-                        decode_responses=self._decode_responses,
-                    )
-                print(f"Connected to Redis: {self._host}")
-                break
-            except (redis.ConnectionError,) as e:
-                print(f"Redis connection error: {e}")
-                print(f"retrying to connect...")
-
-    def get_redis_client(self):
-        if self.redis_client is None:
-            self.connect()
-        return self.redis_client
-
-    def get_redis_client_async(self):
-        if self.redis_client_async is None:
-            self.connect(async_client=True)
-        return self.redis_client_async
-
-
-def get_redis_access(
-    app_name: str,
-    password: str,
-    decode_responses: bool = False,
-    restart: bool = False,
-    async_client=False,
-):
-    global _redis_access
-    global _redis_access_async
-
-    def init_access(async_client=False):
-        global _redis_access
-        global _redis_access_async
-
-        _redis_access = None
-        _redis_access_async = None
-
-        if not async_client:
-            _redis_access = RedisConnector(
-                host=app_name, password=password, decode_responses=decode_responses
-            )
-        else:
-            _redis_access_async = RedisConnector(
-                host=app_name, password=password, decode_responses=decode_responses
-            )
-
-    try:
-        if not async_client:
-            if not isinstance(_redis_access, RedisConnector) or restart:
-                init_access()
-        else:
-            if not isinstance(_redis_access_async, RedisConnector) or restart:
-                init_access(True)
-    except NameError:
-        if not async_client:
-            init_access()
-        else:
-            init_access(True)
-        pass
-    return _redis_access if not async_client else _redis_access_async