LiteyukiStudio/nonebot-plugin-marshoai
4
4
Fork 0
mirror of https://github.com/LiteyukiStudio/nonebot-plugin-marshoai.git synced 2025-02-07 21:46:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

统一格式(无实质改动) & 添加了图片 (#23)

Browse Source 
* 修复了style和script标签无法去除的问题

* 添加了图片展示 & 统一了引号格式

* 添加了图片展示 & 统一了引号格式

* 统一引号格式并异步处理函数
This commit is contained in:
Nya_Twisuki 2024-12-12 13:47:28 +08:00 committed by GitHub
parent 6a4b0bbd0d
commit b939a48b0b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
6 changed files with 74 additions and 65 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
nonebot_plugin_marshoai/tools/marshoai-megakits/__init__.py
View File

@ -6,32 +6,32 @@ from . import mk_NyaCode
# Twisuki # Twisuki
async def twisuki(): async def twisuki():
return str(mk_Info.twisuki()) return str(await mk_Info.twisuki())
# MegaKits # MegaKits
async def megakits(): async def megakits():
return str(mk_Info.megakits()) return str(await mk_Info.megakits())
# Random Turntable # Random Turntable
async def random_turntable(upper: int, lower: int = "0"): async def random_turntable(upper: int, lower: int = 0):
return str(mk_Common.random_turntable(upper, lower)) return str(await mk_Common.random_turntable(upper, lower))
# Number Calc # Number Calc
async def number_calc(a: str, b: str, op: str): async def number_calc(a: str, b: str, op: str):
return str(mk_Common.number_calc(a, b, op)) return str(await mk_Common.number_calc(a, b, op))
# MorseCode Encrypt # MorseCode Encrypt
async def morse_encrypt(msg: str): async def morse_encrypt(msg: str):
return str(mk_MorseCode.morse_encrypt(msg)) return str(await mk_MorseCode.morse_encrypt(msg))
# MorseCode Decrypt # MorseCode Decrypt
async def morse_decrypt(msg: str): async def morse_decrypt(msg: str):
return str(mk_MorseCode.morse_decrypt(msg)) return str(await mk_MorseCode.morse_decrypt(msg))
# NyaCode Encrypt # NyaCode Encrypt
async def nya_encode(msg: str): async def nya_encode(msg: str):
return str(mk_NyaCode.nya_encode(msg)) return str(await mk_NyaCode.nya_encode(msg))
# NyaCode Decrypt # NyaCode Decrypt
async def nya_decode(msg: str): async def nya_decode(msg: str):
return str(mk_NyaCode.nya_decode(msg)) return str(await mk_NyaCode.nya_decode(msg))

4
nonebot_plugin_marshoai/tools/marshoai-megakits/mk_Common.py
View File

@ -1,11 +1,11 @@
import random import random
# Random Turntable # Random Turntable
async def random_turntable(upper: int, lower: int = "0"): async def random_turntable(upper: int, lower: int):
return random.randint(lower, upper) return random.randint(lower, upper)
# Number Calc # Number Calc
def number_calc(a: str, b: str, op: str): async def number_calc(a: str, b: str, op: str):
a, b = float(a), float(b) a, b = float(a), float(b)
match op: match op:
case "+": case "+":

8
nonebot_plugin_marshoai/tools/marshoai-megakits/mk_Info.py
View File

@ -1,7 +1,7 @@
# Twisuki # Twisuki
def twisuki(): async def twisuki():
return "Twiuski(苏阳)是megakits插件作者, Github : 'https://github.com/Twisuki'" return "Twiuski(苏阳)是megakits插件作者, Github : \"https://github.com/Twisuki\""
# MegaKits # MegaKits
def megakits(): async def megakits():
return "MegaKits插件是一个功能混杂的MarshoAI插件, 由Twisuki(Github : 'https://github.com/Twisuki')开发, 插件仓库 : 'https://github.com/LiteyukiStudio/marsho-toolsets/tree/main/Twisuki/marshoai-megakits'" return "MegaKits插件是一个功能混杂的MarshoAI插件, 由Twisuki(Github : \"https://github.com/Twisuki\")开发, 插件仓库 : \"https://github.com/LiteyukiStudio/marsho-toolsets/tree/main/Twisuki/marshoai-megakits\""

34
nonebot_plugin_marshoai/tools/marshoai-megakits/mk_MorseCode.py
View File

@ -1,37 +1,37 @@
# MorseCode # MorseCode
MorseEncode = { MorseEncode = {
'A': '.-', 'B': '-...', 'C': '-.-.', 'D': '-..', 'E': '.', 'F': '..-.', "A": ".-", "B": "-...", "C": "-.-.", "D": "-..", "E": ".", "F": "..-.",
'G': '--.', 'H': '....', 'I': '..', 'J': '.---', 'K': '-.-', 'L': '.-..', "G": "--.", "H": "....", "I": "..", "J": ".---", "K": "-.-", "L": ".-..",
'M': '--', 'N': '-.', 'O': '---', 'P': '.--.', 'Q': '--.-', 'R': '.-.', "M": "--", "N": "-.", "O": "---", "P": ".--.", "Q": "--.-", "R": ".-.",
'S': '...', 'T': '-', 'U': '..-', 'V': '...-', 'W': '.--', 'X': '-..-', "S": "...", "T": "-", "U": "..-", "V": "...-", "W": ".--", "X": "-..-",
'Y': '-.--', 'Z': '--..', "Y": "-.--", "Z": "--..",
'1': '.----', '2': '..---', '3': '...--', '4': '....-', '5': '.....', "1": ".----", "2": "..---", "3": "...--", "4": "....-", "5": ".....",
'6': '-....', '7': '--...', '8': '---..', '9': '----.', '0': '-----', "6": "-....", "7": "--...", "8": "---..", "9": "----.", "0": "-----",
'.': '.-.-.-', ':': '---...', ',': '--..--', ';': '-.-.-.', ".": ".-.-.-", ":": "---...", ",": "--..--", ";": "-.-.-.",
'?': '..--..', '=': '-...-', '\'': '.----.', '/': '-..-.', "?": "..--..", "=": "-...-", "\'": ".----.", "/": "-..-.",
'!': '-.-.--', '-': '-....-', '_': '..--.-', '\"': '.-..-.', "!": "-.-.--", "-": "-....-", "_": "..--.-", "\"": ".-..-.",
'(': '-.--.', ')': '-.--.-', '$': '...-..-', '&': '....', "(": "-.--.", ")": "-.--.-", "$": "...-..-", "&": "....",
'@': '.--.-.', ' ': ' ' "@": ".--.-.", " ": " "
} }
MorseDecode = {value: key for key, value in MorseEncode.items()} MorseDecode = {value: key for key, value in MorseEncode.items()}
# MorseCode Encrypt # MorseCode Encrypt
def morse_encrypt(msg: str): async def morse_encrypt(msg: str):
result = "" result = ""
msg = msg.upper() msg = msg.upper()
for char in msg: for char in msg:
if char in MorseEncode: if char in MorseEncode:
result += MorseEncode[char] result += MorseEncode[char]
else: else:
result += '..--..' result += "..--.."
result += ' ' result += " "
return result return result
# MorseCode Decrypt # MorseCode Decrypt
def morse_decrypt(msg: str): async def morse_decrypt(msg: str):
result = "" result = ""
msg_arr = msg.split() msg_arr = msg.split()
@ -39,6 +39,6 @@ def morse_decrypt(msg: str):
if char in MorseDecode: if char in MorseDecode:
result += MorseDecode[char] result += MorseDecode[char]
else: else:
result += '?' result += "?"
return result return result

20
nonebot_plugin_marshoai/tools/marshoai-megakits/mk_NyaCode.py
View File

@ -3,26 +3,26 @@ import base64
# NyaCode # NyaCode
NyaCodeCharset = [ NyaCodeCharset = [
'喵', '呜', '?', '~' "喵", "呜", "?", "~"
] ]
NyaCodeSpecialCharset = [ NyaCodeSpecialCharset = [
'唔', '!', '...', '..' "唔", "!", "...", ".."
] ]
NyaCodeEncode = {} NyaCodeEncode = {}
for i in range(64): for i in range(64):
triplet = ''.join(NyaCodeCharset[(i // (4 ** j)) % 4] for j in range(3)) triplet = "".join(NyaCodeCharset[(i // (4 ** j)) % 4] for j in range(3))
NyaCodeEncode[chr(65 + i if i < 26 else 97 + (i - 26) if i < 52 else 48 + (i - 52) if i < 62 else ( NyaCodeEncode[chr(65 + i if i < 26 else 97 + (i - 26) if i < 52 else 48 + (i - 52) if i < 62 else (
43 if i == 62 else 47))] = triplet 43 if i == 62 else 47))] = triplet
NyaCodeDecode = {value: key for key, value in NyaCodeEncode.items()} NyaCodeDecode = {value: key for key, value in NyaCodeEncode.items()}
# NyaCode Encrypt # NyaCode Encrypt
def nya_encode(msg: str): async def nya_encode(msg: str):
msg_b64str = base64.b64encode(msg.encode()).decode().replace('=', '') msg_b64str = base64.b64encode(msg.encode()).decode().replace("=", "")
msg_nyastr = ''.join(NyaCodeEncode[base64_char] for base64_char in msg_b64str) msg_nyastr = "".join(NyaCodeEncode[base64_char] for base64_char in msg_b64str)
result = "" result = ""
for char in msg_nyastr: for char in msg_nyastr:
if char == '呜' and random.random() < 0.5: if char == "呜" and random.random() < 0.5:
result += "!" result += "!"
if random.random() < 0.25: if random.random() < 0.25:
@ -33,8 +33,8 @@ def nya_encode(msg: str):
# NyaCode Decrypt # NyaCode Decrypt
def nya_decode(msg: str): async def nya_decode(msg: str):
msg = msg.replace('唔', '').replace('!', '').replace('.', '') msg = msg.replace("唔", "").replace("!", "").replace(".", "")
msg_nyastr = [] msg_nyastr = []
i = 0 i = 0
if len(msg) % 3 != 0 : if len(msg) % 3 != 0 :
@ -47,7 +47,7 @@ def nya_decode(msg: str):
i += 3 i += 3
except Exception: except Exception:
return "这句话不是正确的猫语" return "这句话不是正确的猫语"
msg_b64str = ''.join(NyaCodeDecode[nya_char] for nya_char in msg_nyastr) msg_b64str = "".join(NyaCodeDecode[nya_char] for nya_char in msg_nyastr)
msg_b64str += "=" * (4 - len(msg_b64str) % 4) msg_b64str += "=" * (4 - len(msg_b64str) % 4)
try: try:
result = base64.b64decode(msg_b64str.encode()).decode() result = base64.b64decode(msg_b64str.encode()).decode()

53
nonebot_plugin_marshoai/tools/marshoai-meogirl/mg_Search.py
View File

@ -6,7 +6,7 @@ import urllib.parse
from bs4 import BeautifulSoup from bs4 import BeautifulSoup
headers = { headers = {
'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/100.0.4896.127 Safari/537.36' "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/100.0.4896.127 Safari/537.36"
} }
async def get_async_data (url): async def get_async_data (url):
@ -19,7 +19,7 @@ async def search(msg : str, num : int):
url = "https://mzh.moegirl.org.cn/index.php?search=" + urllib.parse.quote_plus(msg) url = "https://mzh.moegirl.org.cn/index.php?search=" + urllib.parse.quote_plus(msg)
response = await get_async_data(url) response = await get_async_data(url)
logger.success(f"连接{url}完成, 状态码 : {response.status_code}") logger.success(f"连接\"{url}\"完成, 状态码 : {response.status_code}")
# 正常搜索 # 正常搜索
if response.status_code == 200: if response.status_code == 200:
@ -35,25 +35,25 @@ async def search(msg : str, num : int):
└── li ... └── li ...
└── li ... └── li ...
""" """
soup = BeautifulSoup(response.text, 'html.parser') soup = BeautifulSoup(response.text, "html.parser")
# 检测ul.mw-search-results, 是否有结果 # 检测ul.mw-search-results, 是否有结果
if soup.find('ul', class_='mw-search-results'): ul_tag = soup.find("ul", class_ = "mw-search-results")
ul_tag = soup.select('ul.mw-search-results')[0] if ul_tag:
li_tags = ul_tag.select('li') li_tags = ul_tag.find_all("li")
for li_tag in li_tags: for li_tag in li_tags:
div_heading = li_tag.select('div.mw-search-result-heading')[0] div_heading = li_tag.find("div", class_ = "mw-search-result-heading")
if div_heading: if div_heading:
a_tag = div_heading.select('a')[0] a_tag = div_heading.find("a")
result += a_tag['title'] + "\n" result += a_tag["title"] + "\n"
logger.info(f"搜索到 : \"{a_tag['title']}\"") logger.info(f"搜索到 : \"{a_tag["title"]}\"")
div_result = li_tag.find('div', class_='searchresult') div_result = li_tag.find("div", class_="searchresult")
if div_result: if div_result:
content = str(div_result).replace('<div class=\"searchresult\">', '').replace('</div>', '') content = str(div_result).replace("<div class=\"searchresult\">", "").replace("</div>", "")
content = content.replace('<span class=\"searchmatch\">', '').replace('</span>', '') content = content.replace("<span class=\"searchmatch\">", "").replace("</span>", "")
result += content + "\n\n" result += content + "\n"
num -= 1 num -= 1
if num == 0: if num == 0:
@ -67,10 +67,10 @@ async def search(msg : str, num : int):
# 重定向 # 重定向
elif response.status_code == 302: elif response.status_code == 302:
logger.info(f"\"{msg}\"已被重定向至\"{response.headers.get('location')}\"") logger.info(f"\"{msg}\"已被重定向至\"{response.headers.get("location")}\"")
# 读取重定向结果 # 读取重定向结果
response = await get_async_data(response.headers.get('location')) response = await get_async_data(response.headers.get("location"))
soup = BeautifulSoup(response.text, 'html.parser') soup = BeautifulSoup(response.text, "html.parser")
logger.success("重定向成功") logger.success("重定向成功")
num = 0 num = 0
@ -83,13 +83,22 @@ async def search(msg : str, num : int):
└── p # 人物介绍 └── p # 人物介绍
└── ... └── ...
""" """
if soup.find('div', class_='mw-parser-output'):
div = soup.find('div', class_='mw-parser-output') result += msg + "\n"
p_tags = div.select('p') img = soup.find("img", class_="infobox-image")
if img:
logger.info(f"照片{img["src"]}")
result += f"![ {msg} ]( {img["src"]} ) \n"
div = soup.find("div", class_="mw-parser-output")
if div:
p_tags = div.find_all("p")
for p_tag in p_tags: for p_tag in p_tags:
p = str(p_tag) p = str(p_tag)
p = re.sub(r'<.*?>', '', p) p = re.sub(r"<script.*?</script>|<style.*?</style>", "", p, flags=re.DOTALL)
if p != '': p = re.sub(r"<.*?>", "", p, flags = re.DOTALL)
p = re.sub(r"\[.*?]", "", p, flags = re.DOTALL)
if p != "":
result += str(p) result += str(p)
num += 1 num += 1