UncountablePythonSDK 0.0.43__py3-none-any.whl → 0.0.44__py3-none-any.whl

{UncountablePythonSDK-0.0.43.dist-info → UncountablePythonSDK-0.0.44.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: UncountablePythonSDK
-Version: 0.0.43
+Version: 0.0.44
 Summary: Uncountable SDK
 Project-URL: Homepage, https://github.com/uncountableinc/uncountable-python-sdk
 Project-URL: Repository, https://github.com/uncountableinc/uncountable-python-sdk.git

{UncountablePythonSDK-0.0.43.dist-info → UncountablePythonSDK-0.0.44.dist-info}/RECORD

@@ -28,8 +28,8 @@ pkgs/argument_parser/case_convert.py,sha256=NuJLJUJRbyVb6_Slen4uqaStEHbcOS1d-hBB
 pkgs/filesystem_utils/__init__.py,sha256=yxfwtYFvqq_fjMl-tg2jwa6eNPNZUF4ykqsSzALyNdw,1143
 pkgs/filesystem_utils/_gdrive_session.py,sha256=OZudNoP2HikolnpurVJhJdh5fgzqbaZQvn53ReGGXx4,11015
 pkgs/filesystem_utils/_local_session.py,sha256=xFEYhAvNqrOYqwt4jrEYOuYkjJn0zclZhTelW_Q1-rw,2325
-pkgs/filesystem_utils/_sftp_session.py,sha256=jtvxmMs3H1RneuJhztx1pCSbX9hBq3srFV1IVNqUlBA,4729
-pkgs/filesystem_utils/file_type_utils.py,sha256=F6P2chbXIDwAUAiMc_ja1_rxrhGy3fA1-ymAnIK3gVU,1259
+pkgs/filesystem_utils/_sftp_session.py,sha256=gNoUD_b4MuVqWj31nU-FpfpXZlyWkwdEHtX1S8W6gpQ,4727
+pkgs/filesystem_utils/file_type_utils.py,sha256=CC2c7127TE84vlRTy7bpE4Qvy7ytxuRNA3dwP91PXA8,1257
 pkgs/filesystem_utils/filesystem_session.py,sha256=BQ2Go8Mu9-GcnaWh2Pm4x7ugLVsres6XrOQ8RoiEpcE,1045
 pkgs/serialization/__init__.py,sha256=LifasRW0a50A3qRFmo2bf3FQ6TXhZWOTz2-CVTgPjcQ,753
 pkgs/serialization/missing_sentry.py,sha256=aM_9KxbCk9dVvXvcOKgkIQBqFWvLhv8QlIUCiuFEXMo,806
@@ -86,7 +86,7 @@ uncountable/integration/db/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 uncountable/integration/db/connect.py,sha256=YtQHJ1DBGPhxKFRCfiXqohOYUceKSxMVOJ88aPI48Ug,181
 uncountable/integration/executors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 uncountable/integration/executors/executors.py,sha256=v5ClGVUlvrZcMdmGQa8Ll668G_HGTnKpGOnTM7UMZCQ,956
-uncountable/integration/executors/generic_upload_executor.py,sha256=lCHGIMCe3t_UvEgHMBoOneWR2zbYaGXztt5F2zDRBLw,8796
+uncountable/integration/executors/generic_upload_executor.py,sha256=H_pfseSATBvY44xVNFC8VP6wL9pqoXG2qcReA76hl0E,8790
 uncountable/integration/executors/script_executor.py,sha256=OmSBOtU48G3mqza9c2lCm84pGGyaDk-ZBJCx3RsdJXc,846
 uncountable/integration/secret_retrieval/__init__.py,sha256=3QXVj35w8rRMxVvmmsViFYDi3lcb3g70incfalOEm6o,87
 uncountable/integration/secret_retrieval/retrieve_secret.py,sha256=M0qXVJpD8hMYIFypHFeyh598sqmIDX8ZOyXK23CluF0,1323
@@ -235,7 +235,7 @@ uncountable/types/api/triggers/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr
 uncountable/types/api/triggers/run_trigger.py,sha256=_Rpha9nxXI3Xr17CrGDtofg4HZ81x2lt0rMZ6As0qfE,893
 uncountable/types/api/uploader/__init__.py,sha256=gCgbynxG3jA8FQHzercKtrHKHkiIKr8APdZYUniAor8,55
 uncountable/types/api/uploader/invoke_uploader.py,sha256=2RCvnccwEm6J5dt7UJVLUKxfk49vfpGXjBLCWIBD3Ck,1006
-UncountablePythonSDK-0.0.43.dist-info/METADATA,sha256=vhbJVX4f-c2phTYrI3H2MJ4NCXfkOHVeHa3NHq-SEGk,1707
-UncountablePythonSDK-0.0.43.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-UncountablePythonSDK-0.0.43.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
-UncountablePythonSDK-0.0.43.dist-info/RECORD,,
+UncountablePythonSDK-0.0.44.dist-info/METADATA,sha256=pq3HyRKI8gVZ_HmK6Z7WQekcPSlTMWShM9CTwNpotGE,1707
+UncountablePythonSDK-0.0.44.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+UncountablePythonSDK-0.0.44.dist-info/top_level.txt,sha256=1UVGjAU-6hJY9qw2iJ7nCBeEwZ793AEN5ZfKX9A1uj4,31
+UncountablePythonSDK-0.0.44.dist-info/RECORD,,

pkgs/filesystem_utils/_sftp_session.py

@@ -63,7 +63,7 @@ class SFTPSession(FileSystemSession):
         super().__init__()
         self.host: str = sftp_config.ip
         self.username: str = sftp_config.username
-        self.key_file: str | paramiko.AgentKey | None = (
+        self.key_file: str | paramiko.RSAKey | None = (
             sftp_config.pem_path
             if sftp_config.pem_path is not None
             else sftp_config.pem_key

pkgs/filesystem_utils/file_type_utils.py

@@ -55,7 +55,7 @@ class FileSystemSFTPConfig:
     ip: str
     username: str
     pem_path: str | None
-    pem_key: paramiko.AgentKey | None = None
+    pem_key: paramiko.RSAKey | None = None
     password: str | None = None
     valid_extensions: Optional[tuple[str]] = None
     recursive: bool = True

uncountable/integration/executors/generic_upload_executor.py

@@ -4,6 +4,8 @@ import re
 from datetime import datetime, timezone
 from tempfile import NamedTemporaryFile
 
+import paramiko
+
 from pkgs.filesystem_utils import (
     FileObjectData,
     FileSystemFileReference,
@@ -176,14 +178,14 @@ class GenericUploadJob(Job):
                 pem_secret = retrieve_secret(
                     self.data_source.pem_secret, profile_metadata=args.profile_metadata
                 )
-                with NamedTemporaryFile() as pem_file:
-                    pem_file.write(pem_secret.encode())
-                    sftp_config = FileSystemSFTPConfig(
-                        ip=self.data_source.host,
-                        username=self.data_source.username,
-                        pem_path=pem_file.name,
-                    )
-                    return SFTPSession(sftp_config=sftp_config)
+                pem_key = paramiko.RSAKey.from_private_key(io.StringIO(pem_secret))
+                sftp_config = FileSystemSFTPConfig(
+                    ip=self.data_source.host,
+                    username=self.data_source.username,
+                    pem_path=None,
+                    pem_key=pem_key,
+                )
+                return SFTPSession(sftp_config=sftp_config)
 
     def run(self, args: JobArguments) -> JobResult:
         client = args.client