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)
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
 logger = logging.getLogger(__name__)
 
 app = Flask(__name__)
@@ -15,16 +15,23 @@ 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.dz'
+zip_path = 'it_wltv_full.zip'
+
 # Step 1: Download the zip file
 def download_zip_file(url, output_path):
     logger.info(f"Downloading from {url}...")
-    response = requests.get(url)
-    if response.status_code == 200:
-        with open(output_path, 'wb') as f:
-            f.write(response.content)
-        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}")
+    try:
+        response = requests.get(url)
+        if response.status_code == 200:
+            with open(output_path, 'wb') as f:
+                f.write(response.content)
+            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 Exception as e:
+        logger.error(f"An error occurred while downloading the file: {e}")
 
 # Step 2: Extract the file from the zip file
 def extract_file(zip_path, extract_to_dir):
@@ -55,11 +62,14 @@ def update_files():
 # 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
+        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
@@ -85,8 +95,5 @@ def serve_extracted_file():
         return "File not found", 404
 
 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)