fix episode downloading

app.py

@@ -3,7 +3,7 @@ import os
 import json
 import threading
 import urllib.parse
-from hf_scrapper import download_file, get_system_proxies, get_download_progress
+from hf_scrapper import download_film, download_episode, get_system_proxies, get_download_progress
 from indexer import indexer
 from tvdb import fetch_and_cache_json
 import re
@@ -154,7 +154,7 @@ thread.start()
 
 # API Endpoints
 
-@app.route('/api/movie', methods=['GET'])
+@app.route('/api/film', methods=['GET'])
 def get_movie_api():
     """Endpoint to get the movie by title."""
     title = request.args.get('title')
@@ -183,7 +183,7 @@ def get_movie_api():
     
     # Start the download in a separate thread if not already downloading
     if film_id not in download_threads or not download_threads[film_id].is_alive():
-        thread = threading.Thread(target=download_file, args=(file_url, TOKEN, cache_path, proxies, film_id, title))
+        thread = threading.Thread(target=download_film, args=(file_url, TOKEN, cache_path, proxies, film_id, title))
         download_threads[film_id] = thread
         thread.start()
     
@@ -236,7 +236,7 @@ def get_tv_show_api():
     
     # Start the download in a separate thread if not already downloading
     if episode_id not in download_threads or not download_threads[episode_id].is_alive():
-        thread = threading.Thread(target=download_file, args=(file_url, TOKEN, cache_path, proxies, episode_id, title))
+        thread = threading.Thread(target=download_episode, args=(file_url, TOKEN, cache_path, proxies, episode_id, title))
         download_threads[episode_id] = thread
         thread.start()
     

hf_scrapper.py

@@ -29,7 +29,7 @@ def get_system_proxies():
         print(f"Error getting system proxies: {e}")
         return {}
 
-def download_file(file_url, token, cache_path, proxies, film_id, title, chunk_size=100 * 1024 * 1024):
+def download_film(file_url, token, cache_path, proxies, film_id, title, chunk_size=100 * 1024 * 1024):
     """
     Downloads a file from the specified URL and saves it to the cache path.
     Tracks the download progress.
@@ -120,6 +120,74 @@ def update_film_store_json(title, cache_path):
         json.dump(film_store_data, json_file, indent=2)
     print(f'Film store updated with {title}.')
 
+
+###############################################################################
+def download_episode(file_url, token, cache_path, proxies, episode_id, title, chunk_size=100 * 1024 * 1024):
+    """
+    Downloads a file from the specified URL and saves it to the cache path.
+    Tracks the download progress.
+
+    Args:
+        file_url (str): The URL of the file to download.
+        token (str): The authorization token for the request.
+        cache_path (str): The path to save the downloaded file.
+        proxies (dict): Proxies for the request.
+        film_id (str): Unique identifier for the film download.
+        title (str): The title of the film.
+        chunk_size (int): Size of each chunk to download.
+    """
+    print(f"Downloading file from URL: {file_url} to {cache_path} with proxies: {proxies}")
+    headers = {'Authorization': f'Bearer {token}'}
+    try:
+        response = requests.get(file_url, headers=headers, proxies=proxies, stream=True)
+        response.raise_for_status()
+        
+        total_size = int(response.headers.get('content-length', 0))
+        download_progress[film_id] = {"total": total_size, "downloaded": 0, "status": "Downloading", "start_time": time.time()}
+
+        os.makedirs(os.path.dirname(cache_path), exist_ok=True)
+        with open(cache_path, 'wb') as file, tqdm(total=total_size, unit='B', unit_scale=True, desc=cache_path) as pbar:
+            for data in response.iter_content(chunk_size=chunk_size):
+                file.write(data)
+                pbar.update(len(data))
+                download_progress[episode_id]["downloaded"] += len(data)
+        
+        print(f'File cached to {cache_path} successfully.')
+        update_tv_store_json(title, cache_path)
+        download_progress[episode_id]["status"] = "Completed"
+    except RequestException as e:
+        print(f"Error downloading file: {e}")
+        download_progress[episode_id]["status"] = "Failed"
+    except IOError as e:
+        print(f"Error writing file {cache_path}: {e}")
+        download_progress[episode_id]["status"] = "Failed"
+    finally:
+        if download_progress[episode_id]["status"] != "Downloading":
+            download_progress[episode_id]["end_time"] = time.time()
+
+
+def update_tv_store_json(title, cache_path):
+    """
+    Updates the tv store JSON with the new file.
+
+    Args:
+        title (str): The title of the film.
+        cache_path (str): The local path where the file is saved.
+    """
+    TV_STORE_JSON_PATH = os.path.join(CACHE_DIR, "tv_store.json")
+
+    film_store_data = {}
+    if os.path.exists(TV_STORE_JSON_PATH):
+        with open(TV_STORE_JSON_PATH, 'r') as json_file:
+            film_store_data = json.load(json_file)
+
+    film_store_data[title] = cache_path
+
+    with open(TV_STORE_JSON_PATH, 'w') as json_file:
+        json.dump(film_store_data, json_file, indent=2)
+    print(f'TV store updated with {title}.')
+
+###############################################################################
 def get_file_structure(repo, token, path="", proxies=None):
     """
     Fetches the file structure of a specified Hugging Face repository.
@@ -165,4 +233,4 @@ if __name__ == "__main__":
     cache_path = os.path.join(CACHE_DIR, "films/Funky Monkey 2004/Funky Monkey (2004) Web-dl 1080p.mp4")
     proxies = get_system_proxies()
     film_id = "funky_monkey_2004"  # Unique identifier for the film download
-    download_file(file_url, token, cache_path, proxies=proxies, film_id=film_id)
+    download_film(file_url, token, cache_path, proxies=proxies, film_id=film_id)