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

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

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

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

@@ -16,23 +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.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.2 → 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.2
+Version: 3.0.3
 Summary: FTP wrapper from BrynQ
 Home-page: UNKNOWN
 Author: BrynQ

{brynq_sdk_ftp-3.0.2 → 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.2',
+    version='3.0.3',
     description='FTP wrapper from BrynQ',
     long_description='FTP wrapper from Brynq',
     author='BrynQ',