190 lines
11 KiB
Python
190 lines
11 KiB
Python
import os
|
||
import datetime
|
||
import boto3
|
||
import io
|
||
import re
|
||
import sys
|
||
import csv
|
||
|
||
# 環境変数
|
||
CHECK_BUCKET_NAME = os.environ["CHECK_BUCKET_NAME"]
|
||
CONFIG_BUCKET_NAME = os.environ["CONFIG_BUCKET_NAME"]
|
||
RECEIVE_DAILY_FILE_NAME_LIST_PATH = os.environ["RECEIVE_DAILY_FILE_NAME_LIST_PATH"]
|
||
NON_BUSINESS_DAY_LIST_PATH = os.environ["NON_BUSINESS_DAY_LIST_PATH"]
|
||
NOTICE_MAIL_TITLE_TEMPLATE_PATH = os.environ["NOTICE_MAIL_TITLE_TEMPLATE_PATH"]
|
||
NOTICE_MAIL_BODY_TEMPLATE_PATH = os.environ["NOTICE_MAIL_BODY_TEMPLATE_PATH"]
|
||
MBJ_SAP_NOTICE_TOPIC = os.environ["MBJ_SAP_NOTICE_TOPIC"]
|
||
MAIL_BODY_REPLACE_SYMBOL = os.environ["MAIL_BODY_REPLACE_SYMBOL"]
|
||
NDS_NOTICE_TOPIC = os.environ["NDS_NOTICE_TOPIC"]
|
||
NDS_NOTICE_TITLE = os.environ["NDS_NOTICE_TITLE"]
|
||
|
||
# 定数
|
||
ROW_COMMENT_SYMBOL = '#'
|
||
LOG_INFO = 'Info'
|
||
LOG_ERROR = 'Error'
|
||
INDEX_REGEX = 0
|
||
INDEX_DATA_NAME = 1
|
||
INDEX_ROW_COMMENT_SYMBOL = 0
|
||
INDEX_SPLIT_NUM = 1
|
||
INDEX_LAST = -1
|
||
|
||
# 変数
|
||
s3_client = boto3.client('s3')
|
||
s3_resource = boto3.resource('s3')
|
||
sns_client = boto3.client('sns')
|
||
|
||
|
||
def lambda_handler(event, context):
|
||
try:
|
||
# ① 処理開始ログを出力する
|
||
print_log(LOG_INFO, 'I-01-01', '処理開始 SAP_supデータ受領チェック処理（日次）')
|
||
execute_date = datetime.date.today().strftime('%Y/%m/%d')
|
||
print_log(LOG_INFO, 'I-01-02', f'処理稼働日：{execute_date}')
|
||
mail_msg = ''
|
||
|
||
# ② 営業日チェック処理を行う
|
||
print_log(LOG_INFO, 'I-02-01', '営業日チェック処理開始')
|
||
|
||
# 1.設定ファイル[メルク社非営業日設定ファイル]を読み込む
|
||
try:
|
||
print_log(LOG_INFO, 'I-02-02', f'非営業日設定ファイル読込 読込元：{CONFIG_BUCKET_NAME}/{NON_BUSINESS_DAY_LIST_PATH}')
|
||
non_business_day_obj = s3_resource.Object(CONFIG_BUCKET_NAME, NON_BUSINESS_DAY_LIST_PATH)
|
||
non_business_day_response = non_business_day_obj.get()
|
||
print_log(LOG_INFO, 'I-02-03', '非営業日設定ファイルを読み込みました')
|
||
except Exception as e:
|
||
print_log(LOG_ERROR, 'E-02-01', f'非営業日設定ファイルの読み込みに失敗しました エラー内容：{e}')
|
||
error_exit('E-02-01', e)
|
||
|
||
# 2.処理稼働日が「②１．」で読み込んだ「メルク社非営業日設定ファイル」に存在するか確認する
|
||
try:
|
||
print_log(LOG_INFO, 'I-02-04', f'本日が非営業日かチェックします チェック日：{execute_date}')
|
||
none_business_day_list = []
|
||
for i, row in enumerate(io.TextIOWrapper(io.BytesIO(non_business_day_response["Body"].read()), encoding='utf-8'), 1):
|
||
if row[INDEX_ROW_COMMENT_SYMBOL] == ROW_COMMENT_SYMBOL:
|
||
continue
|
||
non_date = datetime.datetime.strptime(row.rstrip('\n'), "%Y/%m/%d").strftime('%Y/%m/%d')
|
||
none_business_day_list .append(non_date)
|
||
if execute_date in none_business_day_list:
|
||
print_log(LOG_INFO, 'I-02-05', '本日は非営業日のため、チェック処理をスキップします')
|
||
sys.exit()
|
||
else:
|
||
print_log(LOG_INFO, 'I-02-06', '本日は営業日のため、チェック処理を実施します')
|
||
except Exception as e:
|
||
print_log(LOG_ERROR, 'E-02-02', f'メルク社非営業日設定ファイルに不備があります 行数：{i} エラー内容：{e}')
|
||
error_exit('E-02-02', e)
|
||
|
||
# ③ 設定ファイル[SAP_supI/Fファイルネーム設定ファイル（日次）]を読み込む
|
||
try:
|
||
print_log(LOG_INFO, 'I-03-01', f'日次I/Fファイルネーム設定ファイル読込 読込元：{CONFIG_BUCKET_NAME}/{RECEIVE_DAILY_FILE_NAME_LIST_PATH}')
|
||
receive_daily_file_name_obj = s3_resource.Object(CONFIG_BUCKET_NAME, RECEIVE_DAILY_FILE_NAME_LIST_PATH)
|
||
receive_daily_file_name_response = receive_daily_file_name_obj.get()
|
||
print_log(LOG_INFO, 'I-03-02', '日次I/Fファイルネーム設定ファイルを読み込みました')
|
||
except Exception as e:
|
||
print_log(LOG_ERROR, 'E-03-01', f'日次I/Fファイルネーム設定ファイルの読み込みに失敗しました エラー内容：{e}')
|
||
error_exit('E-03-01', e)
|
||
|
||
# ④ 日次チェック処理を行う
|
||
print_log(LOG_INFO, 'I-04-01', '日次チェック処理開始')
|
||
|
||
# 1.SAP保管用バケットの処理稼働日に該当するサブフォルダにあるファイル一覧を取得する
|
||
print_log(LOG_INFO, 'I-04-02', f'オブジェクトリストの取得 取得先：{CHECK_BUCKET_NAME}/{execute_date}/')
|
||
object_prefix = f'{execute_date}/'
|
||
object_list = s3_resource.Bucket(CHECK_BUCKET_NAME).objects.filter(Prefix=object_prefix)
|
||
|
||
# 2.日次I/Fファイルチェック処理
|
||
print_log(LOG_INFO, 'I-04-03', '日次I/Fファイルチェック処理開始')
|
||
print_log(LOG_INFO, 'I-04-04', '取得したオブジェクトリストと日次I/Fファイルネーム設定ファイルの突き合わせを開始します')
|
||
row_count = 0
|
||
match_count = 0
|
||
receive_daily_file_name_body = io.TextIOWrapper(io.BytesIO(receive_daily_file_name_response["Body"].read()), encoding='utf-8')
|
||
for i, row in enumerate(csv.reader(receive_daily_file_name_body, delimiter='\t'), 1):
|
||
row_count = i
|
||
file_exists = False
|
||
for object in object_list:
|
||
object_key = object.key.rsplit('/', INDEX_SPLIT_NUM)
|
||
file_name = object_key[INDEX_LAST]
|
||
match_result = re.fullmatch(row[INDEX_REGEX], file_name)
|
||
if match_result is not None:
|
||
file_exists = True
|
||
break
|
||
if file_exists == True:
|
||
match_count += 1
|
||
print_log(LOG_INFO, 'I-04-05', f'日次I/Fファイルの受領を確認しました ファイル名：{file_name}')
|
||
else:
|
||
print_log(LOG_ERROR, 'E-04-01', f'日次I/Fファイルに不足があります ファイル名：{row[INDEX_DATA_NAME]}')
|
||
mail_msg = f'{mail_msg} {row[INDEX_DATA_NAME]}\n'
|
||
if row_count == match_count:
|
||
print_log(LOG_INFO, 'I-04-06', '日次I/Fファイルは全て受領していることを確認しました')
|
||
|
||
# ⑤ 「①」でメモリ保持しているメール挿入用文言に出力内容が存在するか確認する
|
||
print_log(LOG_INFO, 'I-05-01', 'メール送信処理開始')
|
||
|
||
if len(mail_msg) > 0:
|
||
# 1.存在した場合
|
||
print_log(LOG_INFO, 'I-05-02', f'{execute_date} 日次I/Fファイルに不足があるため、メール送信処理を開始します')
|
||
|
||
try:
|
||
print_log(LOG_INFO, 'I-05-03', f'通知メール（タイトル）テンプレートファイル読込 読込元：{CONFIG_BUCKET_NAME}/{NOTICE_MAIL_TITLE_TEMPLATE_PATH}')
|
||
mail_title_obj = s3_client.get_object(Bucket=CONFIG_BUCKET_NAME, Key=NOTICE_MAIL_TITLE_TEMPLATE_PATH)
|
||
mail_title = mail_title_obj['Body'].read().decode('utf-8')
|
||
print_log(LOG_INFO, 'I-05-04', '通知メール（タイトル）テンプレートファイルを読み込みました')
|
||
except Exception as e:
|
||
print_log(LOG_ERROR, 'E-05-01', f'通知メール（タイトル）テンプレートファイルの読み込みに失敗しました エラー内容：{e}')
|
||
error_exit('E-05-01', e)
|
||
|
||
try:
|
||
print_log(LOG_INFO, 'I-05-05', f'通知メール（本文）テンプレートファイル読込 読込元：{CONFIG_BUCKET_NAME}/{NOTICE_MAIL_BODY_TEMPLATE_PATH}')
|
||
mail_body_obj = s3_client.get_object(Bucket=CONFIG_BUCKET_NAME, Key=NOTICE_MAIL_BODY_TEMPLATE_PATH)
|
||
mail_body_response = mail_body_obj['Body'].read().decode('utf-8')
|
||
mail_body = mail_body_response.replace(MAIL_BODY_REPLACE_SYMBOL, mail_msg)
|
||
print_log(LOG_INFO, 'I-05-06', '通知メール（本文）テンプレートファイルを読み込みました')
|
||
except Exception as e:
|
||
print_log(LOG_ERROR, 'E-05-02', f'通知メール（本文）テンプレートファイルの読み込みに失敗しました エラー内容：{e}')
|
||
error_exit('E-05-02', e)
|
||
|
||
print_log(LOG_INFO, 'I-05-07', f'メール送信指示をします 送信先トピック：{MBJ_SAP_NOTICE_TOPIC}')
|
||
params = {
|
||
'TopicArn': MBJ_SAP_NOTICE_TOPIC,
|
||
'Subject': mail_title,
|
||
'Message': mail_body
|
||
}
|
||
sns_client.publish(**params)
|
||
print_log(LOG_INFO, 'I-05-08', 'メール送信指示をしました')
|
||
else:
|
||
# 2.存在しない場合
|
||
print_log(LOG_INFO, 'I-05-09', f'{execute_date} 日次I/Fファイルに不足がなかったため、メール送信処理をスキップします')
|
||
|
||
# ⑥ 処理終了ログを出力する
|
||
print_log(LOG_INFO, 'I-06-01', '処理終了 SAP_supデータ受領チェック処理（日次）')
|
||
except Exception as e:
|
||
print_log(LOG_ERROR, 'E-99', f'想定外のエラーが発生しました エラー内容：{e}')
|
||
error_exit('E-99', e)
|
||
|
||
|
||
def print_log(log_level, log_id, msg):
|
||
try:
|
||
print(f'{datetime.datetime.now():%Y-%m-%d %H:%M:%S} {log_level} {log_id} {msg}')
|
||
except Exception as e:
|
||
print(f'{datetime.datetime.now():%Y-%m-%d %H:%M:%S} Error E-99 想定外のエラーが発生しました エラー内容：{e}')
|
||
error_exit('E-99', e)
|
||
|
||
|
||
def error_exit(error_log_id, exception):
|
||
try:
|
||
print(f'{datetime.datetime.now():%Y-%m-%d %H:%M:%S} Error E-ERR-01 エラー処理開始')
|
||
|
||
print(f'{datetime.datetime.now():%Y-%m-%d %H:%M:%S} Error E-ERR-02 処理異常通知の送信指示をしました 通知先トピック：{NDS_NOTICE_TOPIC}')
|
||
error_msg = f'{error_log_id} のエラーが発生しました。ご確認ください\n詳細：{exception}'
|
||
params = {
|
||
'TopicArn': NDS_NOTICE_TOPIC,
|
||
'Subject': NDS_NOTICE_TITLE,
|
||
'Message': error_msg
|
||
}
|
||
sns_client.publish(**params)
|
||
|
||
print(f'{datetime.datetime.now():%Y-%m-%d %H:%M:%S} Error E-ERR-03 エラー終了 処理を中断します')
|
||
sys.exit()
|
||
except Exception as e:
|
||
print(f'{datetime.datetime.now():%Y-%m-%d %H:%M:%S} Error E-99 想定外のエラーが発生しました エラー内容：{e}')
|
||
sys.exit()
|