refactor code

plugins/pixiv/data_source.py

@@ -5,20 +5,22 @@ from utils.utils import get_local_proxy
 from utils.user_agent import get_user_agent
 from bs4 import BeautifulSoup
 import feedparser
-from utils.init_result import image
+from utils.message_builder import image
 from asyncio.exceptions import TimeoutError
 from configs.config import RSSHUBAPP
 from aiohttp.client_exceptions import ClientConnectorError
 import platform
-if platform.system() == 'Windows':
+
+if platform.system() == "Windows":
     import asyncio
+
     asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
 
 
-async def get_pixiv_urls(mode: str, num: int = 5, date: str = '') -> 'list, list, int':
-    url = f'{RSSHUBAPP}pixiv/ranking/{mode}'
+async def get_pixiv_urls(mode: str, num: int = 5, date: str = "") -> "list, list, int":
+    url = f"{RSSHUBAPP}/pixiv/ranking/{mode}"
     if date:
-        url += f'/{date}'
+        url += f"/{date}"
     try:
         return await parser_data(url, num)
     except ClientConnectorError:
@@ -26,62 +28,70 @@ async def get_pixiv_urls(mode: str, num: int = 5, date: str = '') -> 'list, list
 
 
 async def download_pixiv_imgs(urls: list, user_id: int) -> str:
-    result = ''
+    result = ""
     index = 0
     for img in urls:
         async with aiohttp.ClientSession(headers=get_user_agent()) as session:
             for _ in range(3):
-                async with session.get(img, proxy=get_local_proxy(), timeout=2) as response:
-                    async with aiofiles.open(IMAGE_PATH + f'temp/{user_id}_{index}_pixiv.jpg', 'wb') as f:
+                async with session.get(
+                    img, proxy=get_local_proxy(), timeout=2
+                ) as response:
+                    async with aiofiles.open(
+                        IMAGE_PATH + f"temp/{user_id}_{index}_pixiv.jpg", "wb"
+                    ) as f:
                         try:
                             await f.write(await response.read())
-                            result += image(f'{user_id}_{index}_pixiv.jpg', 'temp')
+                            result += image(f"{user_id}_{index}_pixiv.jpg", "temp")
                             index += 1
                             break
                         except TimeoutError:
                             # result += '\n这张图下载失败了..\n'
                             pass
             else:
-                result += '\n这张图下载失败了..\n'
+                result += "\n这张图下载失败了..\n"
     return result
 
 
-async def search_pixiv_urls(keyword: str, num: int, order: str, r18: int) -> 'list, list':
-    url = f'{RSSHUBAPP}pixiv/search/{keyword}/{order}/{r18}'
+async def search_pixiv_urls(
+    keyword: str, num: int, order: str, r18: int
+) -> "list, list":
+    url = f"{RSSHUBAPP}/pixiv/search/{keyword}/{order}/{r18}"
     return await parser_data(url, num)
 
 
-async def parser_data(url: str, num: int) -> 'list, list, int':
+async def parser_data(url: str, num: int) -> "list, list, int":
     text_list = []
     urls = []
     async with aiohttp.ClientSession() as session:
         for _ in range(3):
             try:
-                async with session.get(url, proxy=get_local_proxy(), timeout=2) as response:
-                    data = feedparser.parse(await response.text())['entries']
+                async with session.get(
+                    url, proxy=get_local_proxy(), timeout=2
+                ) as response:
+                    data = feedparser.parse(await response.text())["entries"]
                     break
             except TimeoutError:
                 pass
         else:
-            return ['网络不太好，也许过一会就好了'], [], 998
+            return ["网络不太好，也许过一会就好了"], [], 998
         try:
             if len(data) == 0:
-                return ['没有搜索到喔'], [], 997
+                return ["没有搜索到喔"], [], 997
             if num > len(data):
                 num = len(data)
             data = data[:num]
             for data in data:
-                soup = BeautifulSoup(data['summary'], 'lxml')
-                title = "标题：" + data['title']
-                pl = soup.find_all('p')
-                author = pl[0].text.split('-')[0].strip()
+                soup = BeautifulSoup(data["summary"], "lxml")
+                title = "标题：" + data["title"]
+                pl = soup.find_all("p")
+                author = pl[0].text.split("-")[0].strip()
                 imgs = []
-                text_list.append(f'{title}\n{author}\n')
+                text_list.append(f"{title}\n{author}\n")
                 for p in pl[1:]:
-                    imgs.append(p.find('img').get('src'))
+                    imgs.append(p.find("img").get("src"))
                 urls.append(imgs)
         except ValueError:
-            return ['是网站坏了啊，也许过一会就好了'], [], 999
+            return ["是网站坏了啊，也许过一会就好了"], [], 999
         return text_list, urls, 200