update 0.0.6.0

plugins/search_anime/data_source.py

@@ -2,12 +2,13 @@ from lxml import etree
 import feedparser
 from urllib import parse
 from services.log import logger
+from typing import List
 import aiohttp
 import time
 from utils.utils import get_local_proxy
 
 
-async def from_anime_get_info(key_word: str, max_: int) -> str:
+async def from_anime_get_info(key_word: str, max_: int) -> List[str]:
     s_time = time.time()
     repass = ""
     url = "https://share.dmhy.org/topics/rss/rss.xml?keyword=" + parse.quote(key_word)
@@ -15,15 +16,16 @@ async def from_anime_get_info(key_word: str, max_: int) -> str:
         repass = await get_repass(url, max_)
     except Exception as e:
         logger.error("Timeout! {}".format(e))
-
-    return f"搜索 {key_word} 结果（耗时 {int(time.time() - s_time)} 秒）：\n" + repass
+    repass.insert(0, f"搜索 {key_word} 结果（耗时 {int(time.time() - s_time)} 秒）：\n")
+    return repass
 
 
-async def get_repass(url: str, max_: int) -> str:
-    putline = []
+async def get_repass(url: str, max_: int) -> List[str]:
+    put_line = []
     async with aiohttp.ClientSession() as session:
         async with session.get(url, proxy=get_local_proxy(), timeout=20) as response:
             d = feedparser.parse(await response.text())
+            max_ = max_ if max_ < len([e.link for e in d.entries]) else len([e.link for e in d.entries])
             url_list = [e.link for e in d.entries][:max_]
             for u in url_list:
                 try:
@@ -44,15 +46,11 @@ async def get_repass(url: str, max_: int) -> str:
                             .replace("\t", "")
                         )
                         size = item[3].xpath("string(.)")[5:].strip()
-                        putline.append(
+                        put_line.append(
                             "【{}】| {}\n【{}】| {}".format(class_a, title, size, magent)
                         )
                 except Exception as e:
                     logger.warning(f"搜番超时 e：{e}")
-
-        repass = "\n\n".join(putline)
-
-        return repass
+    return put_line
 
 
-# print(asyncio.get_event_loop().run_until_complete(from_anime_get_info('进击的巨人', 1234556)))