app/handlers/media_handler.py

# media_handler.py
from abc import ABC, abstractmethod
import logging
from app.services.download_media import download_whatsapp_media

logger = logging.getLogger(__name__)

class MediaHandler(ABC):
    @abstractmethod
    async def download(self, media_id: str, whatsapp_token: str, file_path: str) -> str:
        pass

class WhatsAppMediaHandler(MediaHandler):
    async def download(self, media_id: str, whatsapp_token: str, file_path: str) -> str:
        return await download_whatsapp_media(media_id, whatsapp_token, file_path)