90 lines
3.7 KiB
Python
90 lines
3.7 KiB
Python
"""
|
||
Viewセキュリティオプション付与チェック用Lambda関数のエントリーポイント
|
||
"""
|
||
|
||
import json
|
||
|
||
import botocore
|
||
|
||
from aws.s3 import ConfigBucket
|
||
from aws.ssm import SSMParameterStore
|
||
from constants import (CHECK_TARGET_SCHEMAS, RESPONSE_CODE_NO_SUCH_KEY,
|
||
RESPONSE_CODE_PARAMETER_NOT_FOUND, RESPONSE_ERROR,
|
||
RESPONSE_ERROR_CODE)
|
||
from exceptions import (FileNotFoundException, MeDaCaException,
|
||
ParameterNotFoundException)
|
||
from medaca_logger import MeDaCaLogger
|
||
|
||
|
||
def handler(event, context):
|
||
logger = MeDaCaLogger.get_logger()
|
||
try:
|
||
logger.info('I-01-01', '処理開始 Viewセキュリティオプション付与チェック')
|
||
logger.info('I-02-02', 'チェック対象スキーマ名ファイルを読み込み 開始')
|
||
check_target_schemas = read_check_target_schemas()
|
||
logger.info('I-02-02', f'チェック対象スキーマ名ファイルを読み込み 終了 チェック対象スキーマ名：{check_target_schemas}')
|
||
# print(check_target_schemas)
|
||
logger.info('I-03-01', 'データベースへの接続開始 開始')
|
||
# DB接続のためのパラメータ取得
|
||
db_host, db_user_name, db_user_password = read_db_param_from_parameter_store()
|
||
# print(db_host, db_user_name, db_user_password)
|
||
logger.info('I-03-01', 'データベースへの接続開始 終了')
|
||
|
||
except MeDaCaException as e:
|
||
logger.exception(e.error_id, e)
|
||
raise e
|
||
except Exception as e:
|
||
logger.exception('E-99', f'想定外のエラーが発生しました エラー内容：{e}')
|
||
raise e
|
||
finally:
|
||
logger.info('I-06-01', '処理終了 Viewセキュリティオプション付与チェック')
|
||
|
||
|
||
def read_check_target_schemas() -> list:
|
||
"""設定ファイル[チェック対象スキーマ名ファイル]を読み込む
|
||
|
||
Raises:
|
||
exceptions.FileNotFoundException: ファイルが読み込めなかったエラー
|
||
Exception: 想定外のエラー
|
||
|
||
Returns:
|
||
list: チェック対象のスキーマ名のリスト
|
||
"""
|
||
try:
|
||
config_bucket = ConfigBucket()
|
||
check_target_schema_names = config_bucket.check_target_schema_names()
|
||
return json.loads(check_target_schema_names)[CHECK_TARGET_SCHEMAS]
|
||
except botocore.exceptions.ClientError as e:
|
||
if e.response[RESPONSE_ERROR][RESPONSE_ERROR_CODE] == RESPONSE_CODE_NO_SUCH_KEY:
|
||
raise FileNotFoundException('E-02-01', f'チェック対象スキーマ名ファイルの読み込みに失敗しました エラー内容：{e}')
|
||
else:
|
||
raise Exception(e)
|
||
|
||
|
||
def read_db_param_from_parameter_store() -> tuple:
|
||
"""パラメータストアからDB接続情報を取得する
|
||
|
||
Raises:
|
||
FileNotFoundException: _description_
|
||
Exception: 想定外のエラー
|
||
|
||
Returns:
|
||
tuple: DB接続情報
|
||
"""
|
||
try:
|
||
parameter_store = SSMParameterStore()
|
||
db_host = parameter_store.db_host()
|
||
db_user_name = parameter_store.db_user_name()
|
||
db_user_password = parameter_store.db_user_password()
|
||
return db_host, db_user_name, db_user_password
|
||
except botocore.exceptions.ClientError as e:
|
||
if e.response[RESPONSE_ERROR][RESPONSE_ERROR_CODE] == RESPONSE_CODE_PARAMETER_NOT_FOUND:
|
||
raise ParameterNotFoundException('E-03-02', f'パラメータストアの取得に失敗しました エラー内容：{e}')
|
||
else:
|
||
raise Exception(e)
|
||
|
||
|
||
# ローカル実行用
|
||
if __name__ == '__main__':
|
||
handler({}, {})
|