From 507f03e716478fce3ee9b9c3f3f5964c34b5b2dd Mon Sep 17 00:00:00 2001 From: Samlon Date: Fri, 25 Jul 2025 19:36:56 +0200 Subject: play.py fix --- cogs/downloader.py | 58 +++++------------------------------------------------- 1 file changed, 5 insertions(+), 53 deletions(-) (limited to 'cogs/downloader.py') diff --git a/cogs/downloader.py b/cogs/downloader.py index c35373f..5fb3d1c 100644 --- a/cogs/downloader.py +++ b/cogs/downloader.py @@ -1,34 +1,19 @@ -import base64 import os -import os.path import tempfile +from pathlib import Path import discord -import requests import yt_dlp from discord.ext import commands from dotenv import load_dotenv # Load environment variables load_dotenv() -PIXELDRAIN_API_KEY = os.getenv('PIXELDRAIN_API_KEY') - - -def _get_auth_headers(): - """Generate authentication headers for Pixeldrain API""" - if not PIXELDRAIN_API_KEY: - raise Exception("Pixeldrain API key not found in environment variables") - - auth_string = f":{PIXELDRAIN_API_KEY}" - auth_bytes = auth_string.encode('ascii') - base64_auth = base64.b64encode(auth_bytes).decode('ascii') - return {"Authorization": f"Basic {base64_auth}"} class Downloader(commands.Cog): def __init__(self, bot): self.bot = bot - self.pixeldrain_url = "https://pixeldrain.com/api/file" @commands.slash_command( name="download", @@ -54,47 +39,14 @@ class Downloader(commands.Cog): 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: - with open(actual_filename, 'rb') as f: - files = {'file': f} - response = requests.post( - self.pixeldrain_url, - files=files, - 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}") - 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 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}") + file = discord.File(actual_filename) + await ctx.followup.send("Pobran", file=file) except Exception as e: - await ctx.followup.send(f"Error: {str(e)}") - # No need for explicit cleanup as a tempfile.TemporaryDirectory handles it automatically + await ctx.followup.send(f":(\n{str(e)}") except Exception as e: - await ctx.followup.send(f"Error: {str(e)}") + await ctx.followup.send(f":(\n{str(e)}") def setup(bot): -- cgit v1.2.3