mirror of
https://github.com/pacnpal/Pac-cogs.git
synced 2025-12-20 02:41:06 -05:00
971d52bd00
Core processing logic in VideoProcessor class Dedicated message handling in MessageHandler Queue operations in QueueHandler Progress tracking in ProgressTracker Reaction management in reactions.py Improved Maintainability: Each component has a single responsibility Easier to test individual components Better code organization and readability Reduced file sizes for better version control Better Resource Management: Centralized progress tracking Isolated queue operations Cleaner cleanup processes Optimized Imports: Components can be imported individually as needed Main processor.py provides backward compatibility Clear module interface through init.py
98 lines
3.5 KiB
Python
98 lines
3.5 KiB
Python
"""Reaction emojis and reaction management for VideoProcessor"""
|
||
|
||
import logging
|
||
import asyncio
|
||
import discord
|
||
|
||
logger = logging.getLogger("VideoArchiver")
|
||
|
||
# Reaction emojis
|
||
REACTIONS = {
|
||
'queued': '📹',
|
||
'processing': '⚙️',
|
||
'success': '✅',
|
||
'error': '❌',
|
||
'numbers': ['1️⃣', '2️⃣', '3️⃣', '4️⃣', '5️⃣'],
|
||
'progress': ['⬛', '🟨', '🟩'],
|
||
'download': ['0️⃣', '2️⃣', '4️⃣', '6️⃣', '8️⃣', '🔟']
|
||
}
|
||
|
||
async def update_queue_position_reaction(message: discord.Message, position: int, bot_user) -> None:
|
||
"""Update queue position reaction"""
|
||
try:
|
||
for reaction in REACTIONS["numbers"]:
|
||
try:
|
||
await message.remove_reaction(reaction, bot_user)
|
||
except:
|
||
pass
|
||
|
||
if 0 <= position < len(REACTIONS["numbers"]):
|
||
await message.add_reaction(REACTIONS["numbers"][position])
|
||
logger.info(
|
||
f"Updated queue position reaction to {position + 1} for message {message.id}"
|
||
)
|
||
except Exception as e:
|
||
logger.error(f"Failed to update queue position reaction: {e}")
|
||
|
||
async def update_progress_reaction(message: discord.Message, progress: float, bot_user) -> None:
|
||
"""Update progress reaction based on FFmpeg progress"""
|
||
if not message:
|
||
return
|
||
|
||
try:
|
||
# Remove old reactions in the event loop
|
||
for reaction in REACTIONS["progress"]:
|
||
try:
|
||
await message.remove_reaction(reaction, bot_user)
|
||
except Exception as e:
|
||
logger.error(f"Failed to remove progress reaction: {e}")
|
||
continue
|
||
|
||
# Add new reaction based on progress
|
||
try:
|
||
if progress < 33:
|
||
await message.add_reaction(REACTIONS["progress"][0])
|
||
elif progress < 66:
|
||
await message.add_reaction(REACTIONS["progress"][1])
|
||
else:
|
||
await message.add_reaction(REACTIONS["progress"][2])
|
||
except Exception as e:
|
||
logger.error(f"Failed to add progress reaction: {e}")
|
||
|
||
except Exception as e:
|
||
logger.error(f"Failed to update progress reaction: {e}")
|
||
|
||
async def update_download_progress_reaction(message: discord.Message, progress: float, bot_user) -> None:
|
||
"""Update download progress reaction"""
|
||
if not message:
|
||
return
|
||
|
||
try:
|
||
# Remove old reactions in the event loop
|
||
for reaction in REACTIONS["download"]:
|
||
try:
|
||
await message.remove_reaction(reaction, bot_user)
|
||
except Exception as e:
|
||
logger.error(f"Failed to remove download reaction: {e}")
|
||
continue
|
||
|
||
# Add new reaction based on progress
|
||
try:
|
||
if progress <= 20:
|
||
await message.add_reaction(REACTIONS["download"][0])
|
||
elif progress <= 40:
|
||
await message.add_reaction(REACTIONS["download"][1])
|
||
elif progress <= 60:
|
||
await message.add_reaction(REACTIONS["download"][2])
|
||
elif progress <= 80:
|
||
await message.add_reaction(REACTIONS["download"][3])
|
||
elif progress < 100:
|
||
await message.add_reaction(REACTIONS["download"][4])
|
||
else:
|
||
await message.add_reaction(REACTIONS["download"][5])
|
||
except Exception as e:
|
||
logger.error(f"Failed to add download reaction: {e}")
|
||
|
||
except Exception as e:
|
||
logger.error(f"Failed to update download progress reaction: {e}")
|