fix actions

app/actions/fetch_medias.py

@@ -1,12 +1,22 @@
+from typing import List
+
+from pydantic import Field
+
 from app.actions import BaseAction
 from app.schemas import ActionParams, ActionContext
+from core.config import settings
+from core.event import eventmanager
+from log import logger
+from schemas import RecommendSourceEventData, MediaInfo
+from schemas.types import ChainEventType
+from utils.http import RequestUtils
 
 
 class FetchMediasParams(ActionParams):
     """
     获取媒体数据参数
     """
-    pass
+    sources: List[str] = Field([], description="媒体数据来源")
 
 
 class FetchMediasAction(BaseAction):
@@ -14,6 +24,63 @@ class FetchMediasAction(BaseAction):
     获取媒体数据
     """
 
+    __inner_sources = [
+        {
+            "api_path": 'recommend/tmdb_trending',
+            "name": '流行趋势',
+        },
+        {
+            "api_path": 'recommend/douban_showing',
+            "name": '正在热映',
+        },
+        {
+            "api_path": 'bangumi/calendar',
+            "name": 'Bangumi每日放送',
+        },
+        {
+            "api_path": 'recommend/tmdb_movies',
+            "name": 'TMDB热门电影',
+        },
+        {
+            "api_path": 'recommend/tmdb_tvs?with_original_language=zh|en|ja|ko',
+            "name": 'TMDB热门电视剧',
+        },
+        {
+            "api_path": 'recommend/douban_movie_hot',
+            "name": '豆瓣热门电影',
+        },
+        {
+            "api_path": 'recommend/douban_tv_hot',
+            "name": '豆瓣热门电视剧',
+        },
+        {
+            "api_path": 'recommend/douban_tv_animation',
+            "name": '豆瓣热门动漫',
+        },
+        {
+            "api_path": 'recommend/douban_movies',
+            "name": '豆瓣最新电影',
+        },
+        {
+            "api_path": 'recommend/douban_tvs',
+            "name": '豆瓣最新电视剧',
+        },
+        {
+            "api_path": 'recommend/douban_movie_top250',
+            "name": '豆瓣电影TOP250',
+        },
+        {
+            "api_path": 'recommend/douban_tv_weekly_chinese',
+            "name": '豆瓣国产剧集榜',
+        },
+        {
+            "api_path": 'recommend/douban_tv_weekly_global',
+            "name": '豆瓣全球剧集榜',
+        }
+    ]
+
+    __medias = []
+
     @property
     def name(self) -> str:
         return "获取媒体数据"
@@ -24,7 +91,47 @@ class FetchMediasAction(BaseAction):
 
     @property
     def success(self) -> bool:
-        return True
+        return True if self.__medias else False
+
+    def __get_source(self, source: str):
+        """
+        获取数据源
+        """
+        for s in self.__inner_sources:
+            if s['name'] == source:
+                return s
+        return None
 
     async def execute(self, params: FetchMediasParams, context: ActionContext) -> ActionContext:
-        pass
+        """
+        获取媒体数据，填充到medias
+        """
+        # 广播事件，请示额外的推荐数据源支持
+        event_data = RecommendSourceEventData()
+        event = eventmanager.send_event(ChainEventType.RecommendSource, event_data)
+        # 使用事件返回的上下文数据
+        if event and event.event_data:
+            event_data: RecommendSourceEventData = event.event_data
+            if event_data.extra_sources:
+                self.__inner_sources.extend([s.dict() for s in event_data.extra_sources])
+
+        for name in params.sources:
+            source = self.__get_source(name)
+            if not source:
+                continue
+            logger.info(f"获取媒体数据 {source} ...")
+            # 调用内部API获取数据
+            api_url = f"http://127.0.0.1:{settings.PORT}/api/v1/{source['api_path']}?token={settings.API_TOKEN}"
+            res = RequestUtils(timeout=15).post_res(api_url)
+            if res:
+                results = res.json()
+                logger.info(f"{name} 获取到 {len(results)} 条数据")
+                self.__medias.extend([MediaInfo(**r) for r in results])
+            else:
+                logger.error(f"{name} 获取数据失败")
+
+        if self.__medias:
+            context.medias.extend(self.__medias)
+
+        self.job_done()
+        return context