Merged FTP tasks into ftp.py

Author: vinod-kanigicherla

tasks/upload/ftp/retrieve_from_ftp.py renamed to tasks/upload/ftp.py

@@ -4,6 +4,87 @@
 from netunicorn.base import Failure, Result, Success, Task
 
 
+class UploadToFTP(Task):
+    """
+    Task for uploading a local file to an FTP Server. Establishes a connection to the specified FTP server, navigates to the desired remote directory, and uploads the specified local file.
+    """
+
+    def __init__(
+        self,
+        local_filepath: str,
+        ftp_url: str,
+        username: str,
+        password: str,
+        destination_dir: str = "/",
+        timeout: int = 30,
+        *args,
+        **kwargs,
+    ):
+        """
+        Initializes the UploadToFTP task with parameters.
+
+        Parameters:
+          local_filepath (str): Path to local file to upload.
+          ftp_url (str): URL or IP address of FTP.
+          username (str): Username credential for FTP auth.
+          password (str): Password credential for FTP auth.
+          destination_dir (str): Destination directory on the FTP server where the file will be uploaded to. Defaults to "/".
+          timeout (int, optional): Timeout value for FTP connection measued in seconds. Defaults to 30 seconds.
+        """
+        super().__init__(*args, **kwargs)
+        self.local_filepath = local_filepath
+        self.ftp_url = ftp_url
+        self.username = username
+        self.password = password
+        self.destination_dir = destination_dir
+        self.timeout = timeout
+
+    def run(self) -> Result:
+        """
+        Uploads the local file to FTP server.
+
+        Steps:
+        1. Finds the specified local file.
+        2. Connects to FTP server using provided credentials.
+        3. Sets remote directory if specified, else defaults to "/".
+        4. Uploads the local file to FTP server and closes connection.
+
+        Returns:
+          Result:
+            Success: Contains a success message if successful upload.
+            --OR--
+            Failure: Contains an error message if upload fails.
+
+        """
+        try:
+            if not os.path.isfile(self.local_filepath):
+                return Failure(f"Local file does not exist: {self.local_filepath}")
+
+            ftp = FTP(self.ftp_url, timeout=self.timeout)  # Modify timeout as needed
+            ftp.login(user=self.username, passwd=self.password)
+
+            if self.destination_dir:
+                ftp.cwd(self.destination_dir)
+
+            with open(self.local_filepath, "rb") as f:
+                remote_filename = os.path.basename(self.local_filepath)
+                ftp.storbinary(f"STOR {remote_filename}", f)
+
+            ftp.quit()
+            return Success(
+                f"Successfully uploaded {self.local_filepath} to {self.ftp_url}/{self.destination_dir}"
+            )
+
+        except FileNotFoundError:
+            return Failure(f"File not found: {self.local_filepath}")
+
+        except error_perm as e:
+            return Failure(f"FTP permission error: {e}")
+
+        except Exception as e:
+            return Failure(f"An unexpected error occurred: {str(e)}")
+
+
 class RetrieveFromFTP(Task):
     """
     Task for retrieving a file from an FTP server to a local directory. Establishes a connection to the specified FTP server, navigates to the desired remote directory and downloads the specified file to a local directory.