This commit is contained in:
Paul Glaß 2024-12-02 21:09:10 +01:00
commit a63d8d058c
5 changed files with 119 additions and 0 deletions

4
.gitignore vendored Normal file
View File

@ -0,0 +1,4 @@
.env
.venv
**/fregbot-birthday.log
.idea/**

14
pyproject.toml Normal file
View File

@ -0,0 +1,14 @@
[build-system]
requires = ["setuptools>=61.0", "setuptools-scm"]
build-backend = "setuptools.build_meta"
[project]
name = "fregbot"
dynamic = ["version"]
dependencies = [
"discord.py",
"pydantic",
]
[project.scripts]
fregbot-birthday = "birthday:main"

2
src/.gitignore vendored Normal file
View File

@ -0,0 +1,2 @@
*.egg-info
__pycache__

83
src/birthday.py Normal file
View File

@ -0,0 +1,83 @@
import datetime
import json
import logging
import os
from pathlib import Path
import discord
from discord import Client, Member, TextChannel
from pydantic import BaseModel
class ChannelConfig(BaseModel):
server_id: int
channel_id: int
class BirthdayHaver(BaseModel):
name: str = None
user_id: int
date: str
class BirthdayConfig(BaseModel):
channel: ChannelConfig
birthdays: list[BirthdayHaver]
class BirthdayClient(Client):
birthdays: dict[tuple[Member, str], datetime.date] = {}
config_file: Path = None
channel: TextChannel = None
async def on_ready(self):
logging.info(f'Logged in as {self.user}')
try:
self.channel = self.parse_birthdays()
logging.info(f'Birthdays loaded')
for (member, name), birthday in self.birthdays.items():
if datetime.date.today() == birthday:
await self.tell_birthday(member, name)
except Exception as e:
logging.error(f'Error: {e}')
finally:
await self.close()
async def tell_birthday(self, member: Member, name: str) -> None:
logging.info(f'Gratuliere {member.display_name}')
await self.channel.send(f'Heute hat {name} ({member.mention}) Geburtstag! Seid nett :birthday: :tada:')
def parse_birthdays(self) -> TextChannel:
with open(self.config_file) as fh:
config = BirthdayConfig.model_validate_json(fh.read())
guild = self.get_guild(config.channel.server_id)
channel = discord.utils.get(guild.channels, id=config.channel.channel_id)
for birthday_haver in config.birthdays:
member = discord.utils.get(guild.members, id=birthday_haver.user_id)
birthday = datetime.datetime.strptime(birthday_haver.date, '%d.%m.')
self.birthdays[member, birthday_haver.name] = datetime.date(datetime.date.today().year, birthday.month,
birthday.day)
return channel
def main() -> None:
intents = discord.Intents.default()
intents.message_content = intents.members = intents.messages = True
log_handler = logging.FileHandler(filename='fregbot-birthday.log', encoding='utf-8', mode='w')
client = BirthdayClient(intents=intents)
client.config_file = Path(os.getenv('FREG_BIRTHDAY_CONFIG') or '')
if not client.config_file.is_file():
logging.error(f'{client.config_file} ist leider keine Datei :s, setz mal FREG_BIRTHDAY_CONFIG')
exit(2)
token = os.getenv('FREG_TOKEN')
if not token:
logging.error('Kein Token gefunden. Kann nicht starten. Umgebungsvariable FREG_TOKEN muss gesetzt sein')
exit(1)
client.run(token, log_handler=log_handler)
if __name__ == '__main__':
main()

16
src/main.py Normal file
View File

@ -0,0 +1,16 @@
# This is a sample Python script.
# Press Umschalt+F10 to execute it or replace it with your code.
# Press Double Shift to search everywhere for classes, files, tool windows, actions, and settings.
def print_hi(name):
# Use a breakpoint in the code line below to debug your script.
print(f'Hi, {name}') # Press Strg+8 to toggle the breakpoint.
# Press the green button in the gutter to run the script.
if __name__ == '__main__':
print_hi('PyCharm')
# See PyCharm help at https://www.jetbrains.com/help/pycharm/