init

pt_gen/services/tmdb.py

@@ -0,0 +1,109 @@
+import httpx
+from typing import Optional, List, Dict, Any
+
+class TMDBClient:
+    BASE_URL = "https://api.themoviedb.org/3"
+
+    def __init__(self, api_key: str):
+        self.api_key = api_key
+
+    async def _make_request(self, endpoint: str, params: Optional[Dict[str, Any]] = None) -> Optional[Dict[str, Any]]:
+        if params is None:
+            params = {}
+        # 总是包含 api_key 和 language
+        params['api_key'] = self.api_key
+        params['language'] = 'zh-CN'
+        
+        url = f"{self.BASE_URL}{endpoint}"
+        async with httpx.AsyncClient() as client:
+            try:
+                response = await client.get(url, params=params)
+                response.raise_for_status()
+                return response.json()
+            except httpx.HTTPStatusError as e:
+                print(f"TMDB API 请求失败: {e.response.status_code} for URL: {e.request.url}")
+                return None
+            except httpx.RequestError as e:
+                print(f"TMDB 请求错误: {e}")
+                return None
+
+    async def find_by_imdb_id(self, imdb_id: str) -> Optional[Dict[str, Any]]:
+        endpoint = f"/find/{imdb_id}"
+        return await self._make_request(endpoint, params={'external_source': 'imdb_id'})
+
+    # --- 重构 get_movie_details 方法 ---
+    async def get_movie_details(self, tmdb_id: str) -> Optional[Dict[str, Any]]:
+        params = {
+            'append_to_response': 'credits,alternative_titles'
+        }
+        endpoint = f"/movie/{tmdb_id}"
+        data = await self._make_request(endpoint, params=params)
+        
+        if not data:
+            return None
+        
+        return self._parse_movie_details(data)
+
+    def _parse_movie_details(self, data: Dict[str, Any]) -> Dict[str, Any]:
+        """将从 TMDB API 获取的原始数据解析为干净的字典"""
+        
+        directors = []
+        if 'credits' in data and 'crew' in data['credits']:
+            for member in data['credits']['crew']:
+                if member.get('job') == 'Director':
+                    directors.append(member['name'])
+
+        writers = []
+        writing_jobs = {'Writer', 'Screenplay', 'Story'}
+        if 'credits' in data and 'crew' in data['credits']:
+            for member in data['credits']['crew']:
+                # 检查部门是否为'Writing'并且职位是我们想要的
+                if member.get('department') == 'Writing':#  and member.get('job') in writing_jobs:
+                    # 避免重复添加同一个人
+                    if member['name'] not in writers:
+                        writers.append(member['name'])
+
+        actors = []
+        if 'credits' in data and 'cast' in data['credits']:
+            for member in data['credits']['cast']:
+                actors.append(member['name'])
+
+
+        genres = [genre['name'] for genre in data.get('genres', [])]
+
+
+        countries = [country['name'] for country in data.get('production_countries', [])]
+        
+
+        aka_titles = []
+        if 'alternative_titles' in data and 'titles' in data['alternative_titles']:
+             for item in data['alternative_titles']['titles']:
+                if item.get('iso_3166_1') == 'CN':
+                    aka_titles.append(item['title'])
+        runtime_min = data.get('runtime')
+        runtime = f"{runtime_min}分钟" if runtime_min else None
+        spoken_languages = [lang['english_name'] for lang in data.get('spoken_languages', [])]
+        
+        parsed_data = {
+            "id": data.get('id'),
+            "imdb_id": data.get('imdb_id'),
+            "title": data.get('title'),
+            "original_title": data.get('original_title'),
+            "overview": data.get('overview'),
+            "release_date": data.get('release_date', ''),
+            "runtime": runtime,
+            "spoken_languages": spoken_languages,
+            "tagline": data.get('tagline'),
+            "tagline": data.get('tagline'),
+            "poster_path": data.get('poster_path'),
+            "vote_average": data.get('vote_average', 0),
+            "directors": directors,
+            "writers": writers,
+            "actors": actors,
+            "genres": genres,
+            "countries": countries,
+            "aka_titles": aka_titles
+        }
+
+        return parsed_data
+