brynq-sdk-ftp 3.0.1__tar.gz → 3.0.3__tar.gz

{brynq_sdk_ftp-3.0.1 → brynq_sdk_ftp-3.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq_sdk_ftp
-Version: 3.0.1
+Version: 3.0.3
 Summary: FTP wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_ftp-3.0.1 → brynq_sdk_ftp-3.0.3}/brynq_sdk_ftp/sftp.py

@@ -3,7 +3,6 @@ from io import StringIO
 from paramiko.client import SSHClient, AutoAddPolicy
 from paramiko import RSAKey
 from paramiko.sftp_attr import SFTPAttributes
-import pysftp
 from typing import Union, List, Literal, Optional
 from stat import S_ISREG
 import os
@@ -17,25 +16,57 @@ class SFTP(BrynQ):
         :param debug: If you want to see debug messages
         """
         super().__init__()
-        credentials = self.interfaces.credentials.get(system="sftp", system_type=system_type)
-        credentials = credentials.get('data')
         self.debug = debug
-        if self.debug:
-            print(credentials)
-        self.host = credentials['host']
-        self.port = 22 if credentials['port'] is None else credentials['port']
-        self.username = credentials['username']
-        self.password = credentials['password']
-        self.cnopts = pysftp.CnOpts()
-        self.cnopts.hostkeys = None
-        self.private_key_path = credentials.get('private_key_password', None)
-        self.private_key_passphrase = credentials.get('private_key_password', None)
-        self.private_key = credentials.get('private_key', None)
-        if self.private_key:
-            self.private_key = RSAKey(file_obj=StringIO(credentials.get('private_key')), password=self.private_key_passphrase)
+
         self.client = SSHClient()
         self.client.set_missing_host_key_policy(AutoAddPolicy())
 
+        # Try to fetch credentials from BrynQ; if not present, attributes remain unset
+        try:
+            credentials = self.interfaces.credentials.get(system="sftp", system_type=system_type)
+            credentials = credentials.get('data')
+            if credentials:
+                if self.debug:
+                    print(credentials)
+                self.host = credentials['host']
+                self.port = 22 if credentials.get('port') is None else credentials.get('port')
+                self.username = credentials.get('username')
+                self.password = credentials.get('password')
+                self.private_key_path = credentials.get('private_key_password', None)
+                self.private_key_passphrase = credentials.get('private_key_password', None)
+                self.private_key = None
+                if credentials.get('private_key'):
+                    self.private_key = RSAKey(file_obj=StringIO(credentials.get('private_key')), password=self.private_key_passphrase)
+        except ValueError:
+            print("No credentials found for SFTP. If this was intended, use _set_credentials() to set the credentials to pass variables ['host', 'port', 'username', 'password', 'private_key', 'private_key_password']")
+
+    def _set_credentials(self, credentials: dict):
+        """
+        When a child class(ex:Meta4) needs to set the credentials, use this method.
+        Set SFTP connection credentials programmatically.
+
+        Expected keys in credentials dict:
+        - host (str)
+        - port (int, optional; defaults to 22)
+        - username (str)
+        - password (str, optional)
+        - private_key (str, optional; PEM string)
+        - private_key_password (str, optional; passphrase for private_key)
+        """
+        # Core connection parameters
+        self.host = credentials['host']
+        self.port = 22 if credentials.get('port') is None else credentials.get('port')
+        self.username = credentials.get('username')
+        self.password = credentials.get('password')
+
+        # Key options
+        self.private_key_path = credentials.get('private_key_password')
+        self.private_key_passphrase = credentials.get('private_key_password')
+        if credentials.get('private_key'):
+            self.private_key = RSAKey(file_obj=StringIO(credentials.get('private_key')), password=self.private_key_passphrase)
+
+
+
     def upload_file(self, local_filepath, remote_filepath, confirm=True) -> SFTPAttributes:
         """
         Upload a single file to a remote location. If there is no Private key

{brynq_sdk_ftp-3.0.1 → brynq_sdk_ftp-3.0.3}/brynq_sdk_ftp.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 1.0
 Name: brynq-sdk-ftp
-Version: 3.0.1
+Version: 3.0.3
 Summary: FTP wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_ftp-3.0.1 → brynq_sdk_ftp-3.0.3}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_namespace_packages
 
 setup(
     name='brynq_sdk_ftp',
-    version='3.0.1',
+    version='3.0.3',
     description='FTP wrapper from BrynQ',
     long_description='FTP wrapper from Brynq',
     author='BrynQ',