mirror of
https://github.com/LiteyukiStudio/nonebot-plugin-acgnshow.git
synced 2025-06-02 18:25:25 +00:00
106 lines
2.6 KiB
Python
106 lines
2.6 KiB
Python
import re
|
||
import random
|
||
import datetime
|
||
import json
|
||
from pathlib import Path
|
||
from .config import config
|
||
|
||
|
||
bgpath: Path = Path(config.acgnshow_bgimage_path)
|
||
bg_list: list[Path] = [
|
||
f
|
||
for f in bgpath.iterdir()
|
||
if f.suffix
|
||
in (
|
||
".jpg",
|
||
".png",
|
||
".jpeg",
|
||
".gif",
|
||
".webp",
|
||
".bmp",
|
||
".svg",
|
||
".ico",
|
||
".tiff",
|
||
".tif",
|
||
".jfif",
|
||
".jpe",
|
||
".jif",
|
||
".jfi",
|
||
".jfif",
|
||
)
|
||
]
|
||
|
||
|
||
def choose_random_bgimage() -> str:
|
||
"""
|
||
从背景图片文件夹中随机选择一张图片，返回图片的file协议uri地址
|
||
"""
|
||
# randomfile = random.choice(os.listdir(bgpath)) # 天哪
|
||
|
||
return random.choice(bg_list).as_uri()
|
||
|
||
|
||
def convert_timestamp(timestamp, strformat: str = "%Y-%m-%d %H:%M:%S") -> str:
|
||
"""
|
||
将时间戳转换为日期格式
|
||
|
||
:param timestamp: unix 时间戳
|
||
:param strformat: 日期格式，默认为 "%Y-%m-%d %H:%M:%S"
|
||
|
||
:return: yyyy-mm-dd hh:mm:ss时间
|
||
"""
|
||
return datetime.datetime.fromtimestamp(timestamp).strftime(strformat)
|
||
|
||
|
||
def extract_banner_url(value) -> str:
|
||
a = json.loads(value)
|
||
url = "https:" + a["banner"]["url"]
|
||
return url
|
||
|
||
|
||
def add_https_to_urls(html_content):
|
||
"""
|
||
为 HTML 内容中的缺失 https: 前缀的 URL 添加 https: 前缀
|
||
|
||
:param html_content: 包含 HTML 的字符串
|
||
:return: 修正后的 HTML 字符串
|
||
"""
|
||
# 使用正则表达式查找所有以 "//" 开头的 URL
|
||
updated_html_content = re.sub(r'(?<=src=["\'])//', "https://", html_content)
|
||
return updated_html_content
|
||
|
||
|
||
def split_html_into_fragments(html_content):
|
||
"""
|
||
将 HTML 内容按照元素分割成多个片段，并存储在列表中
|
||
|
||
:param html_content: 包含 HTML 的字符串
|
||
:return: 存储 HTML 片段的列表
|
||
"""
|
||
# 使用正则表达式匹配 HTML 标签及其内容
|
||
pattern = re.compile(r"(<[^>]+>[^<]*<\/[^>]+>|<[^>]+\/>|<[^>]+>)")
|
||
fragments = pattern.findall(html_content)
|
||
return fragments
|
||
|
||
|
||
def join_fragments_in_groups(fragments, image_count=2):
|
||
"""
|
||
|
||
:param fragments: 存储 HTML 片段的列表
|
||
:param image_count: 每个组包含的图片数量，默认为2
|
||
:return: 拼接后的 HTML 列表
|
||
"""
|
||
grouped_html = []
|
||
count = 0
|
||
buffer = ""
|
||
for group in fragments:
|
||
buffer += group
|
||
if "img" in group:
|
||
count += 1 # 发现图片则计数器+1
|
||
if count >= image_count:
|
||
grouped_html.append(buffer)
|
||
count = 0
|
||
buffer = "" # 初始化计数器和缓冲区
|
||
grouped_html.append(buffer) # 把缓冲区剩余内容一起添加
|
||
return grouped_html
|