forust/homelab
1
0
Fork 0
Code Issues Pull Requests Actions Packages Wiki Activity

chore(userbot): apply ruff check --fix and ruff format

Browse Source 
- ruff check --fix: 210 auto-fixed errors (import sorting, trailing
  whitespace, unused imports, f-string fixups, deprecated annotations)
- ruff format: 104 files reformatted to consistent style
- 268 non-auto-fixable issues remain (S113 requests timeout, etc.)
This commit is contained in:
mr-forust
2026-06-19 12:19:01 +02:00
parent bd9724da69
commit 0506aaaac8
104 changed files with 4338 additions and 5319 deletions
Download Patch File Download Diff File Expand all files Collapse all files
userbot/modules/notes.py
+49 -59
View File
@@ -16,128 +16,118 @@
from pyrogram import Client, errors, filters
from pyrogram.types import Message
from utils.db import db
from utils.handlers import NoteSendHandler
from utils.misc import modules_help, prefix
@Client.on_message(filters.command(["save"], prefix) & filters.me)
@Client.on_message(filters.command(['save'], prefix) & filters.me)
async def save_note(client: Client, message: Message):
await message.edit("<b>Loading...</b>")
await message.edit('<b>Loading...</b>')
try:
chat = await client.get_chat(db.get("core.notes", "chat_id", 0))
chat = await client.get_chat(db.get('core.notes', 'chat_id', 0))
except (errors.RPCError, ValueError, KeyError):
# group is not accessible or isn't created
chat = await client.create_supergroup(
"Userbot_Notes_Filters", "Don't touch this group, please"
)
db.set("core.notes", "chat_id", chat.id)
chat = await client.create_supergroup('Userbot_Notes_Filters', "Don't touch this group, please")
db.set('core.notes', 'chat_id', chat.id)
chat_id = chat.id
if message.reply_to_message and len(message.text.split()) >= 2:
note_name = message.text.split(maxsplit=1)[1]
if message.reply_to_message.media_group_id:
checking_note = db.get("core.notes", f"note{note_name}", False)
checking_note = db.get('core.notes', f'note{note_name}', False)
if not checking_note:
get_media_group = [
_.id
for _ in await client.get_media_group(
message.chat.id, message.reply_to_message.id
)
_.id for _ in await client.get_media_group(message.chat.id, message.reply_to_message.id)
]
try:
message_id = await client.forward_messages(
chat_id, message.chat.id, get_media_group
)
message_id = await client.forward_messages(chat_id, message.chat.id, get_media_group)
except errors.ChatForwardsRestricted:
await message.edit(
"<b>Forwarding messages is restricted by chat admins</b>",
'<b>Forwarding messages is restricted by chat admins</b>',
)
return
note = {
"MESSAGE_ID": str(message_id[1].id),
"MEDIA_GROUP": True,
"CHAT_ID": str(chat_id),
'MESSAGE_ID': str(message_id[1].id),
'MEDIA_GROUP': True,
'CHAT_ID': str(chat_id),
}
db.set("core.notes", f"note{note_name}", note)
await message.edit(f"<b>Note {note_name} saved</b>")
db.set('core.notes', f'note{note_name}', note)
await message.edit(f'<b>Note {note_name} saved</b>')
else:
await message.edit("<b>This note already exists</b>")
await message.edit('<b>This note already exists</b>')
else:
checking_note = db.get("core.notes", f"note{note_name}", False)
checking_note = db.get('core.notes', f'note{note_name}', False)
if not checking_note:
try:
message_id = await message.reply_to_message.forward(chat_id)
except errors.ChatForwardsRestricted:
message_id = await message.copy(chat_id)
note = {
"MEDIA_GROUP": False,
"MESSAGE_ID": str(message_id.id),
"CHAT_ID": str(chat_id),
'MEDIA_GROUP': False,
'MESSAGE_ID': str(message_id.id),
'CHAT_ID': str(chat_id),
}
db.set("core.notes", f"note{note_name}", note)
await message.edit(f"<b>Note {note_name} saved</b>")
db.set('core.notes', f'note{note_name}', note)
await message.edit(f'<b>Note {note_name} saved</b>')
else:
await message.edit("<b>This note already exists</b>")
await message.edit('<b>This note already exists</b>')
elif len(message.text.split()) >= 3:
note_name = message.text.split(maxsplit=1)[1].split()[0]
checking_note = db.get("core.notes", f"note{note_name}", False)
checking_note = db.get('core.notes', f'note{note_name}', False)
if not checking_note:
message_id = await client.send_message(
chat_id, message.text.split(note_name)[1].strip()
)
message_id = await client.send_message(chat_id, message.text.split(note_name)[1].strip())
note = {
"MEDIA_GROUP": False,
"MESSAGE_ID": str(message_id.id),
"CHAT_ID": str(chat_id),
'MEDIA_GROUP': False,
'MESSAGE_ID': str(message_id.id),
'CHAT_ID': str(chat_id),
}
db.set("core.notes", f"note{note_name}", note)
await message.edit(f"<b>Note {note_name} saved</b>")
db.set('core.notes', f'note{note_name}', note)
await message.edit(f'<b>Note {note_name} saved</b>')
else:
await message.edit("<b>This note already exists</b>")
await message.edit('<b>This note already exists</b>')
else:
await message.edit(
f"<b>Example: <code>{prefix}save note_name</code></b>",
f'<b>Example: <code>{prefix}save note_name</code></b>',
)
@Client.on_message(filters.command("note", prefix) & filters.me)
@Client.on_message(filters.command('note', prefix) & filters.me)
async def note_send(client: Client, message: Message):
handler = NoteSendHandler(client, message)
await handler.handle_note_send()
@Client.on_message(filters.command(["notes"], prefix) & filters.me)
@Client.on_message(filters.command(['notes'], prefix) & filters.me)
async def notes(_, message: Message):
await message.edit("<b>Loading...</b>")
text = "Available notes:\n\n"
collection = db.get_collection("core.notes")
await message.edit('<b>Loading...</b>')
text = 'Available notes:\n\n'
collection = db.get_collection('core.notes')
for note in collection.keys():
if note[:4] == "note":
text += f"<code>{note[4:]}</code>\n"
if note[:4] == 'note':
text += f'<code>{note[4:]}</code>\n'
await message.edit(text)
@Client.on_message(filters.command(["clear"], prefix) & filters.me)
@Client.on_message(filters.command(['clear'], prefix) & filters.me)
async def clear_note(_, message: Message):
if len(message.text.split()) >= 2:
note_name = message.text.split(maxsplit=1)[1]
find_note = db.get("core.notes", f"note{note_name}", False)
find_note = db.get('core.notes', f'note{note_name}', False)
if find_note:
db.remove("core.notes", f"note{note_name}")
await message.edit(f"<b>Note {note_name} deleted</b>")
db.remove('core.notes', f'note{note_name}')
await message.edit(f'<b>Note {note_name} deleted</b>')
else:
await message.edit("<b>There is no such note</b>")
await message.edit('<b>There is no such note</b>')
else:
await message.edit(f"<b>Example: <code>{prefix}clear note_name</code></b>")
await message.edit(f'<b>Example: <code>{prefix}clear note_name</code></b>')
modules_help["notes"] = {
"save [name]*": "Save note",
"note [name]*": "Get saved note",
"notes": "Get note list",
"clear [name]*": "Delete note",
modules_help['notes'] = {
'save [name]*': 'Save note',
'note [name]*': 'Get saved note',
'notes': 'Get note list',
'clear [name]*': 'Delete note',
}