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 a channel
-def get_current_program(channel_id):
+# Function to get the current program for Sky TG24
+def get_current_program(channel_id="skytg24"):
     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")
@@ -61,30 +61,75 @@ def get_current_program(channel_id):
         root = tree.getroot()
 
         # Get the current time
-        current_time = datetime.now().strftime("%Y%m%d%H%M%S")
+        current_time = datetime.now()
 
         for programme in root.findall('programme'):
             if programme.get('channel') == channel_id:
                 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 <= datetime.now() < stop_time:
-                    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"
+                if start_time <= current_time < stop_time:
+                    title = programme.find('title').text
+                    desc = programme.find('desc').text
                     return {"title": title, "description": desc}
-        logger.info(f"No current program found for {channel_id}.")
+        logger.info("No current program found for Sky TG24.")
         return None
     except Exception as e:
         logger.error(f"Failed to parse XML file: {e}")
         return None
 
-# Route to serve the current program on any channel
-@app.route('/<channel_id>/now', methods=['GET'])
-def current_program(channel_id):
-    program_info = get_current_program(channel_id)
+# Flask route to serve the extracted file
+@app.route('/wltv', methods=['GET'])
+def serve_extracted_file():
+    try:
+        extracted_files = os.listdir(extract_to_dir)
+        logger.info(f"Files in 'extracted' directory: {extracted_files}")
+        for file_name in extracted_files:
+            logger.info(f"File: {file_name}, Extension: {os.path.splitext(file_name)[1]}")
+    except Exception as e:
+        logger.error(f"Error listing files in 'extracted' directory: {e}")
+        return "Error listing files", 500
+    
+    file_name = 'it_wltv_full.xml'  # Updated file name with extension
+    file_path = os.path.join(extract_to_dir, file_name)
+    
+    if os.path.exists(file_path):
+        logger.info(f"Serving file from {file_path}")
+        return send_file(file_path, as_attachment=False, mimetype='text/plain')
+    else:
+        logger.error("File not found")
+        return "File not found", 404
+
+# Route to fetch logs
+@app.route('/logs', methods=['GET'])
+def get_logs():
+    try:
+        with open('app.log', 'r') as log_file:
+            logs = log_file.read()
+        return logs, 200
+    except Exception as e:
+        logger.error(f"Failed to read logs: {e}")
+        return jsonify({"error": "Failed to read logs"}), 500
+
+# Route to update (download and extract again)
+@app.route('/update', methods=['POST'])
+def update_files():
+    file_url = 'http://epg-guide.com/wltv.gz'
+    file_path = '/tmp/it_wltv_full.gz'
+
+    # Download and extract the gz file
+    download_file(file_url, file_path)
+    extract_gz_file(file_path, extract_to_dir)
+
+    return "Files updated", 200
+
+# 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.", 404
+        return "No current program found for Sky TG24.", 404
 
 if __name__ == '__main__':
     file_url = 'http://epg-guide.com/wltv.gz'