danvics/ytptube
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Separate downloads fragments into their own temp dir

Browse source
This commit is contained in:
ArabCoders 2023-12-27 16:28:42 +03:00
parent a26453448a
commit 46f1ea616c
2 changed files with 12 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
README.md
View file

@ -21,7 +21,6 @@ Your `yt-dlp` config should include the following options for optimal working co
```json
{
"windowsfilenames": true,
"continue_dl": true,
"live_from_start": true,
"format_sort": [
"codec:avc:m4a"

14
app/src/Download.py
View file

@ -5,6 +5,7 @@ import logging
import multiprocessing
import os
import re
import shutil
import yt_dlp
from src.Utils import get_format, get_opts, jsonCookie, mergeConfig
from src.DTO.ItemDTO import ItemDTO
@ -23,6 +24,7 @@ class Download:
info: ItemDTO = None
default_ytdl_opts: dict = None
debug: bool = False
tempPath: str = None
_ytdlp_fields: tuple = (
'tmpfilename',
@ -85,12 +87,17 @@ class Download:
}
)
# Create temp dir for each download.
self.tempPath = os.path.join(self.temp_dir, self.info._id)
if not os.path.exists(self.tempPath):
os.makedirs(self.tempPath, exist_ok=True)
params: dict = {
'no_color': True,
'format': self.format,
'paths': {
'home': self.download_dir,
'temp': self.temp_dir
'temp': self.tempPath,
},
'outtmpl': {
'default': self.output_template,
@ -111,7 +118,7 @@ class Download:
if not data:
logging.warning(
f'The cookie string that was provided for {self.info.title} is empty or not in expected spec.')
with open(os.path.join(self.temp_dir, f'{self.info._id}.txt'), 'w') as f:
with open(os.path.join(self.tempPath, f'cookie_{self.info._id}.txt'), 'w') as f:
f.write(data)
params['cookiefile'] = f.name
@ -129,6 +136,9 @@ class Download:
except yt_dlp.utils.YoutubeDLError as exc:
self.status_queue.put({'status': 'error', 'msg': str(exc)})
if self.tempPath and self.info._id and os.path.exists(self.tempPath):
shutil.rmtree(self.tempPath, ignore_errors=True)
async def start(self, notifier):
if self.manager is None:
self.manager = multiprocessing.Manager()