cantamen seat attributes fix (#172)

mobidata-bw · Oct 17, 2024 · c2d3f2a · c2d3f2a
1 parent 9ef9ecd
commit c2d3f2a
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/x2gbfs/providers/cantamen.py b/x2gbfs/providers/cantamen.py
@@ -122,9 +122,15 @@ def _parse_attributes(self, attributes):
             if lowercasedClass.startswith('seats'):
                 # seats are encoded differently per provider: SEATSX, seats_x
                 seatsClass = lowercasedClass.replace('_', '')
-                numberOfSeats_str = seatsClass[len('seats') :]
-                numberOfSeats = 7 if numberOfSeats_str == '5plus2' else int(numberOfSeats_str)
-                self.seats[attributeId] = numberOfSeats
+                numberOfSeats = seatsClass[len('seats') :]
+                if numberOfSeats == '5plus2':
+                    self.seats[attributeId] = 7
+                elif numberOfSeats == '6bus':
+                    self.seats[attributeId] = 6
+                elif numberOfSeats.isdigit():
+                    self.seats[attributeId] = int(numberOfSeats)
+                else:
+                    logger.info(f'Seat attribute {attribute["Class"]} is unknown and will be ignored')
 
     def _all_bookees(self) -> Generator[Dict[str, Any], None, None]:
         bookees = self._load_response()['Bookee']