Skip to content

Commit

Permalink
Переход на disnake (#42)
Browse files Browse the repository at this point in the history
* add disnake

* fix requirements

* ТУПО ЗАМЕНЯЕМ DISCORD НА DISNAKE

* Fix

* ffixx

* кк

* Fix?

* fix?

* fix
  • Loading branch information
Schrodinger71 authored Jan 3, 2025
1 parent 1d60b69 commit 85abb88
Show file tree
Hide file tree
Showing 38 changed files with 154 additions and 153 deletions.
6 changes: 3 additions & 3 deletions bot_init.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,9 +3,9 @@
Настроены необходимые параметры для запуска и обработки команд.
"""

import discord
from discord.ext import commands
import disnake
from disnake.ext import commands

bot = commands.Bot(
command_prefix="&", help_command=None, intents=discord.Intents.all()
command_prefix="&", help_command=None, intents=disnake.Intents.all()
)
10 changes: 5 additions & 5 deletions commands/adt_team/add_role_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
Модуль содержащий команду add_role
"""

import discord
from discord.utils import get
import disnake
from disnake.utils import get

from bot_init import bot
from commands.misc.check_roles import has_any_role_by_id
Expand All @@ -12,7 +12,7 @@

@bot.command()
@has_any_role_by_id(HEAD_ADT_TEAM)
async def add_role(ctx, user: discord.Member, *role_names: str):
async def add_role(ctx, user: disnake.Member, *role_names: str):
"""
Добавляет одну или несколько ролей указанному пользователю.
"""
Expand Down Expand Up @@ -42,12 +42,12 @@ async def add_role(ctx, user: discord.Member, *role_names: str):
# Добавляем роль пользователю
await user.add_roles(role)
added_roles.append(role.name)
except discord.Forbidden:
except disnake.Forbidden:
errors.append(
f"⚠️ У бота недостаточно прав для добавления роли '{role.name}' "
f"пользователю {user.mention}."
)
except discord.HTTPException as e:
except disnake.HTTPException as e:
print(f"Ошибка при добавлении роли '{role.name}': {e}")
errors.append(
f"❌ Не удалось добавить роль '{role.name}' для {user.mention} из-за ошибки: {e}"
Expand Down
12 changes: 6 additions & 6 deletions commands/adt_team/add_vacation_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
Модуль команды add_vacation
"""

import discord
import disnake

from bot_init import bot
from commands.misc.check_roles import has_any_role_by_id
Expand All @@ -11,7 +11,7 @@

@bot.command()
@has_any_role_by_id(HEAD_ADT_TEAM)
async def add_vacation(ctx, user: discord.Member, end_date: str, reason: str):
async def add_vacation(ctx, user: disnake.Member, end_date: str, reason: str):
"""
Выдача отпуска пользователю. Добавляется роль отпуска с указанием срока и причины.
"""
Expand Down Expand Up @@ -42,10 +42,10 @@ async def add_vacation(ctx, user: discord.Member, end_date: str, reason: str):
)

# Создаем Embed для уведомления в админ-канале
embed = discord.Embed(
embed = disnake.Embed(
title="Выдача отпуска",
description=f"{ctx.author.mention} выдал(а) отпуск для {user.mention}.",
color=discord.Color.purple(),
color=disnake.Color.purple(),
)
embed.add_field(name="Пользователь", value=user.mention, inline=False)
embed.add_field(
Expand All @@ -58,11 +58,11 @@ async def add_vacation(ctx, user: discord.Member, end_date: str, reason: str):
# Отправляем Embed в админ-канал
await admin_channel.send(embed=embed)

except discord.Forbidden:
except disnake.Forbidden:
await ctx.send(
"⚠️ Ошибка: У бота недостаточно прав для добавления роли."
)
except discord.HTTPException as e:
except disnake.HTTPException as e:
await ctx.send(f"❌ Ошибка: Не удалось добавить роль. Подробнее: {e}")
except Exception as e:
print(f"Неожиданная ошибка: {e}")
Expand Down
12 changes: 6 additions & 6 deletions commands/adt_team/end_vacation_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
Модуль команды end_vacation
"""

import discord
import disnake

from bot_init import bot
from commands.misc.check_roles import has_any_role_by_id
Expand All @@ -11,7 +11,7 @@

@bot.command()
@has_any_role_by_id(HEAD_ADT_TEAM)
async def end_vacation(ctx, user: discord.Member):
async def end_vacation(ctx, user: disnake.Member):
"""
Завершает отпуск указанного пользователя, удаляя роль отпуска.
"""
Expand Down Expand Up @@ -40,10 +40,10 @@ async def end_vacation(ctx, user: discord.Member):
)

# Создаем Embed для уведомления в админ-канал
embed = discord.Embed(
embed = disnake.Embed(
title="Окончание отпуска",
description=f"{ctx.author.mention} завершил(а) отпуск для {user.mention}.",
color=discord.Color.purple(),
color=disnake.Color.purple(),
)
embed.set_author(name=ctx.author.name, icon_url=ctx.author.avatar.url)
embed.add_field(name="Пользователь", value=user.mention, inline=False)
Expand All @@ -52,9 +52,9 @@ async def end_vacation(ctx, user: discord.Member):
# Отправляем Embed в админ-канал
await admin_channel.send(embed=embed)

except discord.Forbidden:
except disnake.Forbidden:
await ctx.send("⚠️ Ошибка: У бота недостаточно прав для снятия роли.")
except discord.HTTPException as e:
except disnake.HTTPException as e:
await ctx.send(f"❌ Ошибка: Не удалось снять роль. Подробнее: {e}")
except Exception as e:
print(f"Неожиданная ошибка: {e}")
Expand Down
12 changes: 6 additions & 6 deletions commands/adt_team/extend_vacation_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
Модуль команды extend_vacation
"""

import discord
import disnake

from bot_init import bot
from commands.misc.check_roles import has_any_role_by_id
Expand All @@ -12,7 +12,7 @@
@bot.command()
@has_any_role_by_id(HEAD_ADT_TEAM)
async def extend_vacation(
ctx, user: discord.Member, new_end_date: str, reason: str
ctx, user: disnake.Member, new_end_date: str, reason: str
):
"""
Продление отпуска пользователю. Обновляется срок отпуска и причина.
Expand All @@ -39,10 +39,10 @@ async def extend_vacation(

try:
# Создаем Embed для уведомления в админ-канале
embed = discord.Embed(
embed = disnake.Embed(
title="Продление отпуска",
description=f"{ctx.author.mention} продлил отпуск для {user.mention}.",
color=discord.Color.purple(),
color=disnake.Color.purple(),
)
embed.add_field(name="Пользователь", value=user.mention, inline=False)
embed.add_field(
Expand All @@ -62,11 +62,11 @@ async def extend_vacation(
f"✅ Срок отпуска {user.mention} был успешно продлен до {new_end_date}."
)

except discord.Forbidden:
except disnake.Forbidden:
await ctx.send(
"⚠️ Ошибка: У бота недостаточно прав для отправки уведомлений."
)
except discord.HTTPException as e:
except disnake.HTTPException as e:
await ctx.send(f"❌ Ошибка: Не удалось продлить отпуск. Подробнее: {e}")
except Exception as e:
print(f"Неожиданная ошибка: {e}")
Expand Down
10 changes: 5 additions & 5 deletions commands/adt_team/new_team_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@
Модуль команды new_team
"""

import discord
import disnake

from bot_init import bot
from commands.misc.check_roles import has_any_role_by_id
Expand All @@ -11,7 +11,7 @@

@bot.command()
@has_any_role_by_id(HEAD_ADT_TEAM)
async def new_team(ctx, user: discord.Member, *roles: discord.Role):
async def new_team(ctx, user: disnake.Member, *roles: disnake.Role):
"""
Команда для назначения пользователя на должность в команде.
Требует две роли: <роль отдела> и <роль должности>.
Expand All @@ -37,11 +37,11 @@ async def new_team(ctx, user: discord.Member, *roles: discord.Role):
try:
await user.add_roles(role)
assigned_roles.append(role.name)
except discord.Forbidden:
except disnake.Forbidden:
await ctx.send(
f"⚠️ У бота нет прав для добавления роли **{role.name}** у {user.mention}."
)
except discord.HTTPException as e:
except disnake.HTTPException as e:
await ctx.send(
f"❌ Ошибка при добавлении роли **{role.name}**: {str(e)}"
)
Expand All @@ -60,7 +60,7 @@ async def new_team(ctx, user: discord.Member, *roles: discord.Role):
if len(assigned_roles) == 2:
admin_channel = bot.get_channel(ADMIN_TEAM)
if admin_channel:
embed = discord.Embed(
embed = disnake.Embed(
title="Назначение на должность",
description=f"{ctx.author.mention} назначает {user.mention}",
color=role_position.color,
Expand Down
10 changes: 5 additions & 5 deletions commands/adt_team/remove_role_command.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,8 +2,8 @@
Модуль команды remove_role
"""

import discord
from discord.utils import get
import disnake
from disnake.utils import get

from bot_init import bot
from commands.misc.check_roles import has_any_role_by_id
Expand All @@ -12,7 +12,7 @@

@bot.command()
@has_any_role_by_id(HEAD_ADT_TEAM)
async def remove_role(ctx, user: discord.Member, *role_names: str):
async def remove_role(ctx, user: disnake.Member, *role_names: str):
"""
Команда для снятия указанных ролей у пользователя.
"""
Expand Down Expand Up @@ -48,11 +48,11 @@ async def remove_role(ctx, user: discord.Member, *role_names: str):
# Пытаемся снять роль
await user.remove_roles(role)
removed_roles.append(role.name)
except discord.Forbidden:
except disnake.Forbidden:
await ctx.send(
f"⚠️ У бота нет прав для снятия роли '{role.name}' у {user.name}."
)
except discord.HTTPException as e:
except disnake.HTTPException as e:
await ctx.send(f"❌ Ошибка при снятии роли '{role.name}': {str(e)}")
except Exception as e:
await ctx.send(
Expand Down
10 changes: 5 additions & 5 deletions commands/adt_team/remove_team_command.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import discord
import disnake
from bot_init import bot
from commands.misc.check_roles import has_any_role_by_id
from config import ADMIN_TEAM, HEAD_ADT_TEAM
Expand Down Expand Up @@ -91,9 +91,9 @@ async def send_results(ctx, removed_roles, errors):
@has_any_role_by_id(HEAD_ADT_TEAM)
async def remove_team(
ctx,
user: discord.Member,
role_dep: discord.Role,
role_job: discord.Role,
user: disnake.Member,
role_dep: disnake.Role,
role_job: disnake.Role,
*,
reason: str,
):
Expand Down Expand Up @@ -128,7 +128,7 @@ async def remove_team(

# Если обе роли успешно удалены, отправляем Embed
if len(removed_roles) == 2:
embed = discord.Embed(
embed = disnake.Embed(
title="Снятие с должности",
description=f"{ctx.author.mention} снял с должности {user.mention}.",
color=role_job.color,
Expand Down
6 changes: 3 additions & 3 deletions commands/adt_team/team_help_command.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
import discord
import disnake

from bot_init import bot
from config import ADMIN_TEAM
Expand Down Expand Up @@ -90,10 +90,10 @@ async def team_help(ctx):
embed_description += f"- {note}\n"

# Создание Embed сообщения
embed = discord.Embed(
embed = disnake.Embed(
title="📚 Команды для управления сотрудниками отдела",
description=embed_description,
color=discord.Color.gold(),
color=disnake.Color.gold(),
)

# Информация об отправителе команды
Expand Down
16 changes: 8 additions & 8 deletions commands/adt_team/tweak_team_command.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
import discord
from discord.ext import commands
import disnake
from disnake.ext import commands

from bot_init import bot
from commands.misc.check_roles import has_any_role_by_id
Expand All @@ -10,9 +10,9 @@
@has_any_role_by_id(HEAD_ADT_TEAM)
async def tweak_team(
ctx: commands.Context,
user: discord.Member,
old_role: discord.Role,
new_role: discord.Role,
user: disnake.Member,
old_role: disnake.Role,
new_role: disnake.Role,
reason: str,
):
"""
Expand Down Expand Up @@ -63,7 +63,7 @@ async def tweak_team(
color = new_role.color # Цвет для Embed сообщения

# Создаем Embed сообщение для лог-канала
embed = discord.Embed(
embed = disnake.Embed(
title=action,
description=action_description,
color=color,
Expand All @@ -87,11 +87,11 @@ async def tweak_team(
f"✅ Роль **{old_role.name}** была успешно заменена на **{new_role.name}** у {user.name}. Причина: {reason}" # pylint: disable=C0301
)

except discord.Forbidden:
except disnake.Forbidden:
await ctx.send(
"⚠️ У бота нет прав для изменения ролей. Пожалуйста, проверьте права бота."
)
except discord.HTTPException as e:
except disnake.HTTPException as e:
await ctx.send(f"❌ Произошла ошибка при изменении ролей: {e}")
print(f"Ошибка при изменении ролей: {e}")
except Exception as e:
Expand Down
8 changes: 4 additions & 4 deletions commands/dbCommand/help_command.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,9 @@
import discord
from discord.ext import commands
import disnake
from disnake.ext import commands

from bot_init import bot

COLOR = discord.Color.dark_purple()
COLOR = disnake.Color.dark_purple()

# Список команд и их описаний для работы с базой данных
DB_COMMANDS = [
Expand All @@ -25,7 +25,7 @@ async def db_help(ctx: commands.Context):
"""
try:
# Создаем embed
embed = discord.Embed(
embed = disnake.Embed(
title="Команды для управления MariaDB",
description="Список доступных команд для работы с базой данных:",
color=COLOR,
Expand Down
Loading

0 comments on commit 85abb88

Please sign in to comment.