Modify/#117 getcategory

tour/consumers.py

@@ -14,79 +14,76 @@ async def connect(self):
           - user_id: string (required)
           - areaCode: string (required)
           - sigunguName: string (required)
-          - contentTypeId: string (required)
+          - categoryName: string (required)
         """
-        query_string = self.scope['query_string'].decode() # 쿼리 스트링을 불러들입니다.
-        params = urllib.parse.parse_qs(query_string) # 쿼리 스트링을 파라미터로 변환합니다.
-        self.user_id = params.pop('user_id', [None])[0] # user 고유 sub를 가져옵니다.
-        self.unique_code = params.pop('unique_code', [""])[0] # 웹소켓 통신을 위한 고유 번호를 가져옵니다.
-        days = params.pop('days', [None])[0] # 여행 기간을 의미합니다.
+        query_string = self.scope['query_string'].decode()
+        params = urllib.parse.parse_qs(query_string)
+        self.user_id = params.pop('user_id', [None])[0]
+        self.unique_code = params.pop('unique_code', [""])[0]
         self.user_id = self.user_id + '_' + self.unique_code if self.unique_code != "" else self.user_id
 
         if self.user_id is None:
             await self.close()
             return
 
-        # 웹소켓 그룹에 가입
         logger.info(f'channel_id: {self.user_id} 웹소켓 가입')
-        await self.channel_layer.group_add(self.user_id, self.channel_name) # user_id를 그룹 이름으로 하고 웹소켓에 가입합니다.
-        await self.accept() # 웹소켓 연결
+        await self.channel_layer.group_add(self.user_id, self.channel_name)
+        await self.accept()
 
-        # 요청을 celery task로 보냅니다.
-        areaCode = params.pop('areaCode', [None])[0] # area_code 가져옴
-        sigunguName = params.pop('sigunguName', [None])[0] # 시군구 이름 가져옴
-        if areaCode is None or days is None: # areaCode가 존재하지 않는다면
+        areaCode = params.pop('areaCode', [None])[0]
+        sigunguName = params.pop('sigunguName', [None])[0]
+        categoryName = params.pop('categoryName', [None])[0]
+
+        if areaCode is None or categoryName is None:
             await self.send(text_data=json.dumps({
                 'state': 'ERROR',
                 'Message': '필수 파라미터 중 일부가 없습니다.'
             }, ensure_ascii=False))
             return
+
         tour = TourApi(MobileOS=MobileOS.ANDROID, MobileApp='AlphaProject2025', service_key=PUBLIC_DATA_PORTAL_API_KEY)
-        sigunguCode = None
         sigunguCodes = None
         if sigunguName is not None:
             sigunguNames = sigunguName.split(',')
             sigunguCodes = []
             for each in sigunguNames:
-                sigunguCode = tour.get_sigungu_code(areaCode, each) # 시군구 이름에 대응되는 코드를 가져옵니다.
-                if sigunguCode is None: # 시군구 코드가 없다면
+                sigunguCode = tour.get_sigungu_code(areaCode, each)
+                if sigunguCode is None:
                     await self.send(text_data=json.dumps({
                         'state': 'ERROR',
                         'Message': '해당 시군구 이름에 대응되는 코드를 가져올 수 없습니다. 시군구 이름을 다시 한번 확인 바랍니다.'
                     }, ensure_ascii=False))
                     return
                 sigunguCodes.append(sigunguCode)
 
-        task_result = app.send_task('tour.tasks.get_recommended_tour_based_area', args=[self.user_id, # 채널 레이어 그룹 특정을 위해 보냅니다.
-                                                                                        areaCode, days, Arrange.TITLE_IMAGE.value, sigunguCodes])
+        task_result = app.send_task(
+            'tour.tasks.get_recommended_place_by_category_task',
+            args=[self.user_id, areaCode, categoryName, sigunguCodes, Arrange.TITLE_IMAGE.value]
+        )
+
         await self.send(text_data=json.dumps({
             'state': 'OK',
             'Message': {
                 'task_id': task_result.task_id,
             }
         }))
 
-
     async def disconnect(self, close_code):
         await self.channel_layer.group_discard(self.user_id, self.channel_name)
 
     async def task_update(self, event):
-        # celery 컨테이너에서 보낸 메시지를 클라이언트로 전송
         await self.send(text_data=json.dumps(event["message"], ensure_ascii=False))
 
     async def receive(self, text_data=None, bytes_data=None):
-        """
-        재시도를 위한 메시지 입니다.
-        """
         data = json.loads(text_data)
         user_id = data.get("user_id", None)
         areaCode = data.get("areaCode", None)
         sigunguName = data.get("sigunguName", None)
-        unique_code = data.get('unique_code', "")  # 웹소켓 통신을 위한 고유 번호를 가져옵니다.
+        categoryName = data.get("categoryName", None)
+        unique_code = data.get('unique_code', "")
         user_id = user_id + '_' + unique_code
-        days = data.get("days", None)
-        if user_id is None or areaCode is None or days is None:
-            # 데이터가 없다면 예외 처리
+
+        if user_id is None or areaCode is None or categoryName is None:
             await self.send(text_data=json.dumps({
                 'state': 'ERROR',
                 'Message': '필수 파라미터 중 일부가 없거나 잘못되었습니다.'
@@ -99,21 +96,20 @@ async def receive(self, text_data=None, bytes_data=None):
             sigunguCodes = []
             sigunguNames = sigunguName.split(',')
             for each in sigunguNames:
-                sigunguCode = tour.get_sigungu_code(areaCode, each)  # 시군구 이름에 대응되는 코드를 가져옵니다.
-                if sigunguCode is None:  # 시군구 코드가 없다면
+                sigunguCode = tour.get_sigungu_code(areaCode, each)
+                if sigunguCode is None:
                     await self.send(text_data=json.dumps({
                         'state': 'ERROR',
                         'Message': '해당 시군구 이름에 대응되는 코드를 가져올 수 없습니다. 시군구 이름을 다시 한번 확인 바랍니다.'
                     }, ensure_ascii=False))
                     return
                 sigunguCodes.append(sigunguCode)
 
-        task_result = app.send_task('tour.tasks.get_recommended_tour_based_area',
-                                    args=[self.user_id,  # 채널 레이어 그룹 특정을 위해 보냅니다.
-                                          areaCode, Arrange.TITLE_IMAGE.value, sigunguCodes])
+        task_result = app.send_task('tour.tasks.get_recommended_place_by_category_task',
+                                    args=[user_id, areaCode, categoryName, sigunguCodes, Arrange.TITLE_IMAGE.value])
         await self.send(text_data=json.dumps({
             'state': 'OK',
             'Message': {
                 'task_id': task_result.task_id,
             }
-        }))
+        }))