easterobot 1.1.1__tar.gz → 1.3.1__tar.gz

{easterobot-1.1.1 → easterobot-1.3.1}/.dockerignore

@@ -3,7 +3,10 @@
 !pyproject.toml
 !LICENSE
 !/easterobot/*.py
-!/easterobot/commands/*.py
+!/easterobot/*/*.py
+!/easterobot/*/*/*.py
+!/easterobot/alembic/script.py.mako
+!/easterobot/py.typed
 !/easterobot/resources
 !README.rst
 !entrypoint.sh

{easterobot-1.1.1 → easterobot-1.3.1}/.github/workflows/tests.yml

@@ -37,3 +37,16 @@ jobs:
 
       - name: Run tests
         run: uv run poe test
+  docker:
+    name: Docker build
+    runs-on: ubuntu-latest
+
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Set up Docker
+        uses: docker/setup-buildx-action@v3
+
+      - name: Build services with Docker Compose
+        run: docker compose -f docker-compose.yml build

{easterobot-1.1.1 → easterobot-1.3.1}/Dockerfile

@@ -2,8 +2,8 @@
 FROM python:3.12-slim
 
 # Metadata for clarity and documentation
-LABEL maintainer="your_email@example.com"
-LABEL description="Docker image for easterobot using uv for dependency management"
+LABEL maintainer="dashstrom.pro@gmail.com"
+LABEL description="Docker image for easterobot, a discord bot for easter events"
 
 # Add the UV binary
 COPY --from=ghcr.io/astral-sh/uv:latest /uv /bin/uv
@@ -17,9 +17,6 @@ RUN mkdir /data
 # Make it read-only
 RUN chown easterobot:easterobot /data
 
-# Use non-root user for security
-USER easterobot
-
 # Change the working directory to the `src` directory
 WORKDIR /src
 
@@ -32,8 +29,16 @@ RUN uv sync --frozen --no-install-project
 # Copy the project into the image
 COPY . .
 
+# Fix permissions
+RUN find /src -type d -exec chmod 755 {} \;
+RUN find /src -type f -exec chmod 644 {} \;
+RUN chmod +x /src/entrypoint.sh
+
 # Sync the project
 RUN uv sync --frozen
 
+# Use non-root user for security
+USER easterobot
+
 # Default command (use exec form for signal handling)
 CMD ["/src/entrypoint.sh"]

