atlan-application-sdk 2.3.0__py3-none-any.whl → 2.3.2__py3-none-any.whl

application_sdk/services/objectstore.py

@@ -2,13 +2,13 @@
 
 import json
 import os
-import shutil
 from typing import List, Union
 
 import orjson
 from dapr.clients import DaprClient
 from temporalio import activity
 
+from application_sdk.common.file_ops import SafeFileOps
 from application_sdk.constants import (
     DAPR_MAX_GRPC_MESSAGE_LENGTH,
     DEPLOYMENT_OBJECT_STORE_NAME,
@@ -304,7 +304,7 @@ class ObjectStore:
             ... )
         """
         try:
-            with open(source, "rb") as f:
+            with SafeFileOps.open(source, "rb") as f:
                 file_content = f.read()
         except IOError as e:
             logger.error(f"Error reading file {source}: {str(e)}")
@@ -366,7 +366,7 @@ class ObjectStore:
             ...     recursive=False
             ... )
         """
-        if not os.path.isdir(source):
+        if not SafeFileOps.isdir(source):
             raise ValueError(f"The provided path '{source}' is not a valid directory.")
 
         try:
@@ -423,13 +423,13 @@ class ObjectStore:
         """
         # Ensure directory exists
 
-        if not os.path.exists(os.path.dirname(destination)):
-            os.makedirs(os.path.dirname(destination), exist_ok=True)
+        if not SafeFileOps.exists(os.path.dirname(destination)):
+            SafeFileOps.makedirs(os.path.dirname(destination), exist_ok=True)
 
         try:
             response_data = await cls.get_content(source, store_name)
 
-            with open(destination, "wb") as f:
+            with SafeFileOps.open(destination, "wb") as f:
                 f.write(response_data)
 
             logger.info(f"Successfully downloaded file: {source}")
@@ -552,10 +552,10 @@ class ObjectStore:
             path: The path to the file or directory to remove.
         """
         try:
-            if os.path.isfile(path) or os.path.islink(path):
-                os.remove(path)
-            elif os.path.isdir(path):
-                shutil.rmtree(path)
+            if SafeFileOps.isfile(path) or os.path.islink(path):
+                SafeFileOps.remove(path)
+            elif SafeFileOps.isdir(path):
+                SafeFileOps.rmtree(path)
         except FileNotFoundError:
             pass  # ignore if the file or directory doesn't exist
         except Exception as e:

application_sdk/services/statestore.py

@@ -8,6 +8,7 @@ from typing import Any, Dict
 from temporalio import activity
 
 from application_sdk.activities.common.utils import get_object_store_prefix
