Update wltv_server.py

wltv_server.py

@@ -2,21 +2,16 @@ import os
 import requests
 import zipfile
 import logging
-from flask import Flask, send_file, Response, jsonify
-from io import StringIO  # For Python 3
+from flask import Flask, send_file, jsonify
 
 # Set up logging
-log_stream = StringIO()
-logging.basicConfig(level=logging.INFO, stream=log_stream, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
 
 # Ensure the extracted directory exists
 extract_to_dir = 'extracted'
-zip_path = 'it_wltv_full.zip'
-zip_url = 'http://api.epg-guide.com/wltv.dz'
-
 if not os.path.exists(extract_to_dir):
     os.makedirs(extract_to_dir, exist_ok=True)
 
@@ -34,6 +29,9 @@ def download_zip_file(url, output_path):
 # Step 2: Extract the file from the zip file
 def extract_file(zip_path, extract_to_dir):
     logger.info(f"Extracting {zip_path} to {extract_to_dir}...")
+    if not os.path.exists(zip_path):
+        logger.error(f"Zip file does not exist: {zip_path}")
+        return
     try:
         with zipfile.ZipFile(zip_path, 'r') as zip_ref:
             zip_ref.extractall(extract_to_dir)
@@ -47,15 +45,31 @@ def extract_file(zip_path, extract_to_dir):
     except Exception as e:
         logger.error(f"An error occurred while extracting the file: {e}")
 
-# Step 3: Set up a Flask route to expose the extracted file
+# Route to manually trigger the download and extraction process
+@app.route('/update', methods=['POST'])
+def update_files():
+    download_zip_file(zip_url, zip_path)
+    extract_file(zip_path, extract_to_dir)
+    return jsonify({'status': 'Update complete'}), 200
+
+# Route to view logs
+@app.route('/logs', methods=['GET'])
+def get_logs():
+    # Read logs from the log file
+    try:
+        with open('app.log', 'r') as f:
+            logs = f.read()
+        return jsonify({'logs': logs}), 200
+    except Exception as e:
+        logger.error(f"Failed to read logs: {e}")
+        return jsonify({'error': 'Failed to read logs'}), 500
+
+# Route to serve the extracted file
 @app.route('/wltv', methods=['GET'])
 def serve_extracted_file():
-    # Log contents of extracted directory for debugging
     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
@@ -70,19 +84,9 @@ def serve_extracted_file():
         logger.error("File not found")
         return "File not found", 404
 
-# New route to serve the logs
-@app.route('/logs', methods=['GET'])
-def serve_logs():
-    log_contents = log_stream.getvalue()
-    return Response(log_contents, mimetype='text/plain')
-
-# Route to manually trigger the download and extraction process
-@app.route('/update', methods=['POST'])
-def update_files():
-    download_zip_file(zip_url, zip_path)
-    extract_file(zip_path, extract_to_dir)
-    return jsonify({'status': 'Update complete'}), 200
-
 if __name__ == '__main__':
+    zip_url = 'http://epg-guide.com/wltv.dz'  # Updated URL
+    zip_path = 'it_wltv_full.zip'
+
     # Start the Flask web server
     app.run(host='0.0.0.0', port=8080)