better filtering by station_ids

app.py

@@ -22,6 +22,7 @@ logger = logging.getLogger(__name__)
 
 
 flags.DEFINE_string('station_id', None, "Station Id")
+flags.DEFINE_list('station_ids', [], 'Station IDs')
 flags.DEFINE_string('file', None, "File where to save the history")
 
 
@@ -123,11 +124,19 @@ def on_data_handler(packet_logger: PacketLogger):
         # noinspection PyBroadException
         try:
             kr_packet = KrPacketV2.load(message, unknown=EXCLUDE)
-            station_id = flags.FLAGS.station_id
-            if station_id is None or \
-                    station_id.lower() == kr_packet.station_id.lower():
-                logger.info(f"{kr_packet}")
-                packet_logger.log(kr_packet)
+            station_ids = flags.FLAGS.station_ids
+            if len(station_ids) > 0:
+                for station_id in station_ids:
+                    if station_id is None or \
+                            station_id.lower() == kr_packet.station_id.lower():
+                        logger.info(f"{kr_packet}")
+                        packet_logger.log(kr_packet)
+            else:
+                station_id = flags.FLAGS.station_id
+                if station_id is None or \
+                        station_id.lower() == kr_packet.station_id.lower():
+                    logger.info(f"{kr_packet}")
+                    packet_logger.log(kr_packet)
         except Exception:
             logging.error("handle_message: Error!", exc_info=sys.exc_info())