diff --git a/src/config/constants.py b/src/config/constants.py index cbeafa4..359f9fc 100644 --- a/src/config/constants.py +++ b/src/config/constants.py @@ -16,3 +16,9 @@ class ERROR: FAILED_CREATE_SUBTITLE = "자막 파일 생성에 실패했습니다." FAILED_CONNECT_CHANNEL = "채널 연결에 실패했습니다." SERVER_ERROR = "서버 오류가 발생했습니다." + +class CODE: + class SUCCESSS: + ANALYZE = "SUCCESS_ANALYZE" + class FAILED: + ANALYZE = "FAILED_ANALYZE" diff --git a/src/rabbitmq/consume.py b/src/rabbitmq/consume.py index 3fbacae..312b042 100644 --- a/src/rabbitmq/consume.py +++ b/src/rabbitmq/consume.py @@ -2,7 +2,7 @@ import os from analyze_video import analyze_frame -from config.constants import MESSAGES +from config.constants import MESSAGES, CODE from gcs.generate_signed_url import generate_signed_url from gcs.read import get_video from gcs.write import upload_video @@ -48,6 +48,7 @@ def process_message(body): signed_url = generate_signed_url(file_name=file_name) message = { "email": email, + "code": CODE.SUCCESSS.ANALYZE, "message": MESSAGES.SUCCESS.ANALYZE, "url": signed_url } @@ -57,8 +58,9 @@ def process_message(body): except Exception as err: message = { "email": email, + "code": CODE.FAILED.ANALYZE, "message": str(err), - "url": "" + "url": None } return message