mirror of
https://github.com/yasirarism/MissKatyPyro.git
synced 2025-12-29 09:44:50 +00:00
a7008c664e
* ci: Update .deepsource.toml
* ci: Update .deepsource.toml
* style: format code with black and isort
Format code with black and isort
This commit fixes the style issues introduced in 0fb651a according to the output
from Black and isort.
Details: https://app.deepsource.com/gh/yasirarism/MissKatyPyro/transform/d8f2f66e-b496-4686-aca6-9830236eda12/
---------
Co-authored-by: deepsource-io[bot] <42547082+deepsource-io[bot]@users.noreply.github.com>
Co-authored-by: deepsource-autofix[bot] <62050782+deepsource-autofix[bot]@users.noreply.github.com>
132 lines
4.5 KiB
Python
132 lines
4.5 KiB
Python
"""
|
|
MIT License
|
|
|
|
Copyright (c) 2021 TheHamkerCat
|
|
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
|
|
The above copyright notice and this permission notice shall be included in all
|
|
copies or substantial portions of the Software.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
|
SOFTWARE.
|
|
"""
|
|
from re import findall
|
|
|
|
from pyrogram import filters
|
|
|
|
from database.notes_db import delete_note, get_note, get_note_names, save_note
|
|
from misskaty import app
|
|
from misskaty.core.decorator.errors import capture_err
|
|
from misskaty.core.decorator.permissions import adminsOnly
|
|
from misskaty.core.decorator.ratelimiter import ratelimiter
|
|
from misskaty.core.keyboard import ikb
|
|
from misskaty.helper.functions import extract_text_and_keyb
|
|
|
|
__MODULE__ = "Notes"
|
|
__HELP__ = """/notes To Get All The Notes In The Chat.
|
|
|
|
(/save, /addnote) [NOTE_NAME] To Save A Note (Can be a sticker or text).
|
|
|
|
#NOTE_NAME To Get A Note.
|
|
|
|
(/clear, /delnote) [NOTE_NAME] To Delete A Note.
|
|
"""
|
|
|
|
|
|
@app.on_message(filters.command(["addnote", "save"]) & ~filters.private)
|
|
@adminsOnly("can_change_info")
|
|
@ratelimiter
|
|
async def save_notee(_, message):
|
|
if len(message.command) < 2 or not message.reply_to_message:
|
|
await message.reply(
|
|
text="**Usage:**\nReply to a text or sticker with /addnote [NOTE_NAME] to save it.",
|
|
)
|
|
|
|
elif not message.reply_to_message.text and not message.reply_to_message.sticker:
|
|
await message.reply("__**You can only save text or stickers in notes.**__")
|
|
else:
|
|
name = message.text.split(None, 1)[1].strip()
|
|
if not name:
|
|
return await message.reply("**Usage**\n__/save [NOTE_NAME]__")
|
|
_type = "text" if message.reply_to_message.text else "sticker"
|
|
note = {
|
|
"type": _type,
|
|
"data": message.reply_to_message.text.markdown
|
|
if _type == "text"
|
|
else message.reply_to_message.sticker.file_id,
|
|
}
|
|
chat_id = message.chat.id
|
|
await save_note(chat_id, name, note)
|
|
await message.reply(f"__**Saved note {name}.**__")
|
|
|
|
|
|
@app.on_message(filters.command("notes") & ~filters.private)
|
|
@capture_err
|
|
@ratelimiter
|
|
async def get_notes(_, message):
|
|
chat_id = message.chat.id
|
|
|
|
_notes = await get_note_names(chat_id)
|
|
|
|
if not _notes:
|
|
return await message.reply("**No notes in this chat.**")
|
|
_notes.sort()
|
|
msg = f"List of notes in {message.chat.title} - {message.chat.id}\n"
|
|
for note in _notes:
|
|
msg += f"**-** `{note}`\n"
|
|
await message.reply(msg)
|
|
|
|
|
|
@app.on_message(filters.regex(r"^#.+") & filters.text & ~filters.private)
|
|
@capture_err
|
|
@ratelimiter
|
|
async def get_one_note(_, message):
|
|
name = message.text.replace("#", "", 1)
|
|
if not name:
|
|
return
|
|
_note = await get_note(message.chat.id, name)
|
|
if not _note:
|
|
return
|
|
if _note["type"] == "text":
|
|
data = _note["data"]
|
|
keyb = None
|
|
if findall(r"\[.+\,.+\]", data):
|
|
if keyboard := extract_text_and_keyb(ikb, data):
|
|
data, keyb = keyboard
|
|
await message.reply_text(
|
|
data,
|
|
reply_markup=keyb,
|
|
disable_web_page_preview=True,
|
|
)
|
|
else:
|
|
await message.reply_sticker(_note["data"])
|
|
|
|
|
|
@app.on_message(filters.command(["delnote", "clear"]) & ~filters.private)
|
|
@adminsOnly("can_change_info")
|
|
@ratelimiter
|
|
async def del_note(_, message):
|
|
if len(message.command) == 1:
|
|
return await message.reply("**Usage**\n__/delnote [NOTE_NAME]__")
|
|
name = message.text.split(None, 1)[1].strip()
|
|
if not name:
|
|
return await message.reply("**Usage**\n__/delnote [NOTE_NAME]__")
|
|
|
|
chat_id = message.chat.id
|
|
|
|
deleted = await delete_note(chat_id, name)
|
|
if deleted:
|
|
await message.reply(f"**Deleted note {name} successfully.**")
|
|
else:
|
|
await message.reply("**No such note.**")
|