From cf47a5b2adae16747c1b6bc2f2e37b980dd176ad Mon Sep 17 00:00:00 2001 From: suti7yk5032 Date: Mon, 9 Sep 2024 14:04:19 +0900 Subject: [PATCH] =?UTF-8?q?=E8=87=AA=E5=89=8D=E3=81=AE=E3=83=9C=E3=82=BF?= =?UTF-8?q?=E3=83=B3=E3=82=92=E8=BF=BD=E5=8A=A0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dislocker.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/dislocker.py b/dislocker.py index 7e98f04..64de3fa 100644 --- a/dislocker.py +++ b/dislocker.py @@ -619,6 +619,8 @@ class Bot(discord.Client): for i in range(1, 11): device_register_button = discord.ui.Button(style=discord.ButtonStyle.primary, label=f"{i}", custom_id=f"deviceregister_{str(pc_number)}_{i}") device_register_view.add_item(device_register_button) + device_own_register_button = discord.ui.Button(style=discord.ButtonStyle.primary, label="自前", custom_id=f"device_register_{str(pc_number)}_own") + device_register_view.add_item(device_own_register_button) await interaction.response.send_message(f"# :keyboard: デバイス番号を選んでください！\n>>> # PC番号 | {str(pc_number)}", view=device_register_view, ephemeral=True) @@ -641,7 +643,11 @@ class Bot(discord.Client): if len(custom_id_split) >= 4: if custom_id_split[3] == "quick": reason = custom_id_split[4] - device_number = custom_id_split[2] + if device_number == "own": + device_number = 0 + else: + device_number = custom_id_split[2] + register = bot.register(user_id=interaction.user.id, name=interaction.user.name, display_name=interaction.user.display_name, pc_number=pc_number, device_number=device_number, detail=reason) print(register["result"]) @@ -800,7 +806,11 @@ class Reason(Modal): print(custom_id) custom_id_split = custom_id.split("_") pc_number = custom_id_split[1] - device_number = custom_id_split[2] + if device_number == "own": + device_number = 0 + else: + device_number = custom_id_split[2] + register = bot.register(user_id=interaction.user.id, name=interaction.user.name, display_name=interaction.user.display_name, pc_number=pc_number, device_number=device_number, detail=self.reason_input_form.value) print(register["result"])