+from application_sdk.common.file_ops import SafeFileOps
 from application_sdk.constants import (
     APPLICATION_NAME,
     STATE_STORE_PATH_TEMPLATE,
@@ -167,10 +168,10 @@ class StateStore:
             # update the state with the new value
             current_state[key] = value
 
-            os.makedirs(os.path.dirname(state_file_path), exist_ok=True)
+            SafeFileOps.makedirs(os.path.dirname(state_file_path), exist_ok=True)
 
             # save the state to a local file
-            with open(state_file_path, "w") as file:
+            with SafeFileOps.open(state_file_path, "w") as file:
                 json.dump(current_state, file)
 
             # save the state to the object store
@@ -243,10 +244,10 @@ class StateStore:
             # update the state with the new value
             current_state.update(value)
 
-            os.makedirs(os.path.dirname(state_file_path), exist_ok=True)
+            SafeFileOps.makedirs(os.path.dirname(state_file_path), exist_ok=True)
 
             # save the state to a local file
-            with open(state_file_path, "w") as file:
+            with SafeFileOps.open(state_file_path, "w") as file:
                 json.dump(current_state, file)
 
             # save the state to the object store

application_sdk/version.py

@@ -2,4 +2,4 @@
 Version information for the application_sdk package.
 """
 
-__version__ = "2.3.0"
+__version__ = "2.3.2"

{atlan_application_sdk-2.3.0.dist-info → atlan_application_sdk-2.3.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: atlan-application-sdk
-Version: 2.3.0
+Version: 2.3.2
 Summary: Atlan Application SDK is a Python library for developing applications on the Atlan Platform
 Project-URL: Repository, https://github.com/atlanhq/application-sdk
 Project-URL: Documentation, https://github.com/atlanhq/application-sdk/README.md
@@ -93,8 +93,23 @@ poetry add atlan-application-sdk
 
 ## Getting Started
 
-- Want to develop locally or run examples from this repository? Check out our [Getting Started Guide](docs/docs/guides/getting-started.md) for a step-by-step walkthrough!
-- Detailed documentation for the application-sdk is available at [docs](https://github.com/atlanhq/application-sdk/blob/main/docs/docs/) folder.
+- Want to develop locally or run examples from this repository? Check out our [Getting Started Guide](docs/guides/getting-started.md) for a step-by-step walkthrough!
+- Detailed documentation for the application-sdk is available at [docs](https://github.com/atlanhq/application-sdk/blob/main/docs/) folder.
+
+## 🐳 Docker
+
+### Pre-built Images (Harbor)
+
+```bash
+# Latest main image
+docker pull registry.atlan.com/public/application-sdk:main-latest
+
+# Versioned image
+docker pull registry.atlan.com/public/application-sdk:main-2.3.1
+
+# Commit-specific image
+docker pull registry.atlan.com/public/application-sdk:sha-49c027f
+```
 
 ## Contributing
 

{atlan_application_sdk-2.3.0.dist-info → atlan_application_sdk-2.3.2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 application_sdk/__init__.py,sha256=2e2mvmLJ5dxmJGPELtb33xwP-j6JMdoIuqKycEn7hjg,151
-application_sdk/constants.py,sha256=_JprQls6AurPBHMG8TgGkLT7q4Lj8YxMAabYQx8Heks,12544
-application_sdk/version.py,sha256=Lnq43x1REzO0vMqhztEDG59AS0CKBDL70U5_-QOtwwc,84
+application_sdk/constants.py,sha256=PneIcT8-ypvCDRZo0uEzfn43X41MMQQPeGpBE1AAm5E,13511
+application_sdk/version.py,sha256=QlDJbSgyks5cBqICIKJQf6EUYOPJztONFx8hFKl5iXY,84
 application_sdk/worker.py,sha256=5V_zxkx64moHSkMaZTNnbfstoygMBk7SAOSEy6AZZZM,11552
 application_sdk/activities/__init__.py,sha256=i7iY6aL1VFg185n2rLLvD_sI2BA9zJ33jL5rD_sY__U,12350
 application_sdk/activities/lock_management.py,sha256=6Wdf3jMKitoarHQP91PIJOoGFz4aaOLS_40c7n1yAOA,3902
@@ -14,7 +14,7 @@ application_sdk/activities/metadata_extraction/base.py,sha256=ENFojpxqKdN_eVSL4i
 application_sdk/activities/metadata_extraction/rest.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/activities/metadata_extraction/sql.py,sha256=CmE77EsgbOuDL5AKaRCnq1jApJnDWNVxx-RZ49cJwus,27415
 application_sdk/activities/query_extraction/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-application_sdk/activities/query_extraction/sql.py,sha256=Gsa79R8CYY0uyt3rA2nLMfQs8-C4_zg1pJ_yYSF2cZw,21193
+application_sdk/activities/query_extraction/sql.py,sha256=1KmDY_B1saz8wv2hGtGy5aOu07qClRDv1M_wCwhQQ8k,21298
 application_sdk/application/__init__.py,sha256=sTgyt4MBDvsQJ2yI0TxzB1Wp68WC-5fBkSExUgtW8LY,12559
 application_sdk/application/metadata_extraction/sql.py,sha256=TmHTLdWLxqtSbVyzfNmnQNe6BUePVIlZxaiQHZXXbyI,9812
 application_sdk/clients/__init__.py,sha256=C9T84J7V6ZumcoWJPAxdd3tqSmbyciaGBJn-CaCCny0,1341
@@ -35,6 +35,8 @@ application_sdk/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3
 application_sdk/common/aws_utils.py,sha256=xlSMIQyjvQ-CydEXaxXrnPUygv7AAbCLsxhZ2wtKnzg,11219
 application_sdk/common/error_codes.py,sha256=BGdOSfhKe4pd9oM9T2N8GASZeIJU2Be1NcThL8LkKBQ,14900
 application_sdk/common/file_converter.py,sha256=ta0PVh7uIEGJg0BTPUJnSjj55ve2iVAOkqwAeg96_-g,3079
+application_sdk/common/file_ops.py,sha256=CSl3wVh7Bt3EBEWzdGvbUyh5R4aBZ5TkkbNS8d4frcA,4338
+application_sdk/common/path.py,sha256=zCyrK4OA1fcMgV8lkxAeCgH6VP9SEIpYmi2GLi3NPIU,1169
 application_sdk/common/types.py,sha256=qkVr3SAR1zn5_0w9hFt18vGtcnaPLKlMJLSBhTSKePU,134
 application_sdk/common/utils.py,sha256=czcWvqoe2PawDvHOahk_AI88Zqth-CM3KzdGmehHQJ4,19286
 application_sdk/common/.cursor/BUGBOT.md,sha256=OkB5TMAEJFzaBfbNb3g9ZDPW2r1krQE_KEuJbytMPuI,12176
@@ -63,19 +65,21 @@ application_sdk/handlers/sql.py,sha256=WkFSo61SLiY3FoHO35Gqxx6YXRTftVJGrWBdLRELw
 application_sdk/interceptors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/interceptors/cleanup.py,sha256=JlFcM_2Y5AIEfGTSNe0aoon7eoE68MIXI0rA3LHsSeY,5966
 application_sdk/interceptors/correlation_context.py,sha256=j486uXZbLA85eTBJ8hxXS13vA8Ucf3_Nt5KiyA_cJVY,5864
-application_sdk/interceptors/events.py,sha256=e0O6uK9_aCTmOORaTGN9RbcTg9_KNaakq-Meh8l2lsI,6477
+application_sdk/interceptors/events.py,sha256=z6aptp7D6gBunT9niKvb8KwdvfhQ9-dmSE1xcsvKB-Y,6914
 application_sdk/interceptors/lock.py,sha256=5ETm20zrTaH2b9fepN4Ckp1tGJV-uINqDrno_5RW3aw,6169
 application_sdk/interceptors/models.py,sha256=kEzJKvb-G1M7aKrLPgAmsukJXLXeh8hIJKwEkOiaY28,6115
 application_sdk/interceptors/.cursor/BUGBOT.md,sha256=pxmUF2c7dtaXAX8yAa1-LBa6FCrj_uw7aQcHrppjf1A,14570
 application_sdk/io/__init__.py,sha256=Fse-fEyrpMlLUxwyFkH8vWWSXz8rdWGlAjZy5ulAZCU,27767
-application_sdk/io/json.py,sha256=sNSyHZCM_ZeaiJHUrolYVHKreBQqSCBsfsjD3JSkoD0,19729
-application_sdk/io/parquet.py,sha256=zy9H_TvWI5CkktJ582NH7Ut_5rUH_S0Jy7ZbTD0JxeI,34227
+application_sdk/io/json.py,sha256=tQvi8pCTvZGhf5MD7b3FzE-LslQZNhadpjb4qx47Xgo,20476
+application_sdk/io/parquet.py,sha256=csscTnu4pULQ81ShfB54HSu5qBjrJpt3QqJGtc_orWI,34371
 application_sdk/io/utils.py,sha256=sn_8Q6HgjeC8uyZp2XGMAfqdJ8XzkIllOEVYXIH54DY,10724
 application_sdk/observability/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/observability/context.py,sha256=lJjpfxEjMY_hrdSDqq519YaWcztgc_1nM4d-mGV5shs,634
-application_sdk/observability/logger_adaptor.py,sha256=Fq5OE579ozr0EzsNYEh2H0q3POVAxtlWfJ-PSwWDGLM,30194
-application_sdk/observability/metrics_adaptor.py,sha256=5Oz02lUED60duryoVDF9mbD11fpxhbXi7P1609n_15Y,16446
+application_sdk/observability/logger_adaptor.py,sha256=dwyRMRHj4qZu_HhwK12SDpJ8NPpRhHVHFgOoT2WCFwY,30280
+application_sdk/observability/metrics_adaptor.py,sha256=555Z8AaABrD4whqkr4sF5ddE5mtToZIecTezpS8bWv4,14008
+application_sdk/observability/models.py,sha256=zAVhk4cKMCKco_9yFBZR0dh31VqeAhO2B1GCe2wcGNg,3847
 application_sdk/observability/observability.py,sha256=O2rBal_0pmFRen7Yx4c4dSH1NyiT937b4bY2w63q-4U,23751
+application_sdk/observability/segment_client.py,sha256=Ny2lxdl7lek9_oNP1nduRGNzziy5kb3fQ72nLpHjewI,15540
 application_sdk/observability/traces_adaptor.py,sha256=0eQJPN-tYA_dV8D3uEa5ZiX9g12NDuLnPaFuQMVDdL0,18242
 application_sdk/observability/utils.py,sha256=-02GAFom8Bg4SNyCTNYySmen2dzvLfTu43bqsNq1AH0,3096
 application_sdk/observability/decorators/observability_decorator.py,sha256=yd6qfrg1MmH5KcZ5Ydzb0RaBzmxx5FrmiI9qwvZx3EU,8963
@@ -94,10 +98,10 @@ application_sdk/server/mcp/server.py,sha256=HG8tFmcc-f9Wj3vZzs2oRoNJzN1s5hwjnKyk
 application_sdk/services/__init__.py,sha256=H-5HZEPdr53MUfAggyHqHhRXDRLZFZsxvJgWbr257Ds,465
 application_sdk/services/_utils.py,sha256=0yHqDP6qNb1OT-bX2XRYQPZ5xkGkV13nyRw6GkPlHs8,1136
 application_sdk/services/atlan_storage.py,sha256=TKzXxu0yXeUcmZehwp8PcnQTC4A9w9RlZ0Fl-Xp1bLE,8509
-application_sdk/services/eventstore.py,sha256=wCT921KRzUe3fAWKC-bbM6_OtIJTKpSQrOutPQzMEgs,6745
-application_sdk/services/objectstore.py,sha256=dLljCsPPSr24bPKh0l3-xRblofzKVQ4LDfqDrMp8JGc,20819
+application_sdk/services/eventstore.py,sha256=MsRDpnDboSzmh2tyrJBccTPgQvyA3EgeB8JEcGag45Q,10991
+application_sdk/services/objectstore.py,sha256=vnasFrOP0AIUAEW9DuNsemiyPVfzMAhRUfODTd4gbSA,20924
 application_sdk/services/secretstore.py,sha256=Pmn1WlmHmgaDhWz5OXBB5_rKXQQMyLMzadwZSNKwc6Q,19070
-application_sdk/services/statestore.py,sha256=3-afiM3Vsoe1XDYRokdGTB5I5CwOKyieuX5RwIZf77o,9413
+application_sdk/services/statestore.py,sha256=aVEvk_KqJCkqo_gBVk77qnX3MIfCASC0WqdeYBqNimg,9511
 application_sdk/test_utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 application_sdk/test_utils/workflow_monitoring.py,sha256=gqq6CsT62GrMt2GqtNSb1iD_-t4MBffQvpO0BXboZek,3490
 application_sdk/test_utils/e2e/__init__.py,sha256=qswG5goB5Xzt3yQyZsKGq20EnZMZCmXYTe0CK3Qk3xo,10125
@@ -155,8 +159,8 @@ application_sdk/workflows/metadata_extraction/__init__.py,sha256=jHUe_ZBQ66jx8bg
 application_sdk/workflows/metadata_extraction/sql.py,sha256=6ZaVt84n-8U2ZvR9GR7uIJKv5v8CuyQjhlnoRJvDszc,12435
 application_sdk/workflows/query_extraction/__init__.py,sha256=n066_CX5RpJz6DIxGMkKS3eGSRg03ilaCtsqfJWQb7Q,117
 application_sdk/workflows/query_extraction/sql.py,sha256=kT_JQkLCRZ44ZpaC4QvPL6DxnRIIVh8gYHLqRbMI-hA,4826
-atlan_application_sdk-2.3.0.dist-info/METADATA,sha256=F-6xR-DchbERcVzifQhz_M4jjUhcYke0tZhe3vIwHao,6014
-atlan_application_sdk-2.3.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-atlan_application_sdk-2.3.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-atlan_application_sdk-2.3.0.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
-atlan_application_sdk-2.3.0.dist-info/RECORD,,
+atlan_application_sdk-2.3.2.dist-info/METADATA,sha256=UkJTSVogF_LaR_OFCl4BFODCJhw0OSbLT4jxzv0W2zs,6325
+atlan_application_sdk-2.3.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+atlan_application_sdk-2.3.2.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+atlan_application_sdk-2.3.2.dist-info/licenses/NOTICE,sha256=A-XVVGt3KOYuuMmvSMIFkg534F1vHiCggEBp4Ez3wGk,1041
+atlan_application_sdk-2.3.2.dist-info/RECORD,,