diff --git a/stepfunctions/TOOLS/convert_definition.py b/stepfunctions/TOOLS/convert_definition.py index eac0e088..aead3f47 100644 --- a/stepfunctions/TOOLS/convert_definition.py +++ b/stepfunctions/TOOLS/convert_definition.py @@ -25,40 +25,36 @@ class StateMachineTemplate(Template): super().__init__(template) def main(args=None): - try: - """ - args1: StepFunctionsステートマシン名 - args2: 変換先環境名(product or staging) - """ + """ + args1: StepFunctionsステートマシン名 + args2: 変換先環境名(product or staging) + """ - # カレントディレクトリ移動 - os.chdir(os.path.dirname(os.path.abspath(__file__))) + # カレントディレクトリ移動 + os.chdir(os.path.dirname(os.path.abspath(__file__))) - # 引数チェック - state_name, env_name = check_args(args) + # 引数チェック + state_name, env_name = check_args(args) - print('引数確認OK') + print('引数確認OK') - # ファイル存在チェック - check_file_exist(state_name) + # ファイル存在チェック + check_file_exist(state_name) - print('ファイル存在確認OK') - - # フォルダがなければ作る - env_name_folder = f'{DEFINITION_FOLDER_BASE}/{state_name}/{CONVERTED_FOLDER_BASE}/{env_name}' - if not os.path.isdir(env_name_folder): - os.makedirs(env_name_folder, exist_ok=True) - print(f'定義生成用フォルダを作成しました。フォルダ名:{env_name_folder}') + print('ファイル存在確認OK') + + # フォルダがなければ作る + env_name_folder = f'{DEFINITION_FOLDER_BASE}/{state_name}/{CONVERTED_FOLDER_BASE}/{env_name}' + if not os.path.isdir(env_name_folder): + os.makedirs(env_name_folder, exist_ok=True) + print(f'定義生成用フォルダを作成しました。フォルダ名:{env_name_folder}') - # 変換 - converted_file_name = convert_definition(state_name, env_name) + # 変換 + converted_file_name = convert_definition(state_name, env_name) - print(f'変換が完了しました ファイル名:{converted_file_name}') + print(f'変換が完了しました ファイル名:{converted_file_name}') - return - - except Exception as e: - raise Exception(e) + return def check_args(args): @@ -81,10 +77,11 @@ def check_length(args): def check_aws_environment(args): - if args[2] == STG_NAME or args[2] == PRD_NAME: - return + if args[2] not in [STG_NAME, PRD_NAME]: + raise Exception('第2引数が不正です') + + return - raise Exception('第2引数が不正です') def check_file_exist(state_name): @@ -116,7 +113,7 @@ def convert_definition(state_name, env_name): # 書き込みファイルオープン with open(f'{to_folder}/{state_name}.json', mode='w', - encoding=CHAR_CODE, newline='\n') as to_file: + encoding=CHAR_CODE, newline='\n') as to_file: to_file.write(substituted_state_json) return f'{to_folder}/{state_name}.json'