{easterobot-1.1.1 → easterobot-1.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: easterobot
-Version: 1.1.1
+Version: 1.3.1
 Summary: Discord bot for Easter.
 Project-URL: Homepage, https://github.com/Dashstrom/easterobot
 Project-URL: Repository, https://github.com/Dashstrom/easterobot
@@ -121,6 +121,9 @@ from `PyPI <https://pypi.org/project>`_
     cd easterobot
     echo "DISCORD_TOKEN=YOU_MUST_PUT_YOUR_TOKEN_HERE" > .env
 
+    # Can be unsafe (and for each update)
+    chmod -R 700 . && mkdir data -p && chmod 777 data
+
     # Run the docker container
     docker compose up -d
 
@@ -130,6 +133,12 @@ from `PyPI <https://pypi.org/project>`_
     # Remove the container (not the data)
     docker compose down --rmi all
 
+    # Update
+    git reset --hard HEAD && git pull
+
+    # One-line update
+    docker compose down --rmi all && git reset --hard HEAD && git pull && chmod -R 700 . && mkdir data -p && chmod 777 data && docker compose up -d
+
 Configuration directory
 #######################
 

{easterobot-1.1.1 → easterobot-1.3.1}/README.rst

@@ -83,6 +83,9 @@ from `PyPI <https://pypi.org/project>`_
     cd easterobot
     echo "DISCORD_TOKEN=YOU_MUST_PUT_YOUR_TOKEN_HERE" > .env
 
+    # Can be unsafe (and for each update)
+    chmod -R 700 . && mkdir data -p && chmod 777 data
+
     # Run the docker container
     docker compose up -d
 
@@ -92,6 +95,12 @@ from `PyPI <https://pypi.org/project>`_
     # Remove the container (not the data)
     docker compose down --rmi all
 
+    # Update
+    git reset --hard HEAD && git pull
+
+    # One-line update
+    docker compose down --rmi all && git reset --hard HEAD && git pull && chmod -R 700 . && mkdir data -p && chmod 777 data && docker compose up -d
+
 Configuration directory
 #######################
 

easterobot-1.3.1/conftest.py

@@ -0,0 +1,44 @@
+"""Configuration for all tests."""
+
+from collections.abc import AsyncIterator
+from typing import Any
+
+import py
+import pytest
+import pytest_asyncio
+from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession
+
+from easterobot import __author__
+from easterobot.bot import Easterobot
+from easterobot.config import MConfig
+
+
+@pytest.fixture(autouse=True)
+def _add_author(doctest_namespace: dict[str, Any]) -> None:
+    """Update doctest namespace."""
+    doctest_namespace["author"] = __author__
+
+
+@pytest.fixture
+def bot(tmpdir: py.path.LocalPath) -> Easterobot:
+    """Get a bot ready-to-use."""
+    return Easterobot.generate(str(tmpdir), env=True)
+
+
+@pytest.fixture
+def engine(bot: Easterobot) -> AsyncEngine:
+    """Get a bot ready-to-use."""
+    return bot.engine
+
+
+@pytest_asyncio.fixture
+async def session(engine: AsyncEngine) -> AsyncIterator[AsyncSession]:
+    """Get a bot ready-to-use."""
+    async with AsyncSession(engine, expire_on_commit=False) as session:
+        yield session
+
+
+@pytest.fixture
+def config(bot: Easterobot) -> MConfig:
+    """Get a bot ready-to-use."""
+    return bot.config

{easterobot-1.1.1 → easterobot-1.3.1}/docker-compose.yml

@@ -1,6 +1,6 @@
 services:
   bot:
-    container_name: easterobot_bot
+    container_name: bot
     stdin_open: true
     tty: true
     build:

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/bot.py

@@ -1,7 +1,6 @@
 """Main program."""
 
 import logging
-import logging.config
 import pathlib
 import shutil
 from getpass import getpass
@@ -37,7 +36,8 @@ from .config import (
 T = TypeVar("T")
 
 logger = logging.getLogger(__name__)
-INTENTS = discord.Intents.all()
+INTENTS = discord.Intents.default()
+INTENTS.message_content = True
 
 
 class Easterobot(discord.ext.commands.Bot):
@@ -47,15 +47,24 @@ class Easterobot(discord.ext.commands.Bot):
 
     def __init__(self, config: MConfig) -> None:
         """Initialise Easterbot."""
+        # Initialize intents
+        intents = discord.Intents.default()
+        if config.message_content:
+            intents.message_content = True
+
+        # Remove warning about VoiceClient
+        discord.VoiceClient.warn_nacl = False
+
+        # Initialize Bot
         super().__init__(
             command_prefix=".",
             description="Bot discord pour faire la chasse aux œufs",
             activity=discord.Game(name="rechercher des œufs"),
             intents=INTENTS,
         )
-        self.config = config
 
         # Attributes
+        self.config = config
         self.app_commands: list[discord.app_commands.AppCommand] = []
         self.app_emojis: dict[str, discord.Emoji] = {}
 
@@ -89,9 +98,9 @@ class Easterobot(discord.ext.commands.Bot):
         cls,
         destination: Union[Path, str],
         *,
-        token: Optional[str],
-        env: bool,
-        interactive: bool,
+        token: Optional[str] = None,
+        env: bool = False,
+        interactive: bool = False,
     ) -> "Easterobot":
         """Generate all data."""
         destination = Path(destination).resolve()

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/__init__.py

@@ -12,6 +12,7 @@ from easterobot.commands.game import (
     tictactoe_command,
 )
 from easterobot.commands.help import help_command
+from easterobot.commands.info import info_command
 from easterobot.commands.reset import reset_command
 from easterobot.commands.search import search_command
 from easterobot.commands.top import top_command
@@ -24,6 +25,7 @@ __all__ = [
     "egg_command_group",
     "enable_command",
     "help_command",
+    "info_command",
     "reset_command",
     "rockpaperscissor_command",
     "search_command",
@@ -33,4 +35,5 @@ __all__ = [
 
 
 async def setup(bot: Easterobot) -> None:
+    egg_command_group.name = bot.config.group
     bot.tree.add_command(egg_command_group)

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/base.py

@@ -170,7 +170,10 @@ def controlled_command(  # noqa: C901, PLR0915
             try:
                 await f(cast(Context, interaction), *args, **kwargs)
             except InterruptedCommandError:
-                logger.exception("InterruptedCommandError occur")
+                logger.warning(
+                    "InterruptedCommandError occur for %s",
+                    event_repr,
+                )
                 async with (
                     AsyncSession(interaction.client.engine) as session,
                     cooldown_lock,

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/disable.py

@@ -9,7 +9,8 @@ from .base import Context, controlled_command, egg_command_group
 
 
 @egg_command_group.command(
-    name="disable", description="Désactiver la chasse aux œufs dans le salon"
+    name="disable",
+    description="Désactiver la chasse aux œufs dans le salon",
 )
 @controlled_command(cooldown=True, manage_channels=True)
 async def disable_command(ctx: Context) -> None:

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/edit.py

@@ -13,7 +13,8 @@ from .base import Context, controlled_command, egg_command_group
 
 
 @egg_command_group.command(
-    name="edit", description="Editer le nombre d'œufs d'un membre"
+    name="edit",
+    description="Editer le nombre d'œufs d'un membre",
 )
 @controlled_command(cooldown=True, administrator=True)
 async def edit_command(

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/enable.py

@@ -9,7 +9,8 @@ from .base import Context, controlled_command, egg_command_group
 
 
 @egg_command_group.command(
-    name="enable", description="Activer la chasse dans le salon"
+    name="enable",
+    description="Activer la chasse dans le salon",
 )
 @controlled_command(
     cooldown=True,

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/game.py

@@ -130,25 +130,28 @@ async def game_dual(  # noqa: C901, D103, PLR0912
                 await session.commit()
 
             # Play the game
-            game = cls(ctx.user, member, msg)
-            await ctx.client.game.run(game)
-            winner = await game.wait_winner()
-
-            # Give eggs to the winner or remove previous one
-            async with lock:
-                for e in e1:
-                    e.lock = False
-                    if winner:
-                        e.user_id = winner.id
-                for e in e2:
-                    e.lock = False
-                    if winner:
-                        e.user_id = winner.id
-                await session.commit()
+            winner = None
+            try:
+                game = cls(ctx.user, member, msg)
+                await ctx.client.game.run(game)
+                winner = await game.wait_winner()
+            finally:
+                # Give eggs to the winner or remove previous one
+                async with lock:
+                    for e in e1:
+                        e.lock = False
+                        if winner:
+                            e.user_id = winner.id
+                    for e in e2:
+                        e.lock = False
+                        if winner:
+                            e.user_id = winner.id
+                    await session.commit()
 
 
 @egg_command_group.command(
-    name="connect4", description="Lancer une partie de puissance 4."
+    name="connect4",
+    description="Lancer une partie de puissance 4",
 )
 @controlled_command(cooldown=True, channel_permissions={"send_messages": True})
 async def connect4_command(
@@ -161,12 +164,13 @@ async def connect4_command(
 
 
 @egg_command_group.command(
-    name="tictactoe", description="Lancer une partie de morpion."
+    name="tictactoe",
+    description="Lancer une partie de morpion",
 )
-@controlled_command(cooldown=True)
+@controlled_command(cooldown=True, channel_permissions={"send_messages": True})
 async def tictactoe_command(
     ctx: Context,
-    member: discord.Member,
+    member: Optional[discord.Member] = None,
     bet: app_commands.Range[int, 0] = 0,
 ) -> None:
     """Run a tictactoe."""
@@ -175,12 +179,12 @@ async def tictactoe_command(
 
 @egg_command_group.command(
     name="rockpaperscissor",
-    description="Lancer une partie de pierre papier ciseaux.",
+    description="Lancer une partie de pierre papier ciseaux",
 )
-@controlled_command(cooldown=True)
+@controlled_command(cooldown=True, channel_permissions={"send_messages": True})
 async def rockpaperscissor_command(
     ctx: Context,
-    member: discord.Member,
+    member: Optional[discord.Member] = None,
     bet: app_commands.Range[int, 0] = 0,
 ) -> None:
     """Run a rockpaperscissor."""

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/help.py

@@ -7,8 +7,11 @@ from easterobot.hunts.hunt import embed
 from .base import Context, controlled_command, egg_command_group
 
 
-@egg_command_group.command(name="help", description="Obtenir de l'aide")
-@controlled_command(cooldown=True)
+@egg_command_group.command(
+    name="help",
+    description="Obtenir l'aide des commandes",
+)
+@controlled_command(cooldown=True, channel_permissions={"send_messages": True})
 async def help_command(ctx: Context) -> None:
     """Help command."""
     emb = embed(
@@ -30,4 +33,4 @@ async def help_command(ctx: Context) -> None:
                 value=f"{option.description}",
                 inline=False,
             )
-    await ctx.response.send_message(embed=emb, ephemeral=True)
+    await ctx.response.send_message(embed=emb)

easterobot-1.3.1/easterobot/commands/info.py

@@ -0,0 +1,61 @@
+"""Command basket."""
+
+import asyncio
+from typing import Optional
+
+import discord
+from discord import app_commands
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from easterobot.commands.base import (
+    Context,
+    controlled_command,
+    egg_command_group,
+)
+from easterobot.hunts.hunt import embed
+from easterobot.hunts.rank import Ranking
+
+
+@egg_command_group.command(
+    name="info",
+    description="Avoir des informations sur la chance d'un joueur",
+)
+@app_commands.describe(
+    user="Joueur a inspecter",
+)
+@controlled_command(cooldown=True)
+async def info_command(
+    ctx: Context, user: Optional[discord.Member] = None
+) -> None:
+    """Show current user basket."""
+    # Delay the response
+    await ctx.response.defer(ephemeral=True)
+
+    # Set the user of the basket
+    hunter = user or ctx.user
+
+    async with AsyncSession(ctx.client.engine) as session:
+        ranking, member_luck = await asyncio.gather(
+            Ranking.from_guild(session, ctx.guild_id),
+            ctx.client.hunt.get_luck(
+                session=session,
+                guild_id=hunter.guild.id,
+                user_id=hunter.id,
+                sleep_hours=False,
+            ),
+        )
+        hunter_rank = ranking.get(hunter.id)
+
+        await ctx.followup.send(
+            embed=embed(
+                title=f"Informations sur {hunter.display_name}",
+                description=(
+                    f"Classement : {hunter_rank.badge}\n"
+                    f"Nombre d'oeufs : `{hunter_rank.eggs}`\n"
+                    f"Chance brute : `{member_luck.luck:.0%}`\n"
+                    f"Chance de trouver un oeuf : `{member_luck.discovered:.0%}`\n"
+                    f"Chance de se faire voler : `{member_luck.spotted:.0%}`"
+                ),
+            ),
+            ephemeral=True,
+        )

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/reset.py

@@ -14,7 +14,8 @@ from .base import Context, Interaction, controlled_command, egg_command_group
 
 
 @egg_command_group.command(
-    name="reset", description="Réinitialiser la chasse aux œufs"
+    name="reset",
+    description="Réinitialiser la chasse aux œufs",
 )
 @controlled_command(cooldown=True, administrator=True)
 async def reset_command(ctx: Context) -> None:

{easterobot-1.1.1 → easterobot-1.3.1}/easterobot/commands/search.py

@@ -4,10 +4,10 @@ import logging
 from typing import Any
 
 import discord
-from sqlalchemy import and_, func, select
+from sqlalchemy import select
 from sqlalchemy.ext.asyncio import AsyncSession
 
-from easterobot.config import RAND, agree
+from easterobot.config import agree
 from easterobot.hunts.hunt import embed
 from easterobot.models import Egg, Hunt
 
@@ -21,7 +21,10 @@ from .base import (
 logger = logging.getLogger("easterobot")
 
 
-@egg_command_group.command(name="search", description="Rechercher un œuf")
+@egg_command_group.command(
+    name="search",
+    description="Rechercher un œuf",
+)
 @controlled_command(cooldown=True, channel_permissions={"send_messages": True})
 async def search_command(ctx: Context) -> None:
     """Search command."""
@@ -34,7 +37,7 @@ async def search_command(ctx: Context) -> None:
                 "La chasse aux œufs n'est pas activée dans ce salon",
                 ephemeral=True,
             )
-            return
+            raise InterruptedCommandError
     try:
         await ctx.response.defer(ephemeral=False)
     except discord.errors.NotFound as err:
@@ -42,39 +45,15 @@ async def search_command(ctx: Context) -> None:
     name = ctx.user.display_name
 
     async with AsyncSession(ctx.client.engine) as session:
-        egg_max = await session.scalar(
-            select(
-                func.count().label("max"),
-            )
-            .where(Egg.guild_id == ctx.guild_id)
-            .group_by(Egg.user_id)
-            .order_by(func.count().label("max").desc())
-            .limit(1)
-        )
-        egg_max = egg_max or 0
-        egg_count = await session.scalar(
-            select(func.count().label("count")).where(
-                and_(
-                    Egg.guild_id == ctx.guild.id,
-                    Egg.user_id == ctx.user.id,
-                )
-            )
+        luck = await ctx.client.hunt.get_luck(
+            guild_id=ctx.guild_id,
+            user_id=ctx.user.id,
+            session=session,
+            sleep_hours=ctx.client.config.in_sleep_hours(),
         )
-        if egg_count is None:
-            egg_count = 0
-    ratio = egg_count / egg_max if egg_max != 0 else 1.0
-
-    discovered = ctx.client.config.commands.search.discovered
-    prob_d = (discovered.max - discovered.min) * (1 - ratio) + discovered.min
 
-    sample_d = RAND.random()
-    if prob_d > sample_d or egg_count < discovered.shield:
-        sample_s = RAND.random()
-        spotted = ctx.client.config.commands.search.spotted
-        prob_s = (spotted.max - spotted.min) * ratio + spotted.min
-        logger.info("discovered: %.2f > %.2f", prob_d, sample_d)
-        if prob_s > sample_s and egg_count > spotted.shield:
-            logger.info("spotted: %.2f > %.2f", prob_s, sample_s)
+    if luck.sample_discovered():
+        if luck.sample_spotted():
 
             async def send_method(
                 *args: Any, **kwargs: Any
@@ -88,7 +67,6 @@ async def search_command(ctx: Context) -> None:
                 send_method=send_method,
             )
         else:
-            logger.info("found: %.2f > %.2f", prob_s, sample_s)
             emoji = ctx.client.egg_emotes.rand()
             async with AsyncSession(ctx.client.engine) as session:
                 session.add(
@@ -105,7 +83,7 @@ async def search_command(ctx: Context) -> None:
                 "%s (%s) got an egg for a total %s in %s",
                 ctx.user,
                 ctx.user.id,
-                agree("{0} egg", "{0} eggs", egg_count),
+                agree("{0} egg", "{0} eggs", luck.egg_count),
                 ctx.channel.jump_url,
             )
             await ctx.followup.send(
@@ -113,15 +91,14 @@ async def search_command(ctx: Context) -> None:
                     title=f"{name} récupère un œuf",
                     description=ctx.client.config.hidden(ctx.user),
                     thumbnail=emoji.url,
-                    egg_count=egg_count + 1,
+                    egg_count=luck.egg_count + 1,
                 )
             )
     else:
-        logger.info("failed: %.2f > %.2f", prob_d, sample_d)
         await ctx.followup.send(
             embed=embed(
                 title=f"{name} repart bredouille",
                 description=ctx.client.config.failed(ctx.user),
-                egg_count=egg_count,
+                egg_count=luck.egg_count,
             )
         )