[guild] balls drop configuration (#817)

* balls drop configuration

* Update ballsdex/packages/guildconfig/cog.py

Co-authored-by: dormieriancitizen <111535275+dormieriancitizen@users.noreply.github.com>

---------

Co-authored-by: dormieriancitizen <111535275+dormieriancitizen@users.noreply.github.com>

Author: flaree

admin_panel/bd_models/admin/guild.py

@@ -70,8 +70,8 @@ def player(self, obj: BallInstance):
 
 @admin.register(GuildConfig)
 class GuildAdmin(admin.ModelAdmin):
-    list_display = ("guild_id", "spawn_channel", "enabled", "silent", "blacklisted")
-    list_filter = ("enabled", "silent", BlacklistedListFilter)
+    list_display = ("guild_id", "spawn_channel", "enabled", "silent", "manual_drop_enabled", "blacklisted")
+    list_filter = ("enabled", "silent", "manual_drop_enabled", BlacklistedListFilter)
     show_facets = admin.ShowFacets.NEVER  # type: ignore
 
     search_fields = ("guild_id", "spawn_channel")

admin_panel/bd_models/migrations/0018_guildconfig_manual_drop_enabled.py

@@ -0,0 +1,17 @@
+# Generated by Django 6.0 on 2026-07-05 17:45
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [("bd_models", "0017_ballgroup")]
+
+    operations = [
+        migrations.AddField(
+            model_name="guildconfig",
+            name="manual_drop_enabled",
+            field=models.BooleanField(
+                default=True, help_text="Whether players are allowed to use the drop command in this server"
+            ),
+        )
+    ]

admin_panel/bd_models/models.py

@@ -82,6 +82,9 @@ class GuildConfig(models.Model):
     silent = models.BooleanField(
         help_text="Whether the responses of guesses get sent as ephemeral or not", default=False
     )
+    manual_drop_enabled = models.BooleanField(
+        help_text="Whether players are allowed to use the drop command in this server", default=True
+    )
     admin_command_synced = models.BooleanField(
         help_text="True if slash admin commands are present in this server", default=False, editable=False
     )

ballsdex/packages/balls/cog.py

@@ -25,7 +25,7 @@
 )
 from ballsdex.core.utils.utils import can_mention, inventory_privacy, is_staff
 from bd_models.enums import DonationPolicy
-from bd_models.models import BallInstance, Player, Special, Trade, TradeObject, balls, groups
+from bd_models.models import BallInstance, GuildConfig, Player, Special, Trade, TradeObject, balls, groups
 from settings.models import settings
 
 from .bulk_give_selector import BulkGiveSelector
@@ -535,6 +535,13 @@ async def drop(
         if not countryball:
             return
 
+        config = await GuildConfig.objects.aget_or_none(guild_id=interaction.guild_id)
+        if config and not config.manual_drop_enabled:
+            await interaction.response.send_message(
+                "The drop command is currently disabled in this server.", ephemeral=True
+            )
+            return
+
         cog = cast("CountryBallsSpawner | None", self.bot.get_cog("CountryBallsSpawner"))
         if not cog or not interaction.guild_id or cog.cache.get(interaction.guild_id) != interaction.channel_id:
             await interaction.response.send_message(

ballsdex/packages/guildconfig/cog.py

@@ -151,6 +151,25 @@ async def disable(self, interaction: discord.Interaction["BallsDexBot"]):
                     f"spawning channel set. Please configure one with {config_cmd}."
                 )
 
+    @app_commands.command()
+    @app_commands.checks.has_permissions(manage_guild=True)
+    @app_commands.checks.bot_has_permissions(send_messages=True)
+    async def toggledrop(self, interaction: discord.Interaction["BallsDexBot"]):
+        """
+        Allow or disallow players from using the drop command in this server.
+        """
+        config, created = await GuildConfig.objects.aget_or_create(guild_id=interaction.guild_id)
+        config.manual_drop_enabled = not config.manual_drop_enabled  # type: ignore
+        await config.asave()
+        if config.manual_drop_enabled:
+            await interaction.response.send_message(
+                f"Players can now use the drop command to manually spawn {settings.plural_collectible_name}."
+            )
+        else:
+            await interaction.response.send_message(
+                f"Players can no longer use the drop command to manually spawn {settings.plural_collectible_name}."
+            )
+
     @app_commands.command()
     @app_commands.checks.has_permissions(manage_guild=True)
     async def status(self, interaction: discord.Interaction["BallsDexBot"]):
@@ -181,6 +200,7 @@ async def status(self, interaction: discord.Interaction["BallsDexBot"]):
 
         embed.add_field(name="Channel", value=channel.mention, inline=True)
         embed.add_field(name="Status", value="Enabled" if config.enabled else "Disabled", inline=True)
+        embed.add_field(name="Drop command", value="Enabled" if config.manual_drop_enabled else "Disabled", inline=True)
 
         def tick(granted: bool) -> str:
             return "\N{WHITE HEAVY CHECK MARK}" if granted else "\N{CROSS MARK}"