Fix passenger info

ryanking13 · Aug 26, 2024 · 9c12b51 · 9c12b51
1 parent 92383b3
commit 9c12b51
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 19 deletions.
diff --git a/SRT/passenger.py b/SRT/passenger.py
@@ -77,21 +77,22 @@ def get_passenger_dict(passengers, special_seat=False, window_seat=None):
             "totPrnb": Passenger.total_count(passengers),
             "psgGridcnt": str(len(passengers)),
         }
-        for i, passenger in enumerate(passengers):
-            data[f"psgTpCd{i + 1}"] = passenger.type_code
-            data[f"psgInfoPerPrnb{i + 1}"] = str(passenger.count)
+        for _, passenger in enumerate(passengers):
+            code = passenger.type_code
+            data[f"psgTpCd{code}"] = passenger.type_code
+            data[f"psgInfoPerPrnb{code}"] = str(passenger.count)
             # seat location ('000': 기본, '012': 창측, '013': 복도측)
-            data[f"locSeatAttCd{i + 1}"] = WINDOW_SEAT[window_seat]
+            data[f"locSeatAttCd{code}"] = WINDOW_SEAT[window_seat]
             # seat requirement ('015': 일반, '021': 휠체어)
             # TODO: 선택 가능하게
-            data[f"rqSeatAttCd{i + 1}"] = "015"
+            data[f"rqSeatAttCd{code}"] = "015"
             # seat direction ('009': 정방향)
-            data[f"dirSeatAttCd{i + 1}"] = "009"
+            data[f"dirSeatAttCd{code}"] = "009"
 
-            data[f"smkSeatAttCd{i + 1}"] = "000"
-            data[f"etcSeatAttCd{i + 1}"] = "000"
+            data[f"smkSeatAttCd{code}"] = "000"
+            data[f"etcSeatAttCd{code}"] = "000"
             # seat type: ('1': 일반실, '2': 특실)
-            data[f"psrmClCd{i + 1}"] = "2" if special_seat else "1"
+            data[f"psrmClCd{code}"] = "2" if special_seat else "1"
 
         return data
 

diff --git a/SRT/srt.py b/SRT/srt.py
@@ -395,10 +395,6 @@ def _reserve(
             "dirSeatAttCd1": "009",  # 방향좌석속성코드
             "locSeatAttCd1": "000",  # 위치좌석속성코드1
             "rqSeatAttCd1": "015",  # 요구좌석속성코드1
-            "etcSeatAttCd1": "000",  # 기타좌석속성코드1
-            "smkSeatAttCd2": "000",  # 흡연좌석속성코드2
-            "dirSeatAttCd2": "009",  # 방향좌석속성코드2
-            "rqSeatAttCd2": "015",  # 요구좌석속성코드2
             "mblPhone": mblPhone,
         }
 
@@ -410,13 +406,11 @@ def _reserve(
                 }
             )
 
-        # jobid가 RESERVE_JOBID["PERSONAL"]일 경우, data에 windowSeat 추가
-        if jobid == RESERVE_JOBID["PERSONAL"]:
-            data.update(
-                Passenger.get_passenger_dict(
-                    passengers, special_seat=is_special_seat, window_seat=window_seat
-                )
+        data.update(
+            Passenger.get_passenger_dict(
+                passengers, special_seat=is_special_seat, window_seat=window_seat
             )
+        )
 
         r = self._session.post(url=url, data=data)
         parser = SRTResponseData(r.text)