divided function into utilies in order to add a new API

2024-07-08 15:37:04 +02:00
parent 9c7eb1da4d
commit 98a36e996a
11 changed files with 86 additions and 76 deletions
@@ -0,0 +1,26 @@
+import requests
+from tmdbv3api import TMDb, Movie, TV
+from loadenv import load_env
+TMDB_KEY, DOMAIN = load_env()
+
+def get_TMDb_id_from_IMDb_id(imdb_id):
+    if ":" in imdb_id:
+        season = imdb_id.split(":")[1]
+        episode = imdb_id[-1]
+        ismovie = 0
+        imdb_id = imdb_id.split(":")[0]
+    else:
+        ismovie = 1
+    response = requests.get(f'https://api.themoviedb.org/3/find/{imdb_id}', 
+                            params={'external_source': 'imdb_id', 'api_key': f'{TMDB_KEY}'})
+    tmbda = response.json()
+    if tmbda['movie_results']:
+        print(tmbda)
+        return tmbda['movie_results'][0]['id'], ismovie
+    elif tmbda['tv_results']:
+        print(tmbda['tv_results'][0]['id'])
+        return tmbda['tv_results'][0]['id'], season, episode , ismovie 
+    else:
+        return None
+    
+get_TMDb_id_from_IMDb_id("tt1839578:1:1")