mirrors/ArchiveBox
1
0
Fork 0
mirror of https://github.com/ArchiveBox/ArchiveBox.git synced 2025-05-14 07:04:27 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

add FFMPEG_BINARY to ytdlp plugin

Browse source
This commit is contained in:
Nick Sweeting 2024-09-03 01:39:02 -07:00
parent c1c55d6da7
commit b927f5f2a9
No known key found for this signature in database
1 changed files with 18 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

20
archivebox/builtin_plugins/ytdlp/apps.py
View file

@ -1,11 +1,12 @@
import sys import sys
from pathlib import Path from pathlib import Path
from typing import List, Dict, Optional from typing import List, Dict, Optional
from subprocess import run, PIPE, CompletedProcess
from pydantic import InstanceOf, Field from pydantic import InstanceOf, Field
from django.apps import AppConfig from django.apps import AppConfig
from pydantic_pkgr import BinProvider, BinName, PATHStr from pydantic_pkgr import BinProvider, BinName, PATHStr, BinProviderName, ProviderLookupDict
from plugantic.base_plugin import BasePlugin, BaseConfigSet, BaseBinary, BaseBinProvider from plugantic.base_plugin import BasePlugin, BaseConfigSet, BaseBinary, BaseBinProvider
from plugantic.base_configset import ConfigSectionName from plugantic.base_configset import ConfigSectionName
@ -32,7 +33,22 @@ class YtdlpBinary(BaseBinary):
name: BinName = YTDLP_CONFIG.YTDLP_BINARY name: BinName = YTDLP_CONFIG.YTDLP_BINARY
binproviders_supported: List[InstanceOf[BinProvider]] = [env, pip, apt, brew] binproviders_supported: List[InstanceOf[BinProvider]] = [env, pip, apt, brew]
class FfmpegBinary(BaseBinary):
name: BinName = 'ffmpeg'
binproviders_supported: List[InstanceOf[BinProvider]] = [env, apt, brew]
provider_overrides: Dict[BinProviderName, ProviderLookupDict] = {
'env': {'version': lambda: run(['ffmpeg', '-version'], stdout=PIPE, stderr=PIPE, text=True).stdout},
'apt': {'version': lambda: run(['ffmpeg', '-version'], stdout=PIPE, stderr=PIPE, text=True).stdout},
'brew': {'version': lambda: run(['ffmpeg', '-version'], stdout=PIPE, stderr=PIPE, text=True).stdout},
}
# def get_ffmpeg_version(self) -> Optional[str]:
# return self.exec(cmd=['-version']).stdout
YTDLP_BINARY = YtdlpBinary() YTDLP_BINARY = YtdlpBinary()
FFMPEG_BINARY = FfmpegBinary()
class YtdlpPlugin(BasePlugin): class YtdlpPlugin(BasePlugin):
@ -41,7 +57,7 @@ class YtdlpPlugin(BasePlugin):
verbose_name: str = 'YTDLP' verbose_name: str = 'YTDLP'
configs: List[InstanceOf[BaseConfigSet]] = [YTDLP_CONFIG] configs: List[InstanceOf[BaseConfigSet]] = [YTDLP_CONFIG]
binaries: List[InstanceOf[BaseBinary]] = [YTDLP_BINARY] binaries: List[InstanceOf[BaseBinary]] = [YTDLP_BINARY, FFMPEG_BINARY]
PLUGIN = YtdlpPlugin() PLUGIN = YtdlpPlugin()