Add station selection feature to SetupWizard and scrape function

Co-authored-by: nmemmert <67961292+nmemmert@users.noreply.github.com>

journey_fm_app.py

@@ -59,6 +59,9 @@ class Config:
     def save_config(self, config):
         """Save config to QSettings"""
         for key, value in config.items():
+            # Handle list values (convert to comma-separated string)
+            if isinstance(value, list):
+                value = ','.join(value)
             self.set(key, value)
 
 class SetupWizard(QDialog):
@@ -109,6 +112,22 @@ class SetupWizard(QDialog):
         form_group.setLayout(form_layout)
         layout.addWidget(form_group)
 
+        # Station Selection
+        station_group = QGroupBox("Station Selection")
+        station_layout = QVBoxLayout()
+
+        station_layout.addWidget(QLabel("Select which stations to monitor:"))
+        self.journey_fm_checkbox = QCheckBox("Journey FM (myjourneyfm.com)")
+        self.journey_fm_checkbox.setChecked(True)
+        station_layout.addWidget(self.journey_fm_checkbox)
+
+        self.spirit_fm_checkbox = QCheckBox("Spirit FM (spiritfm.com)")
+        self.spirit_fm_checkbox.setChecked(True)
+        station_layout.addWidget(self.spirit_fm_checkbox)
+
+        station_group.setLayout(station_layout)
+        layout.addWidget(station_group)
+
         # Scheduling options
         sched_group = QGroupBox("Scheduling")
         sched_layout = QVBoxLayout()
@@ -147,14 +166,30 @@ class SetupWizard(QDialog):
         self.server_input.setText(config.get('SERVER_IP', ''))
         self.playlist_input.setText(config.get('PLAYLIST_NAME', 'Journey FM Recently Played'))
 
+        # Load station selections
+        selected_stations = config.get('SELECTED_STATIONS', 'journey_fm,spirit_fm')
+        if isinstance(selected_stations, str):
+            selected_stations = selected_stations.split(',')
+        else:
+            selected_stations = ['journey_fm', 'spirit_fm']
+        self.journey_fm_checkbox.setChecked('journey_fm' in selected_stations)
+        self.spirit_fm_checkbox.setChecked('spirit_fm' in selected_stations)
+
     def get_config(self):
+        selected_stations = []
+        if self.journey_fm_checkbox.isChecked():
+            selected_stations.append('journey_fm')
+        if self.spirit_fm_checkbox.isChecked():
+            selected_stations.append('spirit_fm')
+
         return {
             'PLEX_TOKEN': self.token_input.text().strip(),
             'SERVER_IP': self.server_input.text().strip(),
             'PLAYLIST_NAME': self.playlist_input.text().strip(),
             'AUTO_UPDATE': self.auto_update.isChecked(),
             'UPDATE_INTERVAL': self.interval_spin.value(),
-            'UPDATE_UNIT': self.interval_unit.currentText()
+            'UPDATE_UNIT': self.interval_unit.currentText(),
+            'SELECTED_STATIONS': ','.join(selected_stations)
         }
 
 class LogViewer(QWidget):

main.py

@@ -32,8 +32,11 @@ PLEX_TOKEN = 'pU-m3HWYUZU6iXJFhJyA'  # Your Plex.tv token
 SERVER_IP = '172.16.16.106'  # Your local server IP
 PLAYLIST_NAME = 'Journey FM Recently Played'
 
-def scrape_recently_played():
+def scrape_recently_played(selected_stations=None):
     """Scrape recently played songs from Journey FM and Spirit FM"""
+    if selected_stations is None:
+        selected_stations = ['journey_fm', 'spirit_fm']
+
     options = Options()
     options.add_argument("--headless=new")
     options.add_argument("--no-sandbox")
@@ -58,6 +61,7 @@ def scrape_recently_played():
     seen = set()
     
     # Journey FM - uses <strong> for title, normal text for artist
+    if 'journey_fm' in selected_stations:
         try:
             print("Scraping Journey FM...")
             driver.get('https://www.myjourneyfm.com/recently-played/')
@@ -98,6 +102,7 @@ def scrape_recently_played():
             print(f"Error scraping Journey FM: {e}")
     
     # Spirit FM - loads songs from iframe text file
+    if 'spirit_fm' in selected_stations:
         try:
             print("Scraping Spirit FM...")
             driver.get('https://spiritfm.com/ajax/now_playing_history.txt')
@@ -312,12 +317,17 @@ def setup_scheduler(params):
 def main():
     global PLEX_TOKEN, SERVER_IP, PLAYLIST_NAME
     # Load or prompt for config
+    selected_stations = ['journey_fm', 'spirit_fm']  # default
     if os.path.exists('config.json'):
         with open('config.json', 'r') as f:
             config = json.load(f)
         PLEX_TOKEN = config.get('PLEX_TOKEN', PLEX_TOKEN)
         SERVER_IP = config.get('SERVER_IP', SERVER_IP)
         PLAYLIST_NAME = config.get('PLAYLIST_NAME', PLAYLIST_NAME)
+        # Load selected stations
+        selected_stations_str = config.get('SELECTED_STATIONS', 'journey_fm,spirit_fm')
+        if isinstance(selected_stations_str, str):
+            selected_stations = selected_stations_str.split(',')
     else:
         # If no config file, use defaults (GUI should handle this)
         pass
@@ -328,8 +338,8 @@ def main():
     # Initialize history database
     init_history_db()
     
-    # Scrape songs
-    songs = scrape_recently_played()
+    # Scrape songs with selected stations
+    songs = scrape_recently_played(selected_stations)
     
     # Connect to Plex via MyPlexAccount
     account = MyPlexAccount(token=PLEX_TOKEN)