From 1874d8b79befa998cde47c16166ee73b4359be4d Mon Sep 17 00:00:00 2001 From: Quang Date: Thu, 21 Nov 2024 16:08:42 +0700 Subject: [PATCH] add more exception handle --- dsst_etl/upload_pdfs.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/dsst_etl/upload_pdfs.py b/dsst_etl/upload_pdfs.py index d70931f..05a2782 100644 --- a/dsst_etl/upload_pdfs.py +++ b/dsst_etl/upload_pdfs.py @@ -7,7 +7,6 @@ import boto3 import psycopg2 import sqlalchemy -from botocore.exceptions import ClientError from dsst_etl import __version__ from dsst_etl._utils import get_bucket_name, get_compute_context_id @@ -63,8 +62,7 @@ def upload_pdfs(self, pdf_paths: List[str]) -> Tuple[List[str], List[str]]: self.s3_client.upload_file(pdf_path, self.bucket_name, s3_key) successful_uploads.append(pdf_path) logger.info(f"Successfully uploaded {pdf_path} to S3") - - except ClientError as e: + except Exception as e: logger.error(f"Failed to upload {pdf_path}: {str(e)}") failed_uploads.append(pdf_path)