Update wltv_server.py

wltv_server.py

@@ -5,7 +5,7 @@ import logging
 from flask import Flask, send_file, jsonify
 
 # Set up logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
@@ -15,9 +15,7 @@ extract_to_dir = 'extracted'
 if not os.path.exists(extract_to_dir):
     os.makedirs(extract_to_dir, exist_ok=True)
 
-# Update URL to the correct location
-zip_url = 'http://epg-guide.com/wltv.gz'
-zip_path = 'it_wltv_full.zip'
+log_storage = []
 
 # Step 1: Download the zip file
 def download_zip_file(url, output_path):
@@ -30,6 +28,8 @@ def download_zip_file(url, output_path):
             logger.info(f"Downloaded zip file to {output_path} with size {len(response.content)} bytes")
         else:
             logger.error(f"Failed to download file. Status code: {response.status_code}")
+    except PermissionError as e:
+        logger.error(f"Permission error occurred while downloading the file: {e}")
     except Exception as e:
         logger.error(f"An error occurred while downloading the file: {e}")
 
@@ -52,34 +52,14 @@ def extract_file(zip_path, extract_to_dir):
     except Exception as e:
         logger.error(f"An error occurred while extracting the file: {e}")
 
-# 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():
-    try:
-        if os.path.exists('app.log'):
-            with open('app.log', 'r') as f:
-                logs = f.read()
-            return jsonify({'logs': logs}), 200
-        else:
-            logger.error("Log file does not exist")
-            return jsonify({'error': 'Log file does not exist'}), 500
-    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
+# Step 3: Set up a Flask route to expose 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
@@ -94,6 +74,36 @@ def serve_extracted_file():
         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():
+    zip_url = 'http://epg-guide.com/wltv.gz'
+    zip_path = '/tmp/it_wltv_full.zip'
+
+    # Download and extract the zip file
+    download_zip_file(zip_url, zip_path)
+    extract_file(zip_path, extract_to_dir)
+
+    return "Files updated", 200
+
 if __name__ == '__main__':
+    zip_url = 'http://epg-guide.com/wltv.gz'
+    zip_path = '/tmp/it_wltv_full.zip'
+
+    # Download and extract the zip file initially
+    download_zip_file(zip_url, zip_path)
+    extract_file(zip_path, extract_to_dir)
+
     # Start the Flask web server
-    app.run(host='0.0.0.0', port=8080)
+    app.run(host='0.0.0.0', port=8080)