TeamPGM/PagerMaid_Plugins_Pyro
2
0
Fork 0
mirror of https://github.com/TeamPGM/PagerMaid_Plugins_Pyro.git synced 2024-11-16 11:42:15 +00:00
Code Issues Actions Projects Releases Wiki Activity

aff 保存预览状态

Browse Source
This commit is contained in:
omg-xtao 2023-01-10 14:51:38 +08:00 committed by GitHub
parent 41991bee1a
commit 14872ca808
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
aff/main.py
View File

@ -1,21 +1,26 @@
import contextlib
from typing import Tuple
from pagermaid.enums import Message
from pagermaid.single_utils import sqlite
from pagermaid.services import sqlite
from pagermaid.listener import listener
def get_aff() -> str:
return sqlite.get("aff.text", "")
def get_aff() -> Tuple[str, bool]:
return sqlite.get("aff.text", ""), sqlite.get("aff.web_page", False)
def set_aff(text: str) -> None:
sqlite.update({"aff.text": text})
def set_aff(text: str, web_page: bool = False) -> None:
sqlite.update({"aff.text": text, "aff.web_page": web_page})
def del_aff() -> None:
if get_aff():
text, web_page = get_aff()
if text:
del sqlite["aff.text"]
if web_page:
del sqlite["aff.web_page"]
@listener(command="aff",
@ -23,18 +28,19 @@ def del_aff() -> None:
parameters="<save|remove> (可选，回复一条消息，用于保存|删除aff信息)")
async def aff(message: Message):
if not message.parameter:
msg = get_aff()
msg, web_page = get_aff()
if not msg:
return await message.edit("出错了呜呜呜 ~ Aff消息不存在。\n(你有提前保存好嘛？)")
with contextlib.suppress(Exception):
await message.edit(msg, disable_web_page_preview=False)
await message.edit(msg, disable_web_page_preview=not web_page)
elif message.parameter[0] == "save":
if not message.reply_to_message:
return await message.edit("出错了呜呜呜 ~ 请回复一条消息以保存新的Aff信息。")
text = message.reply_to_message.text or message.reply_to_message.caption
web_page = message.reply_to_message.web_page or False
if not text:
return await message.edit("出错了呜呜呜 ~ 请回复一条消息以保存新的Aff信息。")
set_aff(text.html)
set_aff(text.html, web_page)
await message.edit("好耶 ！ Aff信息保存成功。")
elif message.parameter[0] == "remove":
del_aff()