mirror of
https://github.com/suyiiyii/nonebot-bison.git
synced 2025-06-05 11:26:43 +08:00
102 lines
3.2 KiB
Python
102 lines
3.2 KiB
Python
import calendar
|
||
import time
|
||
from typing import Any, ClassVar
|
||
|
||
from bs4 import BeautifulSoup as bs
|
||
import feedparser
|
||
from httpx import AsyncClient
|
||
|
||
from nonebot_bison.post import Post
|
||
from nonebot_bison.types import Category, RawPost, Target
|
||
from nonebot_bison.utils import text_similarity
|
||
from nonebot_bison.utils.site import CookieClientManager, Site
|
||
|
||
from .platform import NewMessage
|
||
|
||
|
||
class RssSite(Site):
|
||
name = "rss"
|
||
schedule_type = "interval"
|
||
schedule_setting: ClassVar[dict] = {"seconds": 30}
|
||
client_mgr = CookieClientManager.from_name(name)
|
||
|
||
|
||
class RssPost(Post):
|
||
async def get_plain_content(self) -> str:
|
||
soup = bs(self.content, "html.parser")
|
||
|
||
for img in soup.find_all("img"):
|
||
img.replace_with("[图片]")
|
||
|
||
for br in soup.find_all("br"):
|
||
br.replace_with("\n")
|
||
|
||
for p in soup.find_all("p"):
|
||
p.insert_after("\n")
|
||
|
||
return soup.get_text()
|
||
|
||
|
||
class Rss(NewMessage):
|
||
categories: ClassVar[dict[Category, str]] = {}
|
||
enable_tag = False
|
||
platform_name = "rss"
|
||
name = "Rss"
|
||
enabled = True
|
||
is_common = True
|
||
site = RssSite
|
||
has_target = True
|
||
|
||
@classmethod
|
||
async def get_target_name(cls, client: AsyncClient, target: Target) -> str | None:
|
||
res = await client.get(target, timeout=10.0)
|
||
feed = feedparser.parse(res.text)
|
||
return feed["feed"]["title"]
|
||
|
||
def get_date(self, post: RawPost) -> int:
|
||
if hasattr(post, "published_parsed"):
|
||
return calendar.timegm(post.published_parsed)
|
||
elif hasattr(post, "updated_parsed"):
|
||
return calendar.timegm(post.updated_parsed)
|
||
else:
|
||
return calendar.timegm(time.gmtime())
|
||
|
||
def get_id(self, post: RawPost) -> Any:
|
||
return post.id
|
||
|
||
async def get_sub_list(self, target: Target) -> list[RawPost]:
|
||
client = await self.ctx.get_client(target)
|
||
res = await client.get(target, timeout=10.0)
|
||
feed = feedparser.parse(res)
|
||
entries = feed.entries
|
||
for entry in entries:
|
||
entry["_target_name"] = feed.feed.title
|
||
return feed.entries
|
||
|
||
def _text_process(self, title: str, desc: str) -> tuple[str | None, str]:
|
||
"""检查标题和描述是否相似，如果相似则标题为None, 否则返回标题和描述"""
|
||
similarity = 1.0 if len(title) == 0 or len(desc) == 0 else text_similarity(title, desc)
|
||
if similarity > 0.8:
|
||
return None, title if len(title) > len(desc) else desc
|
||
|
||
return title, desc
|
||
|
||
async def parse(self, raw_post: RawPost) -> Post:
|
||
title = raw_post.get("title", "")
|
||
soup = bs(raw_post.description, "html.parser")
|
||
desc = raw_post.description
|
||
title, desc = self._text_process(title, desc)
|
||
pics = [x.attrs["src"] for x in soup("img")]
|
||
if raw_post.get("media_content"):
|
||
for media in raw_post["media_content"]:
|
||
if media.get("medium") == "image" and media.get("url"):
|
||
pics.append(media.get("url"))
|
||
return RssPost(
|
||
self,
|
||
content=desc,
|
||
title=title,
|
||
url=raw_post.link,
|
||
images=pics,
|
||
nickname=raw_post["_target_name"],
|
||
)
|