Compare commits
3 commits
491723fc3b
...
47789f920a
Author | SHA1 | Date | |
---|---|---|---|
47789f920a | |||
385999a2f3 | |||
c913e63f51 |
5 changed files with 122 additions and 18 deletions
|
@ -1,3 +1,3 @@
|
|||
from . import post, verein
|
||||
from . import advent, post, verein
|
||||
|
||||
COMMANDS = post.COMMANDS + verein.COMMANDS
|
||||
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,
|
||||
]
|
|
@ -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,
|
||||
)
|
||||
|
|
|
@ -8,19 +8,6 @@ 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}")
|
||||
|
||||
# öffentliche Antwort in DM channels und in "allowed" channels
|
||||
in_dm_channel = interaction.channel is not None and isinstance(
|
||||
interaction.channel, discord.DMChannel
|
||||
)
|
||||
in_allowed_channel = interaction.channel_id in CONFIG.ev_info.channels
|
||||
|
||||
# private Antwort sonst
|
||||
return not (in_dm_channel or in_allowed_channel)
|
||||
|
||||
|
||||
@functools.singledispatch
|
||||
def make_command(command) -> discord.app_commands.Command:
|
||||
raise NotImplementedError
|
||||
|
@ -33,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,
|
||||
)
|
||||
|
||||
|
@ -57,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
|
||||
|
|
|
@ -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