41 lines
No EOL
1.4 KiB
Python
41 lines
No EOL
1.4 KiB
Python
import os
|
|
import subprocess
|
|
import logging
|
|
|
|
from discord.ext import commands, tasks
|
|
from cogs.base_cog.bot_base_cog import BotBaseCog
|
|
|
|
class YoutubeDL(BotBaseCog):
|
|
|
|
def __init__(self, bot):
|
|
super().__init__(bot)
|
|
self.setup(__class__.__name__)
|
|
self.check_for_downloads.start()
|
|
|
|
@commands.command()
|
|
async def youtubedl(self, ctx):
|
|
try:
|
|
url = ctx.message.content.split(" ")[1]
|
|
url = '"' + url + '"'
|
|
video_or_audio = ctx.message.content.split(" ")[2]
|
|
process = subprocess.Popen(["python3", "data/ytdl/youtubedl.py", url, video_or_audio])
|
|
process.wait()
|
|
logging.error(process.stdout.read())
|
|
logging.error(process.stderr.read())
|
|
logging.error(process.returncode)
|
|
await ctx.send(f"Downloading {video_or_audio} from {url}...", suppress_embeds=True)
|
|
except:
|
|
await ctx.send("Usage: !youtubedl <url> <video|audio>")
|
|
|
|
#create a task
|
|
@tasks.loop(seconds=10)
|
|
async def check_for_downloads(self):
|
|
for file in os.listdir("data/ytdl"):
|
|
if file.endswith(".txt"):
|
|
with open(f"data/ytdl/{file}", "r") as f:
|
|
url = f.read()
|
|
await self.bot.get_channel(544408659174883328).send(f"{url}")
|
|
os.remove(f"data/ytdl/{file}")
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(YoutubeDL(bot)) |