refactor: publish 메시지에 분석 성공/실패 코드 추가 - close #14

Author: ChaJunWoo-dev

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"

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