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 given 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")
@@ -71,12 +71,44 @@ def get_current_program(channel_id="skytg24"):
                     title = programme.find('title').text
                     desc = programme.find('desc').text
                     return {"title": title, "description": desc}
-        logger.info("No current program found for Sky TG24.")
+        logger.info(f"No current program found for channel {channel_id}.")
         return None
     except Exception as e:
         logger.error(f"Failed to parse XML file: {e}")
         return None
 
+# Function to dynamically create routes for each channel
+def create_channel_routes():
+    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 to create dynamic routes")
+        return
+
+    try:
+        tree = ET.parse(xml_file)
+        root = tree.getroot()
+
+        # Find all channels in the XML
+        channels = {channel.get('id'): channel.find('display-name').text.replace(" ", "").lower()
+                    for channel in root.findall('channel')}
+
+        for channel_id, route_name in channels.items():
+            # Create a dynamic route for each channel
+            create_route_for_channel(channel_id, route_name)
+
+    except Exception as e:
+        logger.error(f"Failed to create dynamic routes from XML file: {e}")
+
+# Helper function to create route for a specific channel
+def create_route_for_channel(channel_id, route_name):
+    @app.route(f'/{route_name}/now', methods=['GET'])
+    def current_program():
+        program_info = get_current_program(channel_id)
+        if program_info:
+            return jsonify(program_info)
+        else:
+            return f"No current program found for {route_name}.", 404
+
 # Flask route to serve the extracted file
 @app.route('/wltv', methods=['GET'])
 def serve_extracted_file():
@@ -120,16 +152,10 @@ def update_files():
     download_file(file_url, file_path)
     extract_gz_file(file_path, extract_to_dir)
 
-    return "Files updated", 200
+    # Create dynamic routes after updating
+    create_channel_routes()
 
-# Route to serve the current program on Sky TG24
-@app.route('/skytg24/now', methods=['GET'])
-def current_skytg24_program():
-    program_info = get_current_program("skytg24")
-    if program_info:
-        return jsonify(program_info)
-    else:
-        return "No current program found for Sky TG24.", 404
+    return "Files updated", 200
 
 if __name__ == '__main__':
     file_url = 'http://epg-guide.com/wltv.gz'
@@ -139,5 +165,8 @@ if __name__ == '__main__':
     download_file(file_url, file_path)
     extract_gz_file(file_path, extract_to_dir)
 
+    # Create dynamic routes based on the extracted XML data
+    create_channel_routes()
+
     # Start the Flask web server
     app.run(host='0.0.0.0', port=8080)