From 2471c86c6ca5db0c4199e72deafe52f7d595e081 Mon Sep 17 00:00:00 2001 From: Nik Afiq Date: Mon, 13 May 2024 16:17:00 +0900 Subject: [PATCH] sap-data-decrypt-fix --- lambda/sap-data-decrypt/datadecrypt/main.py | 18 ++++++------------ 1 file changed, 6 insertions(+), 12 deletions(-) diff --git a/lambda/sap-data-decrypt/datadecrypt/main.py b/lambda/sap-data-decrypt/datadecrypt/main.py index 86eeb63f..836f7c18 100644 --- a/lambda/sap-data-decrypt/datadecrypt/main.py +++ b/lambda/sap-data-decrypt/datadecrypt/main.py @@ -37,7 +37,6 @@ TARGET_BUCKET_BY_DATA_SOURCE = { # 変数 s3_client = boto3.client('s3') -s3_resource = boto3.resource('s3') sns_client = boto3.client('sns') # logger設定 @@ -125,9 +124,8 @@ def handler(event, context): decrypt_file_key = f'{execute_date}/{decrypt_file_name}' decrypt_bucket_name = TARGET_BUCKET_BY_DATA_SOURCE[s3_event.data_source_name] logger.info(f'I-07-02 復号化ファイル出力　ファイル名：{decrypt_file_name}　出力先：{decrypt_bucket_name}/{decrypt_file_key}') - decrypt_file_obj = s3_resource.Object(decrypt_bucket_name, decrypt_file_key) - decrypt_file_obj.put(Body=decrypt_file) - decrypt_file.close + s3_client.put_object(Bucket=decrypt_bucket_name, Key=decrypt_file_key, Body=decrypt_file) + decrypt_file.close() logger.info('I-07-03 復号化ファイルをS3に出力しました') except Exception as e: logger.exception(f'E-07-01 復号化ファイルのS3出力に失敗しました　エラー内容：{e}') @@ -142,8 +140,7 @@ def handler(event, context): backup_file_key = f'{s3_event.data_source_name}/{execute_date}/{s3_event.file_name}' logger.info( f'I-07-04 PGP暗号化ファイル移動　移動元：{s3_event.bucket_name}/{s3_event.file_path}　移動先：{SAP_DATA_BACKUP_BUCKET_NAME}/{backup_file_key}') - backup_file_obj = s3_resource.Object(SAP_DATA_BACKUP_BUCKET_NAME, backup_file_key) - backup_file_obj.copy(copy_source) + s3_client.copy_object(Bucket=SAP_DATA_BACKUP_BUCKET_NAME, Key=backup_file_key, CopySource=copy_source) s3_client.delete_object(Bucket=s3_event.bucket_name, Key=s3_event.file_path) logger.info('I-07-05 PGP暗号化ファイルをバックアップ用バケットに移動しました') except Exception as e: @@ -158,8 +155,7 @@ def handler(event, context): import_file_folder = f'{s3_event.data_source_name}{DIRECTORY_TARGET}' import_file_key = f'{import_file_folder}{decrypt_file_name}' logger.info(f'I-07-06 復号化ファイル出力　ファイル名：{decrypt_file_name}　出力先：{s3_event.bucket_name}/{import_file_folder}') - import_file_obj = s3_resource.Object(s3_event.bucket_name, import_file_key) - import_file_obj.copy(copy_source) + s3_client.copy_object(Bucket=s3_event.bucket_name, Key=import_file_key, CopySource=copy_source) logger.info(f'I-07-07 復号化ファイルをS3に出力しました') except Exception as e: logger.exception(f'E-07-03 復号化ファイルのS3出力に失敗しました　エラー内容：{e}') @@ -186,8 +182,7 @@ def create_status_file(s3_event, extension) -> None: try: result_error_file_name = s3_event.file_name + extension result_error_key = s3_event.data_source_name + DIRECTORY_RECV + result_error_file_name - result_error_obj = s3_resource.Object(s3_event.bucket_name, result_error_key) - result_error_obj.put(Body='') + s3_client.put_object(Bucket=s3_event.bucket_name, Key=result_error_key, Body='') logger.error( f'E-ERR-01 recvディレクトリにエラーファイルを作成しました　ファイル名：{result_error_file_name}　出力先：{s3_event.bucket_name}/{result_error_key}') except Exception as e: @@ -204,8 +199,7 @@ def move_encrypt_file(s3_event) -> None: } error_file_name = f'{datetime.datetime.now():%Y%m%d%H%M%S}_{s3_event.file_name}' error_key = s3_event.data_source_name + DIRECTORY_RECV_ERROR + error_file_name - error_obj = s3_resource.Object(s3_event.bucket_name, error_key) - error_obj.copy(copy_source) + s3_client.copy_object(Bucket=s3_event.bucket_name, Key=error_key, CopySource=copy_source) s3_client.delete_object(Bucket=s3_event.bucket_name, Key=s3_event.file_path) logger.error( f'E-ERR-02 recv_errorディレクトリにファイルを移動しました　移動元：{s3_event.bucket_name}/{s3_event.file_path}　移動先：{s3_event.bucket_name}/{error_key}')