Skip to content

fix: commands.commandに #37

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
sharkbot-neko wants to merge 2 commits into
base: main
Choose a base branch
from
+107 −30
Open

fix: commands.commandに #37

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
14eba33
fix: commands.commandに
sharkbot-neko Aug 20, 2026
df0ccbd
fix: エラー処理
sharkbot-neko Aug 20, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
137 changes: 107 additions & 30 deletions src/cogs/admin.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -19,16 +19,6 @@


class Admin(commands.Cog):
admin_group = app_commands.Group(
name="admin",
description="管理者向けコマンド",
)
reset_group = app_commands.Group(
name="reset",
description="管理者向けリセットコマンド",
parent=admin_group,
)

def __init__(self, bot: commands.Bot):
self.bot = bot
self._relearn_lock = asyncio.Lock()
Expand Down Expand Up @@ -101,39 +91,50 @@ def _reset_and_relearn_sync(self) -> tuple[int, int, int]:
update_compounds(self.bot.db)
return learned_message_count, cleared_token_count, recalculated_token_count

@reset_group.command(
@commands.command(
name="learn",
description="学習データをリセットして全メッセージを再学習します",
)
async def reset_learn(self, interaction: discord.Interaction):
async def reset_learn(self, ctx: commands.Context):
embed_helper = EmbedHelper(function_name="Admin Reset learn")

if ADMIN_USER_ID is None:
embed = embed_helper.create_error_embed(
title="設定エラー",
description="config.py に ADMIN_USER_ID が設定されていません。",
)
await interaction.response.send_message(embed=embed, ephemeral=True)
await ctx.reply(embed=embed)
return

if not self._is_admin_user(interaction.user.id):
if not self._is_admin_user(ctx.author.id):
embed = embed_helper.create_error_embed(
title="権限エラー",
description="このコマンドは管理者のみ実行できます。",
)
await interaction.response.send_message(embed=embed, ephemeral=True)
await ctx.reply(embed=embed)
return

if self._relearn_lock.locked():
embed = embed_helper.create_warning_embed(
title="処理中",
description="現在、再学習処理を実行中です。完了まで待ってください。",
)
await interaction.response.send_message(embed=embed, ephemeral=True)
try:
await ctx.author.send(embed=embed)
except discord.Forbidden:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
except discord.HTTPException:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
return

await interaction.response.defer(ephemeral=True, thinking=True)

async with self._relearn_lock:
try:
(
Expand All @@ -146,7 +147,20 @@ async def reset_learn(self, interaction: discord.Interaction):
title="再学習エラー",
description="再学習中にエラーが発生しました。ログを確認してください。",
)
await interaction.followup.send(embed=embed, ephemeral=True)
try:
await ctx.author.send(embed=embed)
except discord.Forbidden:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
except discord.HTTPException:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
print(f"[Admin] Reset learn command failed: {error}")
return

Expand All @@ -159,41 +173,78 @@ async def reset_learn(self, interaction: discord.Interaction):
f"tokensを再計算したメッセージ: {recalculated_token_count}件"
),
)
await interaction.followup.send(embed=embed, ephemeral=True)
try:
await ctx.author.send(embed=embed)
except discord.Forbidden:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
except discord.HTTPException:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return

@admin_group.command(
@commands.command(
name="server_stats",
description="導入サーバーごとのメッセージ数と収集ユーザー数を表示します",
)
async def server_stats(self, interaction: discord.Interaction):
async def server_stats(self, ctx: commands.Context):
embed_helper = EmbedHelper(function_name="Admin Server Stats")

if ADMIN_USER_ID is None:
embed = embed_helper.create_error_embed(
title="設定エラー",
description="config.py に ADMIN_USER_ID が設定されていません。",
)
await interaction.response.send_message(embed=embed, ephemeral=True)
try:
await ctx.author.send(embed=embed)
except discord.Forbidden:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
except discord.HTTPException:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
return

if not self._is_admin_user(interaction.user.id):
if not self._is_admin_user(ctx.author.id):
embed = embed_helper.create_error_embed(
title="権限エラー",
description="このコマンドは管理者のみ実行できます。",
)
await interaction.response.send_message(embed=embed, ephemeral=True)
await ctx.reply(embed=embed)
return

await interaction.response.defer(ephemeral=True, thinking=True)

try:
stats = await asyncio.to_thread(get_guild_collection_stats, self.bot.db)
except Exception as error:
embed = embed_helper.create_error_embed(
title="集計エラー",
description="サーバー集計の取得中にエラーが発生しました。",
)
await interaction.followup.send(embed=embed, ephemeral=True)
try:
await ctx.author.send(embed=embed)
except discord.Forbidden:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
except discord.HTTPException:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
print(f"[Admin] Collection stats command failed: {error}")
return

Expand All @@ -203,7 +254,20 @@ async def server_stats(self, interaction: discord.Interaction):
title="データなし",
description="表示可能なサーバー情報がありません。",
)
await interaction.followup.send(embed=embed, ephemeral=True)
try:
await ctx.author.send(embed=embed)
except discord.Forbidden:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
except discord.HTTPException:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
return

max_desc_len = 3800
Expand All @@ -223,7 +287,20 @@ async def server_stats(self, interaction: discord.Interaction):
title="導入サーバー集計",
description="\n".join(description_lines),
)
await interaction.followup.send(embed=embed, ephemeral=True)
try:
await ctx.author.send(embed=embed)
except discord.Forbidden:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return
except discord.HTTPException:
await ctx.reply(embed_helper.create_error_embed(
title="DMに送信できませんでした。",
description="DMを受け取れるようにしてください。"
))
return


async def setup(bot: commands.Bot):
Expand Down