many fixes

This commit is contained in:
2023-11-02 00:34:55 +07:00
parent eef67f0c4d
commit a882198ec2
7 changed files with 84 additions and 63 deletions
+39 -31
View File
@@ -1,48 +1,56 @@
from vkbottle import VKAPIError
from vkbottle.user import Message, UserLabeler
from config import edit_message, prefix_bot, user
from custom_rules.permission import Permission
from vkbottle.user import Message
from config import bl, edit_message, prefix_bot, user
bl = UserLabeler()
@bl.message(Permission(), text=[prefix_bot + " +др"])
async def add_to_fr(message: Message):
if not message.reply_message:
await message.answer("❗ Ошибка при выполнении, необходимо пересланное сообщение")
async def add_to_friends_cmd(ctx: Message):
if not ctx.reply_message:
await ctx.answer("❗ Ошибка при выполнении, необходимо пересланное сообщение")
return
friend_id = message.reply_message.from_id
friend_id = ctx.reply_message.from_id
try:
await user.api.friends.add(user_id=friend_id)
await edit_message(message, "✅ Все отлично, запрос отправлен")
await edit_message(ctx, "✅ Все отлично, запрос отправлен")
except Exception as e:
if e.code == 174:
await edit_message(message, "❗ Невозможно добавить в друзья самого себя")
elif e.code == 175:
await edit_message(message, "❗ Невозможно добавить в друзья пользователя, который занес Вас в свой черный список")
elif e.code == 176:
await edit_message(message, "❗ Невозможно добавить в друзья пользователя, который занесен в Ваш черный список")
if VKAPIError[174]:
await edit_message(ctx, "❗ Невозможно добавить в друзья самого себя")
elif VKAPIError[175]:
await edit_message(ctx,
"❗ Невозможно добавить в друзья пользователя, который занес Вас в свой черный список")
elif VKAPIError[176]:
await edit_message(ctx,
"❗ Невозможно добавить в друзья пользователя, который занесен в Ваш черный список")
else:
await edit_message(message, f"❗ Невозможно добавить в друзья пользователя: {e.message}")
await edit_message(ctx, f"❗ Невозможно добавить в друзья пользователя: {e.message}")
@bl.message(Permission(), text=[prefix_bot + " -др"])
async def remove_from_fr(message: Message):
if not message.reply_message:
await edit_message(message, "❗ Ошибка при выполнении, необходимо пересланное сообщение")
async def remove_from_friends_cmd(ctx: Message):
if not ctx.reply_message:
await edit_message(ctx, "❗ Ошибка при выполнении, необходимо пересланное сообщение")
return
friend_id = message.reply_message.from_id
friend_id = ctx.reply_message.from_id
response = await user.api.friends.delete(user_id=friend_id)
if response.success:
if response.friend_deleted:
await edit_message(message, "✅ Друг удален")
elif response.out_request_deleted:
await edit_message(message, "✅ Отменена исходящая заявка")
elif response.in_request_deleted:
await edit_message(message, "Отклонена входящая заявка")
elif response.suggestion_deleted:
await edit_message(message, "✅ Отклонена рекомендация друга")
else:
await edit_message(message, "❗ Произошла ошибка")
if not response.success:
await edit_message(ctx, "❗ Произошла ошибка при удалении друга")
return
if response.friend_deleted:
await edit_message(ctx, "Друг удален")
elif response.out_request_deleted:
await edit_message(ctx, "✅ Отменена исходящая заявка")
elif response.in_request_deleted:
await edit_message(ctx, "✅ Отклонена входящая заявка")
elif response.suggestion_deleted:
await edit_message(ctx, "✅ Отклонена рекомендация друга")
else:
await edit_message(message, "❗ Произошла ошибка при удалении друга")
await edit_message(ctx, "❗ Произошла ошибка")