radiobot/main.py

94 lines
2.8 KiB
Python
Raw Normal View History

2022-03-23 19:45:22 -04:00
import os
2022-03-23 19:27:00 -04:00
import discord
2022-03-23 21:13:39 -04:00
from discord.ext import commands
import asyncio
from subprocess import Popen
2022-03-23 19:27:00 -04:00
2022-03-23 19:45:22 -04:00
bot_version = os.environ['version']
print('radiobot ' + bot_version + ' starting')
2022-03-23 19:27:00 -04:00
2022-03-23 21:13:39 -04:00
login_token = os.environ['token']
voice_channel_id = os.environ['channel_voice']
2022-03-23 21:22:35 -04:00
text_channel_id = os.environ['channel_text']
2022-03-23 19:56:13 -04:00
2022-03-23 21:13:39 -04:00
# Configure the bot
2022-03-23 21:22:35 -04:00
description = '''Radiobot'''
2022-03-23 21:13:39 -04:00
bot = commands.Bot(command_prefix='!', description=description)
2022-03-23 19:59:21 -04:00
2022-03-23 21:13:39 -04:00
# Initialize opus
2022-03-23 21:23:56 -04:00
#if not discord.opus.is_loaded():
# discord.opus.load_opus('opus')
2022-03-23 21:13:39 -04:00
# Initialize some global variables
voice_client = None
text_channel = None
2022-03-23 19:27:00 -04:00
isPlaying = False
2022-03-23 21:13:39 -04:00
@bot.event
async def on_ready():
print('Logged in as ' + bot.user.name + ' #' + str(bot.user.id))
2022-03-23 21:13:39 -04:00
# get channels
#voice_channel = bot.get_channel(voice_channel_id)
#text_channel = bot.get_channel(text_channel_id)
#if not voice_channel:
# print("No voice channel " + voice_channel_id + " found!")
#if not text_channel:
# print("No text channel " + text_channel_id + " found!")
2022-03-23 21:13:39 -04:00
@bot.event
2022-03-23 21:55:30 -04:00
async def on_message(message):
2022-03-23 21:13:39 -04:00
# don't respond to ourselves
2022-03-23 21:55:30 -04:00
if message.author == bot.user:
2022-03-23 21:13:39 -04:00
return
2022-03-23 19:27:00 -04:00
if not str(message.channel.id) == text_channel_id:
return
print('<' + message.author.nick + '> ' + message.content)
2022-03-23 19:27:00 -04:00
2022-03-23 21:55:30 -04:00
if message.content == '!version':
await message.channel.send('] radiobot ' + bot_version + ' - python ' + os.environ['PYTHON_VERSION'])
2022-03-23 21:13:39 -04:00
@bot.event
async def on_voice_state_update(member, before, after):
"""
Starts events when a user changes their voice state.
Such as connecting, disconnecting and moving between channels.
:type member: discord.Member
:type before: discord.VoiceState
:type after: discord.VoiceState
:param member: The member that changed their voice state.
:param before: The member as they were before the change.
:param after: The member as they are after the change.
:return:
"""
2022-03-27 21:07:44 -04:00
print('before: ' + str(before))
print('after: ' + str(after))
return
clients_before = len(before.channel.members)
if member.bot:
print("self event detection")
return
2022-03-23 19:27:00 -04:00
if str(before.voice_channel) == "None":
print(after.name + " joined " + str(after.voice_channel))
if str(after.channel.id) == voice_channel_id:
print("Connecting to voice channel " + voice_channel_id)
voiceChannel = await after.channel.connect()
2022-03-23 19:27:00 -04:00
elif str(after.voice_channel) == "None":
print(after.name + " left " + str(before.voice_channel))
print("Disconnecting from voice channel " + voice_channel_id)
voiceChannel = await after.channel.connect()
2022-03-23 21:13:39 -04:00
await voiceChannel.disconnect()
2022-03-23 19:27:00 -04:00
2022-03-23 21:55:30 -04:00
# Start the bot with multiprocess compatiblity
if __name__ == "__main__":
try:
bot.loop.run_until_complete(bot.start(login_token))
finally:
bot.loop.close()