hoelee/tubesync
1
0
Fork 0
mirror of https://github.com/meeb/tubesync.git synced 2025-06-24 14:06:36 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Use a temporary directory per task

Browse Source 
I am seeing multiple ffmpeg processes that mess each other up.

I do not know why this is happening yet, but for now let them operate in their own temporary directories.
This commit is contained in:
tcely 2025-01-29 03:55:52 -05:00 committed by GitHub
parent 0eb7a4dab1
commit 78b9aadcbf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
tubesync/sync/youtube.py
View File

@ -8,8 +8,10 @@ import os
from collections import namedtuple from collections import namedtuple
from common.logger import log from common.logger import log
from copy import copy, deepcopy from copy import deepcopy
from pathlib import Path from pathlib import Path
from tempfile import TemporaryDirectory
from urllib.parse import urlsplit, parse_qs
from django.conf import settings from django.conf import settings
from .utils import mkdir_p from .utils import mkdir_p
@ -41,7 +43,7 @@ class YouTubeError(yt_dlp.utils.DownloadError):
def get_yt_opts(): def get_yt_opts():
opts = copy(_defaults) opts = deepcopy(_defaults)
cookie_file = settings.COOKIES_FILE cookie_file = settings.COOKIES_FILE
if cookie_file.is_file(): if cookie_file.is_file():
cookie_file_path = str(cookie_file.resolve()) cookie_file_path = str(cookie_file.resolve())
@ -244,12 +246,22 @@ def download_media(url, media_format, extension, output_file, info_json,
} }
opts = get_yt_opts() opts = get_yt_opts()
ytopts['paths'] = opts.get('paths', {}) ytopts['paths'] = opts.get('paths', {})
output_dir = os.path.dirname(output_file)
temp_dir_parent = output_dir
temp_dir_prefix = '.yt_dlp-'
if 'temp' in ytopts['paths']:
v_key = parse_qs(urlsplit(url).query).get('v').pop()
temp_dir_parent = ytopts['paths']['temp']
temp_dir_prefix = f'{temp_dir_prefix}{v_key}-'
temp_dir = TemporaryDirectory(prefix=temp_dir_prefix,dir=temp_dir_parent)
(Path(temp_dir) / '.ignore').touch(exist_ok=True)
ytopts['paths'].update({ ytopts['paths'].update({
'home': os.path.dirname(output_file), 'home': output_dir,
'temp': temp_dir,
}) })
codec_options = [] codec_options = []
ofn = os.path.basename(output_file) ofn = ytopts['outtmpl']
if 'av1-' in ofn: if 'av1-' in ofn:
codec_options = ['-c:v', 'libsvtav1', '-preset', '8', '-crf', '35'] codec_options = ['-c:v', 'libsvtav1', '-preset', '8', '-crf', '35']
elif 'vp9-' in ofn: elif 'vp9-' in ofn: