From 78983826244cf76fc2f00d72b2cf4aa2ebb5b63d Mon Sep 17 00:00:00 2001
From: "shimoda.m@nds-tyo.co.jp" <shimoda.m@nds-tyo.co.jp>
Date: Wed, 21 Sep 2022 08:44:39 +0900
Subject: [PATCH] =?UTF-8?q?refactor:=20=E6=97=A9=E6=9C=9Freturn=E5=8C=96?=
 =?UTF-8?q?=E3=80=81main=E9=96=A2=E6=95=B0=E3=81=AF=E5=8F=97=E3=81=91?=
 =?UTF-8?q?=E5=8F=96=E3=81=A3=E3=81=9F=E4=BE=8B=E5=A4=96=E3=82=92=E6=8A=95?=
 =?UTF-8?q?=E3=81=92=E3=81=A6=E3=81=84=E3=82=8B=E3=81=A0=E3=81=91=E3=81=AA?=
 =?UTF-8?q?=E3=81=AE=E3=81=A7try-except=E3=82=92=E3=81=AA=E3=81=8F?=
 =?UTF-8?q?=E3=81=97=E3=81=9F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 stepfunctions/TOOLS/convert_definition.py | 57 +++++++++++------------
 1 file changed, 27 insertions(+), 30 deletions(-)

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'