46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import logging
|
|
from .DownloadQueue import DownloadQueue
|
|
from .encoder import Encoder
|
|
|
|
LOG = logging.getLogger('common')
|
|
|
|
|
|
class common():
|
|
"""
|
|
This class is used to share common methods between the socket and the API gateways.
|
|
"""
|
|
queue: DownloadQueue = None
|
|
encoder: Encoder = None
|
|
|
|
def __init__(self, queue: DownloadQueue, encoder: Encoder):
|
|
super().__init__()
|
|
self.queue = queue
|
|
self.encoder = encoder
|
|
|
|
async def add(
|
|
self,
|
|
url: str,
|
|
quality: str,
|
|
format: str,
|
|
folder: str,
|
|
ytdlp_cookies: str,
|
|
ytdlp_config: dict,
|
|
output_template: str
|
|
) -> dict[str, str]:
|
|
if ytdlp_config is None:
|
|
ytdlp_config = {}
|
|
|
|
if ytdlp_cookies and isinstance(ytdlp_cookies, dict):
|
|
ytdlp_cookies = self.encoder.encode(ytdlp_cookies)
|
|
|
|
status = await self.queue.add(
|
|
url=url,
|
|
format=format if format else 'any',
|
|
quality=quality if quality else 'best',
|
|
folder=folder,
|
|
ytdlp_cookies=ytdlp_cookies,
|
|
ytdlp_config=ytdlp_config,
|
|
output_template=output_template
|
|
)
|
|
|
|
return status
|