Update wltv_server.py

wltv_server.py

@@ -1,6 +1,7 @@
 import os
 import requests
-import zipfile
+import gzip
+import shutil
 import logging
 from flask import Flask, send_file, jsonify
 
@@ -15,17 +16,15 @@ extract_to_dir = 'extracted'
 if not os.path.exists(extract_to_dir):
     os.makedirs(extract_to_dir, exist_ok=True)
 
-log_storage = []
-
-# Step 1: Download the zip file
-def download_zip_file(url, output_path):
+# Function to download the file
+def download_file(url, output_path):
     logger.info(f"Downloading from {url}...")
     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")
+            logger.info(f"Downloaded 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:
@@ -33,26 +32,22 @@ def download_zip_file(url, output_path):
     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):
-    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}")
+# Function to extract .gz files
+def extract_gz_file(gz_path, extract_to_dir):
+    logger.info(f"Extracting {gz_path} to {extract_to_dir}...")
+    if not os.path.exists(gz_path):
+        logger.error(f"GZ file does not exist: {gz_path}")
         return
     try:
-        with zipfile.ZipFile(zip_path, 'r') as zip_ref:
-            zip_ref.extractall(extract_to_dir)
-            # Log extracted files and their extensions
-            extracted_files = zip_ref.namelist()
-            logger.info(f"Extracted files: {extracted_files}")
-            for file in extracted_files:
-                logger.info(f"File: {file}, Extension: {os.path.splitext(file)[1]}")
-    except zipfile.BadZipFile:
-        logger.error("The downloaded file is not a zip file or is corrupted.")
+        extracted_file_path = os.path.join(extract_to_dir, 'it_wltv_full')
+        with gzip.open(gz_path, 'rb') as f_in:
+            with open(extracted_file_path, 'wb') as f_out:
+                shutil.copyfileobj(f_in, f_out)
+        logger.info(f"Extracted GZ file to {extracted_file_path}")
     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
+# Flask route to serve the extracted file
 @app.route('/wltv', methods=['GET'])
 def serve_extracted_file():
     try:
@@ -88,22 +83,22 @@ def get_logs():
 # 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'
+    file_url = 'http://epg-guide.com/wltv.gz'
+    file_path = '/tmp/it_wltv_full.gz'
 
-    # Download and extract the zip file
-    download_zip_file(zip_url, zip_path)
-    extract_file(zip_path, extract_to_dir)
+    # Download and extract the gz file
+    download_file(file_url, file_path)
+    extract_gz_file(file_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'
+    file_url = 'http://epg-guide.com/wltv.gz'
+    file_path = '/tmp/it_wltv_full.gz'
 
-    # Download and extract the zip file initially
-    download_zip_file(zip_url, zip_path)
-    extract_file(zip_path, extract_to_dir)
+    # Download and extract the gz file initially
+    download_file(file_url, file_path)
+    extract_gz_file(file_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)