update

app.py

@@ -112,6 +112,40 @@ def bytes_to_human_readable(num, suffix="B"):
         num /= 1024.0
     return f"{num:.1f} Y{suffix}"
 
+def get_all_tv_shows(indexed_cache):
+    """Get all TV shows from the indexed cache structure JSON file."""
+    tv_shows = {}
+    for directory in indexed_cache:
+        if directory['type'] == 'directory' and directory['path'] == 'tv':
+            for sub_directory in directory['contents']:
+                if sub_directory['type'] == 'directory':
+                    show_title = sub_directory['path'].split('/')[-1]
+                    tv_shows[show_title] = []
+                    for season_directory in sub_directory['contents']:
+                        if season_directory['type'] == 'directory':
+                            season = season_directory['path'].split('/')[-1]
+                            for episode in season_directory['contents']:
+                                if episode['type'] == 'file':
+                                    tv_shows[show_title].append({
+                                        "season": season,
+                                        "episode": episode['path'].split('/')[-1],
+                                        "path": episode['path']
+                                    })
+    return tv_shows
+
+def get_all_films(indexed_cache):
+    """Get all films from the indexed cache structure JSON file."""
+    films = []
+    for directory in indexed_cache:
+        if directory['type'] == 'directory' and directory['path'] == 'films':
+            for sub_directory in directory['contents']:
+                if sub_directory['type'] == 'directory':
+                    for item in sub_directory['contents']:
+                        if item['type'] == 'file':
+                            films.append(item['path'])
+    return films
+
+
 def start_prefetching():
     """Start the metadata prefetching in a separate thread."""
     prefetch_metadata()
@@ -124,7 +158,7 @@ thread.start()
 # API Endpoints
 
 @app.route('/api/movie', methods=['GET'])
-def get_movie():
+def get_movie_api():
     """Endpoint to get the movie by title."""
     title = request.args.get('title')
     if not title:
@@ -159,7 +193,7 @@ def get_movie():
     return jsonify({"status": "Download started", "film_id": film_id})
 
 @app.route('/api/tv', methods=['GET'])
-def get_tv_show():
+def get_tv_show_api():
     """Endpoint to get the TV show by title, season, and episode."""
     title = request.args.get('title')
     season = request.args.get('season')
@@ -208,13 +242,13 @@ def get_tv_show():
     return jsonify({"status": "Download started", "episode_id": episode_id})
 
 @app.route('/api/progress/<id>', methods=['GET'])
-def get_progress(id):
+def get_progress_api(id):
     """Endpoint to get the download progress of a movie or TV show episode."""
     progress = get_download_progress(id)
     return jsonify({"id": id, "progress": progress})
 
 @app.route('/api/filmid', methods=['GET'])
-def get_film_id_by_title():
+def get_film_id_by_title_api():
     """Endpoint to get the film ID by providing the movie title."""
     title = request.args.get('title')
     if not title:
@@ -223,7 +257,7 @@ def get_film_id_by_title():
     return jsonify({"film_id": film_id})
 
 @app.route('/api/episodeid', methods=['GET'])
-def get_episode_id():
+def get_episode_id_api():
     """Endpoint to get the episode ID by providing the TV show title, season, and episode."""
     title = request.args.get('title')
     season = request.args.get('season')
@@ -234,12 +268,12 @@ def get_episode_id():
     return jsonify({"episode_id": episode_id})
 
 @app.route('/api/proxy', methods=['GET'])
-def get_proxy():
+def get_proxy_api():
     proxies = get_system_proxies()
     return jsonify({"proxies": proxies})
 
 @app.route('/api/size', methods=['GET'])
-def get_cache_size():
+def get_cache_size_api():
     total_size = 0
     for dirpath, dirnames, filenames in os.walk(CACHE_DIR):
         for f in filenames:
@@ -249,7 +283,7 @@ def get_cache_size():
     return jsonify({"cache_size": readable_size})
 
 @app.route('/api/clear_cache', methods=['POST'])
-def clear_cache():
+def clear_cache_api():
     for dirpath, dirnames, filenames in os.walk(CACHE_DIR):
         for f in filenames:
             fp = os.path.join(dirpath, f)
@@ -257,7 +291,7 @@ def clear_cache():
     return jsonify({"status": "Cache cleared"})
 
 @app.route('/api/store/tv', methods=['GET'])
-def get_tv_store():
+def get_tv_store_api():
     """Endpoint to get the TV store JSON."""
     if os.path.exists(TV_STORE_JSON_PATH):
         with open(TV_STORE_JSON_PATH, 'r') as json_file:
@@ -266,7 +300,7 @@ def get_tv_store():
     return jsonify({}), 404
 
 @app.route('/api/film/metadata', methods=['GET'])
-def get_film_metadata():
+def get_film_metadata_api():
     """Endpoint to get the film metadata by title."""
     title = request.args.get('title')
     if not title:
@@ -282,7 +316,7 @@ def get_film_metadata():
     return jsonify({'error': 'Metadata not found'}), 404
 
 @app.route('/api/tv/metadata', methods=['GET'])
-def get_tv_metadata():
+def get_tv_metadata_api():
     """Endpoint to get the TV show metadata by title."""
     title = request.args.get('title')
     if not title:
@@ -303,6 +337,12 @@ def get_tv_metadata():
     
     return jsonify({'error': 'Metadata not found'}), 404
 
+
+@app.route("/api/all_films")
+def get_all_films_api():
+    get_all_films()
+
+
 # Routes
 @app.route('/')
 def index():

test.py

@@ -4,19 +4,30 @@ import json
 CACHE_DIR = os.getenv("CACHE_DIR")
 INDEX_FILE = os.getenv("INDEX_FILE")
 
-def find_tv_path(json_data, title):
-    """Find the path of the TV show in the JSON data based on the title."""
-    for directory in json_data:
+def get_all_tv_shows(indexed_cache):
+    """Get all TV shows from the indexed cache structure JSON file."""
+    tv_shows = {}
+    for directory in indexed_cache:
         if directory['type'] == 'directory' and directory['path'] == 'tv':
             for sub_directory in directory['contents']:
-                if sub_directory['type'] == 'directory' and title.lower() in sub_directory['path'].lower():
-                    return sub_directory
-    return None
-
+                if sub_directory['type'] == 'directory':
+                    show_title = sub_directory['path'].split('/')[-1]
+                    tv_shows[show_title] = []
+                    for season_directory in sub_directory['contents']:
+                        if season_directory['type'] == 'directory':
+                            season = season_directory['path'].split('/')[-1]
+                            for episode in season_directory['contents']:
+                                if episode['type'] == 'file':
+                                    tv_shows[show_title].append({
+                                        "season": season,
+                                        "episode": episode['path'].split('/')[-1],
+                                        "path": episode['path']
+                                    })
+    return tv_shows
 
 
 with open(INDEX_FILE, 'r') as f:
     file_structure = json.load(f)
 
 
-print(find_tv_path(file_structure,title="Aho Girl"))
+print(get_all_tv_shows(file_structure))