ChandimaPrabath commited on
Commit
da7bab3
·
1 Parent(s): 2f37eba

fix tv metadata api

Browse files
Files changed (2) hide show
  1. app.py +12 -3
  2. test.py +22 -0
app.py CHANGED
@@ -67,6 +67,15 @@ def find_tv_path(json_data, title):
67
  return sub_directory['path']
68
  return None
69
 
 
 
 
 
 
 
 
 
 
70
  def get_film_id(title):
71
  """Generate a film ID based on the title."""
72
  return title.replace(" ", "_").lower()
@@ -286,9 +295,9 @@ def get_tv_metadata():
286
  data = json.load(f)
287
 
288
  # Add the file structure to the metadata
289
- tv_path = find_tv_path(file_structure, title)
290
- if tv_path:
291
- data['file_structure'] = tv_path
292
 
293
  return jsonify(data)
294
 
 
67
  return sub_directory['path']
68
  return None
69
 
70
+ def get_tv_structure(json_data,title):
71
+ """Find the path of the TV show in the JSON data based on the title."""
72
+ for directory in json_data:
73
+ if directory['type'] == 'directory' and directory['path'] == 'tv':
74
+ for sub_directory in directory['contents']:
75
+ if sub_directory['type'] == 'directory' and title.lower() in sub_directory['path'].lower():
76
+ return sub_directory
77
+ return None
78
+
79
  def get_film_id(title):
80
  """Generate a film ID based on the title."""
81
  return title.replace(" ", "_").lower()
 
295
  data = json.load(f)
296
 
297
  # Add the file structure to the metadata
298
+ tv_structure_data = get_tv_structure(file_structure, title)
299
+ if tv_structure_data:
300
+ data['file_structure'] = tv_structure_data
301
 
302
  return jsonify(data)
303
 
test.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ import json
3
+
4
+ CACHE_DIR = os.getenv("CACHE_DIR")
5
+ INDEX_FILE = os.getenv("INDEX_FILE")
6
+
7
+ def find_tv_path(json_data, title):
8
+ """Find the path of the TV show in the JSON data based on the title."""
9
+ for directory in json_data:
10
+ if directory['type'] == 'directory' and directory['path'] == 'tv':
11
+ for sub_directory in directory['contents']:
12
+ if sub_directory['type'] == 'directory' and title.lower() in sub_directory['path'].lower():
13
+ return sub_directory
14
+ return None
15
+
16
+
17
+
18
+ with open(INDEX_FILE, 'r') as f:
19
+ file_structure = json.load(f)
20
+
21
+
22
+ print(find_tv_path(file_structure,title="Aho Girl"))