This commit is contained in:
BarsTiger
2021-04-14 10:10:04 +03:00
parent 547f07b8ee
commit a60ced932e

64
main.py
View File

@@ -1,27 +1,10 @@
# I M P O R T S # I M P O R T S
import discord
import os import os
import asyncio import asyncio
import discord
import youtube_dl import youtube_dl
from discord.ext import commands from discord.ext import commands
########################################################################################################################
#
# _____ _ _
# / ____| | | |
# | (___ | |_ __ _ _ __| |_ _ _ _ __
# \___ \| __/ _` | '__| __| | | | '_ \
# ____) | || (_| | | | |_| |_| | |_) |
# |_____/ \__\__,_|_| \__|\__,_| .__/
# | |
# |_|
token = os.getenv("TOKEN")
bot = commands.Bot(command_prefix=commands.when_mentioned_or("/"),
description="SnakieCat bot")
@bot.event
async def on_ready():
print('Yeah, I am started and my name is {0.user}'.format(bot))
######################################################################################################################### #########################################################################################################################
@@ -42,6 +25,7 @@ async def on_ready():
# M::::::M M::::::M UU:::::::::UU S:::::::::::::::SS I::::::::I CCC::::::::::::C # M::::::M M::::::M UU:::::::::UU S:::::::::::::::SS I::::::::I CCC::::::::::::C
# MMMMMMMM MMMMMMMM UUUUUUUUU SSSSSSSSSSSSSSS IIIIIIIIII CCCCCCCCCCCCC # MMMMMMMM MMMMMMMM UUUUUUUUU SSSSSSSSSSSSSSS IIIIIIIIII CCCCCCCCCCCCC
# Suppress noise about console usage from errors
youtube_dl.utils.bug_reports_message = lambda: '' youtube_dl.utils.bug_reports_message = lambda: ''
ytdl_format_options = { ytdl_format_options = {
@@ -64,6 +48,7 @@ ffmpeg_options = {
ytdl = youtube_dl.YoutubeDL(ytdl_format_options) ytdl = youtube_dl.YoutubeDL(ytdl_format_options)
class YTDLSource(discord.PCMVolumeTransformer): class YTDLSource(discord.PCMVolumeTransformer):
def __init__(self, source, *, data, volume=0.5): def __init__(self, source, *, data, volume=0.5):
super().__init__(source, volume) super().__init__(source, volume)
@@ -92,21 +77,17 @@ class Music(commands.Cog):
@commands.command() @commands.command()
async def join(self, ctx, *, channel: discord.VoiceChannel): async def join(self, ctx, *, channel: discord.VoiceChannel):
"""Joins a voice channel"""
if ctx.voice_client is not None: if ctx.voice_client is not None:
return await ctx.voice_client.move_to(channel) return await ctx.voice_client.move_to(channel)
await channel.connect() await channel.connect()
@commands.command() @commands.command()
async def play(self, ctx, *, url): async def play(self, ctx, *, url):
"""Streams from a url (same as yt, but doesn't predownload)"""
async with ctx.typing(): async with ctx.typing():
player = await YTDLSource.from_url(url, loop=self.bot.loop, stream=True) player = await YTDLSource.from_url(url, loop=self.bot.loop, stream=True)
ctx.voice_client.play(player, after=lambda e: print('Player error: %s' % e) if e else None) ctx.voice_client.play(player, after=lambda e: print('Player error: %s' % e) if e else None)
await ctx.send('Ok, (beep boop) I am turning on {}'.format(player.title)) await ctx.send('Now playing: {}'.format(player.title))
@commands.command() @commands.command()
async def volume(self, ctx, volume: int): async def volume(self, ctx, volume: int):
@@ -116,11 +97,12 @@ class Music(commands.Cog):
return await ctx.send("Not connected to a voice channel.") return await ctx.send("Not connected to a voice channel.")
ctx.voice_client.source.volume = volume / 100 ctx.voice_client.source.volume = volume / 100
await ctx.send("Now volume is {}%".format(volume)) await ctx.send("Changed volume to {}%".format(volume))
@commands.command() @commands.command()
async def stop(self, ctx): async def stop(self, ctx):
"""Stops and disconnects the bot from voice""" """Stops and disconnects the bot from voice"""
await ctx.voice_client.disconnect() await ctx.voice_client.disconnect()
@play.before_invoke @play.before_invoke
@@ -134,13 +116,28 @@ class Music(commands.Cog):
elif ctx.voice_client.is_playing(): elif ctx.voice_client.is_playing():
ctx.voice_client.stop() ctx.voice_client.stop()
########################################################################################################################
#
# _____ _ _
# / ____| | | |
# | (___ | |_ __ _ _ __| |_ _ _ _ __
# \___ \| __/ _` | '__| __| | | | '_ \
# ____) | || (_| | | | |_| |_| | |_) |
# |_____/ \__\__,_|_| \__|\__,_| .__/
# | |
# |_|
bot = commands.Bot(command_prefix=commands.when_mentioned_or("/"),
description="SnakieCat bot")
token = os.getenv("TOKEN")
######################################################################################################################## ########################################################################################################################
# #
# /$$ /$$ /$$ # /$$ /$$ /$$
# | $$ | $$ |__/ # | $$ | $$ |__/
# /$$$$$$ /$$$$$$ /$$ /$$ /$$$$$$ /$$$$$$ /$$$$$$$ /$$$$$$$ /$$ /$$ /$$ /$$$$$$ /$$$$$$ /$$ /$$$$$$$ /$$$$$$ # /$$$$$$ /$$$$$$ /$$ /$$ /$$$$$$ /$$$$$$ /$$$$$$$ /$$$$$$$ /$$ /$$ /$$ /$$$$$$ /$$$$$$ /$$ /$$$$$$$ /$$$$$$
#|_ $$_/ /$$__ $$| $$ /$$/|_ $$_/ |____ $$| $$__ $$ /$$_____/| $$ | $$ | $$ /$$__ $$ /$$__ $$| $$| $$__ $$ /$$__ $$ # |_ $$_/ /$$__ $$| $$ /$$/|_ $$_/ |____ $$| $$__ $$ /$$_____/| $$ | $$ | $$ /$$__ $$ /$$__ $$| $$| $$__ $$ /$$__ $$
# | $$ | $$$$$$$$ \ $$$$/ | $$ /$$$$$$$| $$ \ $$| $$$$$$ | $$ | $$ | $$| $$$$$$$$| $$ \__/| $$| $$ \ $$| $$ \ $$ # | $$ | $$$$$$$$ \ $$$$/ | $$ /$$$$$$$| $$ \ $$| $$$$$$ | $$ | $$ | $$| $$$$$$$$| $$ \__/| $$| $$ \ $$| $$ \ $$
# | $$ /$$| $$_____/ >$$ $$ | $$ /$$ /$$__ $$| $$ | $$ \____ $$| $$ | $$ | $$| $$_____/| $$ | $$| $$ | $$| $$ | $$ # | $$ /$$| $$_____/ >$$ $$ | $$ /$$ /$$__ $$| $$ | $$ \____ $$| $$ | $$ | $$| $$_____/| $$ | $$| $$ | $$| $$ | $$
# | $$$$/| $$$$$$$ /$$/\ $$ | $$$$/ | $$$$$$$| $$ | $$ /$$$$$$$/| $$$$$/$$$$/| $$$$$$$| $$ | $$| $$ | $$| $$$$$$$ # | $$$$/| $$$$$$$ /$$/\ $$ | $$$$/ | $$$$$$$| $$ | $$ /$$$$$$$/| $$$$$/$$$$/| $$$$$$$| $$ | $$| $$ | $$| $$$$$$$
@@ -149,8 +146,12 @@ class Music(commands.Cog):
# | $$$$$$/ # | $$$$$$/
# \______/ # \______/
@bot.event class Answers(commands.Cog):
async def on_message(message): def __init__(self, bot):
self.bot = bot
@bot.listen('on_message')
async def on_message(message):
if message.author == bot.user: if message.author == bot.user:
return return
@@ -158,15 +159,20 @@ async def on_message(message):
await message.channel.send('Hello!') await message.channel.send('Hello!')
if message.content.startswith('aboba') or message.content.startswith('ABOBA'): if message.content.startswith('aboba') or message.content.startswith('ABOBA'):
await message.channel.send(':regional_indicator_a: :regional_indicator_b: :regional_indicator_o: :regional_indicator_b: :regional_indicator_a:') await message.channel.send(
':regional_indicator_a: :regional_indicator_b: :regional_indicator_o: :regional_indicator_b: :regional_indicator_a:')
######################################################################################################################## ########################################################################################################################
# ____ __ ___ __ # ____ __ ___ __
# / __ \/ / / / | / / # / __ \/ / / / | / /
# / /_/ / / / / |/ / # / /_/ / / / / |/ /
# / _, _/ /_/ / /| / # / _, _/ /_/ / /| /
#/_/ |_|\____/_/ |_/ #/_/ |_|\____/_/ |_/
@bot.event
async def on_ready():
print('Yeah, I have started and my name is {0.user}'.format(bot))
bot.add_cog(Music(bot)) bot.add_cog(Music(bot))
bot.add_cog(Answers(bot))
bot.run(token) bot.run(token)