Update wltv_server.py

wltv_server.py

@@ -49,8 +49,8 @@ def extract_gz_file(gz_path, extract_to_dir):
     except Exception as e:
         logger.error(f"An error occurred while extracting the file: {e}")
 
-# Function to get the current program for Sky TG24
-def get_current_program(channel_id="skytg24"):
+# Function to get the current program for a channel
+def get_current_program(channel_id):
     xml_file = os.path.join(extract_to_dir, 'it_wltv_full.xml')
     if not os.path.exists(xml_file):
         logger.error("XML file does not exist for program data")
@@ -68,10 +68,10 @@ def get_current_program(channel_id="skytg24"):
                 start_time = datetime.strptime(programme.get('start')[:14], "%Y%m%d%H%M%S")
                 stop_time = datetime.strptime(programme.get('stop')[:14], "%Y%m%d%H%M%S")
                 if start_time <= current_time < stop_time:
-                    title = programme.find('title').text
-                    desc = programme.find('desc').text
+                    title = programme.find('title').text if programme.find('title') is not None else "No title"
+                    desc = programme.find('desc').text if programme.find('desc') is not None else "No description"
                     return {"title": title, "description": desc}
-        logger.info("No current program found for Sky TG24.")
+        logger.info(f"No current program found for {channel_id}.")
         return None
     except Exception as e:
         logger.error(f"Failed to parse XML file: {e}")
@@ -122,14 +122,14 @@ def update_files():
 
     return "Files updated", 200
 
-# Route to serve the current program on Sky TG24
-@app.route('/<id>/now', methods=['GET'])
-def current_skytg24_program(id):
-    program_info = get_current_program(id)
+# Route to serve the current program for a specific channel
+@app.route('/<channel_id>/now', methods=['GET'])
+def current_program(channel_id):
+    program_info = get_current_program(channel_id)
     if program_info:
         return jsonify(program_info)
     else:
-        return "No current program found for Sky TG24.", 404
+        return "No current program found for this channel.", 404
 
 if __name__ == '__main__':
     file_url = 'http://epg-guide.com/wltv.gz'