Autopep-ed the code and simplified the intents

This commit is contained in:
Louis Lacoste 2022-11-18 12:14:05 +01:00
parent 7cf89decd3
commit 618ada202a

59
main.py
View file

@ -1,12 +1,13 @@
#Polarobot is a discord bot designed mainly to play music
#Copyright (C) 2022 Louis Lacoste
# Polarobot is a discord bot designed mainly to play music
# Copyright (C) 2022 Louis Lacoste
#This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
#This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more details. You should have received a copy of the GNU Affero General Public License along with this program. If not, see <http://www.gnu.org/licenses/>.
# This program is free software: you can redistribute it and/or modify it under the terms of the GNU Affero General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more details. You should have received a copy of the GNU Affero General Public License along with this program. If not, see <http://www.gnu.org/licenses/>.
#Also add information on how to contact you by electronic and paper mail.
# Also add information on how to contact you by electronic and paper mail.
import os, sys
import os
import sys
import re
import random
import discord
@ -18,16 +19,20 @@ from dotenv import load_dotenv
load_dotenv()
TOKEN = os.getenv('DISCORD_TOKEN')
intents = discord.Intents.default()
intents.message_content = True
intents.members = True
#intents = discord.Intents.default()
#intents.message_content = True
#intents.members = True
intents = discord.Intents().all()
client = discord.Client(intents=intents)
bot = commands.Bot(command_prefix="!p ", intents=intents)
youtubeUrlRegex = "https?:\/\/(www\.)?(youtu|youtube)\.(com|be)"
def is_a_youtube_url(url: str):
return bool(re.match(youtubeUrlRegex, url))
return bool(re.match(youtubeUrlRegex, url))
def random_playing_gif():
gifList = ["https://media.tenor.com/Ra6rkhDtYPwAAAAd/anime-dj.gif",
@ -38,51 +43,65 @@ def random_playing_gif():
"https://media.tenor.com/6z00WD2k8foAAAAM/boombox-jamming.gif"]
return random.choice(gifList)
@bot.event
async def on_ready():
print("Starting")
await bot.change_presence(activity= discord.Activity(type = discord.ActivityType.playing, name = 'PolaroBot | !p to call'))
await bot.change_presence(activity=discord.Activity(type=discord.ActivityType.playing, name='PolaroBot | !p to call'))
def restart_bot():
os.execv(sys.executable, ['python'] + sys.argv)
@bot.command(aliases=['reboot'])
async def restart(ctx):
"""Reboot bot"""
bot.current_ctx = ctx
print("Rebooting...")
embedDeco=discord.Embed(title="🤖PolaroBot Statut",
description="PolaroBot Redémarre : Patientez quelques secondes.. ",color=0xF1D50E)
embedDeco = discord.Embed(title="🤖PolaroBot Statut",
description="PolaroBot redémarre : Patientez quelques secondes.. ", color=0xF1D50E)
await ctx.send(embed=embedDeco)
await bot.change_presence(activity=discord.Activity(type=discord.ActivityType.playing, name='PolaroBot redémarre...'))
restart_bot()
@bot.command(aliases=['play_music'],pass_context=True)
@bot.command(aliases=['play_music'], pass_context=True)
async def play(ctx, url: str):
"""Playing a song"""
if is_a_youtube_url(url):
embedPlaying=discord.Embed(title="🤖PolaroBot is playing", description=f"Je sais pas encore jouer de musique mais tkt frère, vla ton URL mon reuf : {url}")
embedPlaying = discord.Embed(
title="🤖PolaroBot joue", description=f"Je sais pas encore jouer de musique mais tkt frère, vla ton URL mon reuf : {url}")
embedPlaying.set_image(url=random_playing_gif())
else:
embedPlaying=discord.Embed(title="🤖PolaroBot veut casser ta gueule", description=f"FREROT ?! C'EST QUOI {url} ? C'EST DE LA MERDE 💩💩💩 ! ")
embedPlaying.set_image(url="https://media.tenor.com/wQH1Lm24wLwAAAAM/de-la-merde-jean.gif")
embedPlaying = discord.Embed(title="🤖PolaroBot veut casser ta gueule",
description=f"FREROT ?! C'EST QUOI {url} ? C'EST DE LA MERDE 💩💩💩 ! ")
embedPlaying.set_image(
url="https://media.tenor.com/wQH1Lm24wLwAAAAM/de-la-merde-jean.gif")
await ctx.send(embed=embedPlaying)
@bot.command()
async def join(ctx):
channel = ctx.author.voice.channel
if channel:
print(f"Voice : connecting to {channel.id}")
bot.voice_channel = await channel.connect()
embedMessage=discord.Embed(title="PolaroBot connected", description=f"PolaroBot has connected to {ctx.author}'s channel")
embedMessage = discord.Embed(
title="PolaroBot connected", description=f"PolaroBot has connected to {ctx.author}'s channel")
else:
embedMessage=discord.Embed(title="PolaroBot can't connect", description=f"Are you connected to a voice channel ?")
embedMessage = discord.Embed(
title="PolaroBot can't connect", description=f"Are you connected to a voice channel ?")
await ctx.send(embed=embedMessage)
@bot.command()
async def leave(ctx):
print(f"Voice : leaving {bot.voice_channel.id}")
await ctx.voice_client.disconnect()
@bot.event
async def on_command_error(ctx, error):
channel = ctx.message.channel