From 42b7999f7854359362634971a9c34f184a263acc Mon Sep 17 00:00:00 2001 From: keeagnsmith21 Date: Tue, 21 Apr 2026 09:12:15 +0800 Subject: [PATCH] except IO error for already existing directories --- observatory_platform/sftp.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/observatory_platform/sftp.py b/observatory_platform/sftp.py index f9ccf9ca7..19c20059a 100644 --- a/observatory_platform/sftp.py +++ b/observatory_platform/sftp.py @@ -102,7 +102,10 @@ def move_files_to_in_progress(self, upload_files: Union[list, str]): upload_files = [upload_files] with make_sftp_connection(self.sftp_conn_id) as sftp: - sftp.mkdir(self.in_progress) + try: + sftp.mkdir(self.in_progress) + except IOError: + print(f"Directory {self.in_progress} already exists. Skipping creation") for file in upload_files: file_name = os.path.basename(file) upload_file = os.path.join(self.upload, file_name) @@ -120,7 +123,10 @@ def move_files_to_finished(self, in_progress_files: Union[list, str]): in_progress_files = [in_progress_files] with make_sftp_connection(self.sftp_conn_id) as sftp: - sftp.mkdir(self.finished) + try: + sftp.mkdir(self.finished) + except IOError: + print(f"Directory {self.finished} already exists. Skipping creation") for file in in_progress_files: file_name = os.path.basename(file) in_progress_file = os.path.join(self.in_progress, file_name)