51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
|
import os
|
||
|
|
import discord
|
||
|
|
import database
|
||
|
|
|
||
|
|
|
||
|
|
def read_env(file):
|
||
|
|
for line in file:
|
||
|
|
splitline = line.split('=', maxsplit=1)
|
||
|
|
if len(splitline) == 2:
|
||
|
|
os.environ[splitline[0]] = splitline[1]
|
||
|
|
|
||
|
|
|
||
|
|
class AnnouncementWatcherClient(discord.Client):
|
||
|
|
def __init__(self, *args, **kwargs):
|
||
|
|
super().__init__(*args, **kwargs)
|
||
|
|
self.db = database.DatabaseManager('announcement.db')
|
||
|
|
|
||
|
|
async def on_guild_join(self, guild):
|
||
|
|
self.db.create_guild()
|
||
|
|
|
||
|
|
# TODO: on_guild_leave() delete all data?
|
||
|
|
|
||
|
|
async def on_message(self, message: discord.Message):
|
||
|
|
if message.author == self.user:
|
||
|
|
return
|
||
|
|
|
||
|
|
if message.content.startswith('$hello'):
|
||
|
|
await message.channel.send('Hello!')
|
||
|
|
|
||
|
|
async def on_guild_channel_pins_update(self, channel: discord.GuildChannel, last_pin):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def register_channel(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def unregister_channel(self):
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
with open('.env') as f:
|
||
|
|
read_env(f)
|
||
|
|
discord_token = os.environ.get('DISCORD_TOKEN')
|
||
|
|
if discord_token is None:
|
||
|
|
print('Could not read token')
|
||
|
|
exit(1)
|
||
|
|
intents = discord.Intents.default()
|
||
|
|
intents.message_content = True
|
||
|
|
client = AnnouncementWatcherClient(intents=intents)
|
||
|
|
client.run(discord_token)
|