nik/newdwh2021
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: 早期return化、main関数は受け取った例外を投げているだけなのでtry-exceptをなくした

Browse Source
This commit is contained in:
shimoda.m@nds-tyo.co.jp 2022-09-21 08:44:39 +09:00
parent 34c35c81ce
commit 7898382624
1 changed files with 27 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

57
stepfunctions/TOOLS/convert_definition.py
View File

@ -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'