mirror of
https://code.lenaisten.de/Lenaisten/lenaverse-bot.git
synced 2024-11-21 22:43:01 +00:00
Merge branch 'develop' into release/0.3.0
This commit is contained in:
commit
76cba673df
6 changed files with 127 additions and 13 deletions
3
lenaverse_bot/commands/__init__.py
Normal file
3
lenaverse_bot/commands/__init__.py
Normal file
|
@ -0,0 +1,3 @@
|
|||
from . import advent, post, verein
|
||||
|
||||
COMMANDS = post.COMMANDS + verein.COMMANDS + advent.COMMANDS
|
97
lenaverse_bot/commands/advent.py
Normal file
97
lenaverse_bot/commands/advent.py
Normal file
|
@ -0,0 +1,97 @@
|
|||
import logging
|
||||
from io import BytesIO
|
||||
|
||||
import aiohttp
|
||||
import discord
|
||||
|
||||
from ..core.config import CONFIG
|
||||
|
||||
_logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
@discord.app_commands.command(name=CONFIG.command_prefix + "advent")
|
||||
async def advent(interaction: discord.Interaction, day: int | None = None) -> None:
|
||||
"""
|
||||
Türchen vom Lenaisten-Adventskalender öffnen (https://advent.lenaisten.de)
|
||||
"""
|
||||
|
||||
_logger.debug(
|
||||
f"User {interaction.user.name}({interaction.user.id}) used /advent (day: {day})"
|
||||
)
|
||||
|
||||
async with aiohttp.ClientSession(
|
||||
auth=aiohttp.BasicAuth(login="", password="")
|
||||
) as session:
|
||||
if day is None:
|
||||
# Kein Tag angegeben => neuesten Tag finden
|
||||
|
||||
async with session.get(
|
||||
"https://advent.lenaisten.de/api/user/doors"
|
||||
) as http_response:
|
||||
if http_response.status != 200:
|
||||
# user/doors Anfrage hat nicht geklappt
|
||||
await interaction.response.send_message(
|
||||
content="Fehler: Ich konnte das aktuelle Türchen nicht finden :zany_face: (Probier's nochmal?)",
|
||||
# nur für ausführenden User
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
|
||||
if not isinstance(doors := await http_response.json(), list):
|
||||
# user/doors Antwort falsches Format (keine Liste)
|
||||
await interaction.response.send_message(
|
||||
content="Fehler: Ich konnte das aktuelle Türchen nicht finden :sweat_smile: (Probier's nochmal?)",
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
|
||||
days: list[int] = []
|
||||
for door in doors:
|
||||
if not isinstance(door, dict) or "day" not in door:
|
||||
# user/doors Antwort falsches Format (Liste enthält falsche Daten)
|
||||
await interaction.response.send_message(
|
||||
content="Fehler: Ich konnte das aktuelle Türchen nicht finden :face_with_monocle: (Probier's nochmal?)",
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
|
||||
days.append(door["day"])
|
||||
|
||||
day = max(days)
|
||||
|
||||
async with session.get(
|
||||
f"https://advent.lenaisten.de/api/user/image_{day}"
|
||||
) as http_response:
|
||||
reply_ephemeral = not CONFIG.ev_info.in_allowed_channel(interaction)
|
||||
|
||||
if http_response.status == 401:
|
||||
# Bild (noch) nicht verfügbar
|
||||
await interaction.response.send_message(
|
||||
content=f"Fehler: Tag {day} kann ich (noch?) nicht abrufen. Netter Versuch! :woman_technologist_tone2:",
|
||||
ephemeral=reply_ephemeral,
|
||||
)
|
||||
return
|
||||
|
||||
if http_response.status != 200:
|
||||
# Bild (noch) nicht verfügbar
|
||||
await interaction.response.send_message(
|
||||
content="Fehler: Ich konnte das Bild nicht herunterladen :sob: (Probier's nochmal?)",
|
||||
ephemeral=True,
|
||||
)
|
||||
return
|
||||
|
||||
image = discord.File(
|
||||
fp=BytesIO(await http_response.read()),
|
||||
filename="advent.jpg",
|
||||
)
|
||||
|
||||
await interaction.response.send_message(
|
||||
content=f"Hier ist das Bild für Tag {day}!",
|
||||
file=image,
|
||||
ephemeral=reply_ephemeral,
|
||||
)
|
||||
|
||||
|
||||
COMMANDS = [
|
||||
advent,
|
||||
]
|
|
@ -4,7 +4,7 @@ from enum import Enum, auto
|
|||
import discord
|
||||
from discord import ui
|
||||
|
||||
from .config import CONFIG
|
||||
from ..core.config import CONFIG
|
||||
|
||||
_logger = logging.getLogger(__name__)
|
||||
|
||||
|
@ -111,7 +111,7 @@ async def post(interaction: discord.Interaction) -> None:
|
|||
f"User {interaction.user.name}({interaction.user.id}) tried to /post"
|
||||
)
|
||||
await interaction.response.send_message(
|
||||
content="Du bist nicht berechtigt, den `/post`-Befehl zu benutzen!",
|
||||
content=f"Du bist nicht berechtigt, den `/{CONFIG.command_prefix}post`-Befehl zu benutzen!",
|
||||
# nur für ausführenden User
|
||||
ephemeral=True,
|
||||
)
|
|
@ -3,16 +3,11 @@ import logging
|
|||
|
||||
import discord
|
||||
|
||||
from .config import CONFIG, FileCommand, InfoCommand
|
||||
from ..core.config import CONFIG, FileCommand, InfoCommand
|
||||
|
||||
_logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
def reply_private(interaction: discord.Interaction, name: str) -> bool:
|
||||
_logger.debug(f"User {interaction.user.name}({interaction.user.id}) used /{name}")
|
||||
return interaction.channel_id not in CONFIG.ev_info.channels
|
||||
|
||||
|
||||
@functools.singledispatch
|
||||
def make_command(command) -> discord.app_commands.Command:
|
||||
raise NotImplementedError
|
||||
|
@ -25,11 +20,15 @@ def _(command: FileCommand) -> discord.app_commands.Command:
|
|||
description=command.description,
|
||||
)
|
||||
async def cmd(interaction: discord.Interaction) -> None:
|
||||
_logger.debug(
|
||||
f"User {interaction.user.name}({interaction.user.id}) used FileCommand /{command.name}"
|
||||
)
|
||||
|
||||
if (file := await command.as_discord_file) is not None:
|
||||
await interaction.response.send_message(
|
||||
content=command.content,
|
||||
suppress_embeds=True,
|
||||
ephemeral=reply_private(interaction, command.name),
|
||||
ephemeral=not CONFIG.ev_info.in_allowed_channel(interaction),
|
||||
file=file,
|
||||
)
|
||||
|
||||
|
@ -49,10 +48,14 @@ def _(command: InfoCommand) -> discord.app_commands.Command:
|
|||
description=command.description,
|
||||
)
|
||||
async def cmd(interaction: discord.Interaction) -> None:
|
||||
_logger.debug(
|
||||
f"User {interaction.user.name}({interaction.user.id}) used InfoCommand /{command.name}"
|
||||
)
|
||||
|
||||
await interaction.response.send_message(
|
||||
content=command.content,
|
||||
suppress_embeds=True,
|
||||
ephemeral=reply_private(interaction, command.name),
|
||||
ephemeral=not CONFIG.ev_info.in_allowed_channel(interaction),
|
||||
)
|
||||
|
||||
return cmd
|
|
@ -2,7 +2,7 @@ import logging
|
|||
|
||||
import discord
|
||||
|
||||
from . import post, verein
|
||||
from ..commands import COMMANDS
|
||||
from .config import CONFIG
|
||||
|
||||
_logger = logging.getLogger(__name__)
|
||||
|
@ -16,8 +16,7 @@ class LenaverseBot(discord.Client):
|
|||
)
|
||||
|
||||
self.tree = discord.app_commands.CommandTree(self)
|
||||
commands = post.COMMANDS + verein.COMMANDS
|
||||
for command in commands:
|
||||
for command in COMMANDS:
|
||||
self.tree.add_command(command)
|
||||
|
||||
async def setup_hook(self):
|
||||
|
|
|
@ -68,6 +68,18 @@ class ClubInfo(BaseModel):
|
|||
fest: InfoCommand
|
||||
aktion: InfoCommand
|
||||
|
||||
def in_allowed_channel(self, interaction: discord.Interaction) -> bool:
|
||||
if interaction.channel is None:
|
||||
return False
|
||||
|
||||
# öffentliche Antwort erlaubt in:
|
||||
# - DM channels
|
||||
# - "allowed" channels
|
||||
is_dm_channel = isinstance(interaction.channel, discord.DMChannel)
|
||||
is_listed = interaction.channel.id in self.channels
|
||||
|
||||
return is_dm_channel or is_listed
|
||||
|
||||
|
||||
class Config(BaseModel):
|
||||
discord_token: str
|
||||
|
|
Loading…
Reference in a new issue