From 791e75cb2beeca45a6edf52b65f7e299e03b5eaf Mon Sep 17 00:00:00 2001 From: SamlonDev Date: Sat, 14 Jun 2025 18:54:02 +0200 Subject: code-style issues --- cogs/downloader.py | 32 ++++++++++++++++++-------------- 1 file changed, 18 insertions(+), 14 deletions(-) (limited to 'cogs/downloader.py') diff --git a/cogs/downloader.py b/cogs/downloader.py index 7f48e57..c35373f 100644 --- a/cogs/downloader.py +++ b/cogs/downloader.py @@ -1,12 +1,13 @@ import base64 import os +import os.path +import tempfile + import discord -from discord.ext import commands import requests import yt_dlp +from discord.ext import commands from dotenv import load_dotenv -import tempfile -import os.path # Load environment variables load_dotenv() @@ -34,9 +35,9 @@ class Downloader(commands.Cog): description="download video from internet", ) async def download( - self, - ctx: discord.ApplicationContext, - url: discord.Option(str, "url", required=True) + self, + ctx: discord.ApplicationContext, + url: discord.Option(str, "url", required=True) ): await ctx.defer() try: @@ -52,7 +53,7 @@ class Downloader(commands.Cog): with yt_dlp.YoutubeDL(ydl_opts) as ydl: info = ydl.extract_info(url, download=True) actual_filename = ydl.prepare_filename(info) - + await ctx.followup.send("Video downloaded successfully, Uploading to pixeldrain...") # upload to pixeldrain try: @@ -61,37 +62,40 @@ class Downloader(commands.Cog): response = requests.post( self.pixeldrain_url, files=files, - headers={"Authorization": f"Basic {base64.b64encode((':' + PIXELDRAIN_API_KEY).encode()).decode()}"} + headers={ + "Authorization": f"Basic {base64.b64encode((':' + PIXELDRAIN_API_KEY).encode()).decode()}"} ) - + try: response_data = response.json() print("Pixeldrain response:", response_data) # Debug log - + if response.status_code == 201: # 201 means successful upload file_id = response_data.get('id') if file_id: file_link = f"https://pixeldrain.com/u/{file_id}" await ctx.followup.send(f"oto twoje videło\n {file_link}") else: - await ctx.followup.send(f"⚠️ Upload successful but no file ID returned. Response: {response_data}") + await ctx.followup.send( + f"⚠️ Upload successful but no file ID returned. Response: {response_data}") else: error_msg = response_data.get('error', 'Unknown error') if response.status_code == 401: error_msg = "Invalid API key. Please check your PIXELDRAIN_API_KEY in .env file." await ctx.followup.send(f"❌ Upload failed (Status {response.status_code}): {error_msg}") except ValueError as e: - # If response is not JSON + # If the response is not JSON error_msg = f"Failed to parse server response: {str(e)}" if response.text: error_msg += f"\nResponse: {response.text[:200]}" await ctx.followup.send(f"❌ {error_msg}") except Exception as e: await ctx.followup.send(f"Error: {str(e)}") - # No need for explicit cleanup as tempfile.TemporaryDirectory handles it automatically + # No need for explicit cleanup as a tempfile.TemporaryDirectory handles it automatically except Exception as e: await ctx.followup.send(f"Error: {str(e)}") + def setup(bot): - bot.add_cog(Downloader(bot)) \ No newline at end of file + bot.add_cog(Downloader(bot)) -- cgit v1.2.3