danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

better docs

Browse source
This commit is contained in:
Jesse Bannon 2022-10-07 22:21:41 -07:00
parent 8d5d65a093
commit 94b8803a3f
3 changed files with 13 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/ytdl_sub/plugins/audio_extract.py
View file

@ -8,7 +8,7 @@ from ytdl_sub.plugins.plugin import Plugin
from ytdl_sub.plugins.plugin import PluginOptions from ytdl_sub.plugins.plugin import PluginOptions
from ytdl_sub.utils.exceptions import FileNotDownloadedException from ytdl_sub.utils.exceptions import FileNotDownloadedException
from ytdl_sub.validators.audo_codec_validator import AUDIO_CODEC_TYPES_EXTENSION_MAPPING from ytdl_sub.validators.audo_codec_validator import AUDIO_CODEC_TYPES_EXTENSION_MAPPING
from ytdl_sub.validators.audo_codec_validator import CodecTypeValidator from ytdl_sub.validators.audo_codec_validator import AudioTypeValidator
from ytdl_sub.validators.validators import FloatValidator from ytdl_sub.validators.validators import FloatValidator
@ -31,7 +31,7 @@ class AudioExtractOptions(PluginOptions):
def __init__(self, name, value): def __init__(self, name, value):
super().__init__(name, value) super().__init__(name, value)
self._codec = self._validate_key(key="codec", validator=CodecTypeValidator).value self._codec = self._validate_key(key="codec", validator=AudioTypeValidator).value
self._quality = self._validate_key_if_present(key="quality", validator=FloatValidator) self._quality = self._validate_key_if_present(key="quality", validator=FloatValidator)
@property @property

11
src/ytdl_sub/plugins/file_convert.py
View file

@ -8,12 +8,12 @@ from ytdl_sub.plugins.plugin import Plugin
from ytdl_sub.plugins.plugin import PluginOptions from ytdl_sub.plugins.plugin import PluginOptions
from ytdl_sub.utils.exceptions import FileNotDownloadedException from ytdl_sub.utils.exceptions import FileNotDownloadedException
from ytdl_sub.utils.file_handler import FileMetadata from ytdl_sub.utils.file_handler import FileMetadata
from ytdl_sub.validators.audo_codec_validator import VideoCodecTypeValidator from ytdl_sub.validators.audo_codec_validator import FileTypeValidator
class FileConvertOptions(PluginOptions): class FileConvertOptions(PluginOptions):
""" """
Converts media files from one extension to another. Converts video files from one extension to another.
Usage: Usage:
@ -30,13 +30,16 @@ class FileConvertOptions(PluginOptions):
def __init__(self, name, value): def __init__(self, name, value):
super().__init__(name, value) super().__init__(name, value)
self._convert_to = self._validate_key( self._convert_to = self._validate_key(
key="convert_to", validator=VideoCodecTypeValidator key="convert_to", validator=FileTypeValidator
).value ).value
@property @property
def convert_to(self) -> str: def convert_to(self) -> str:
""" """
Convert to a desired extension Convert to a desired file type. Supports:
* Video: avi, flv, mkv, mov, mp4, webm
* Audio: aac, flac, mp3, m4a, opus, vorbis, wav
""" """
return self._convert_to return self._convert_to

24
src/ytdl_sub/validators/audo_codec_validator.py
View file

@ -16,30 +16,14 @@ AUDIO_CODEC_TYPES_EXTENSION_MAPPING: Dict[str, str] = {
AUDIO_CODEC_TYPES: Set[str] = set(AUDIO_CODEC_TYPES_EXTENSION_MAPPING.keys()) AUDIO_CODEC_TYPES: Set[str] = set(AUDIO_CODEC_TYPES_EXTENSION_MAPPING.keys())
AUDIO_CODEC_EXTS: Set[str] = set(AUDIO_CODEC_TYPES_EXTENSION_MAPPING.values()) AUDIO_CODEC_EXTS: Set[str] = set(AUDIO_CODEC_TYPES_EXTENSION_MAPPING.values())
VIDEO_CODEC_EXTS: Set[str] = { VIDEO_CODEC_EXTS: Set[str] = {"avi", "flv", "mkv", "mov", "mp4", "webm"}
"avi",
"flv",
"mkv",
"mov",
"mp4",
"webm",
"3g2",
"3gp",
"f4v",
"mk3d",
"divx",
"mpg",
"ogv",
"m4v",
"wmv",
}
class CodecTypeValidator(StringSelectValidator): class AudioTypeValidator(StringSelectValidator):
_expected_value_type_name = "codec" _expected_value_type_name = "codec"
_select_values = AUDIO_CODEC_TYPES _select_values = AUDIO_CODEC_TYPES
class VideoCodecTypeValidator(StringSelectValidator): class FileTypeValidator(StringSelectValidator):
_expected_value_type_name = "codec" _expected_value_type_name = "codec"
_select_values = VIDEO_CODEC_EXTS _select_values = AUDIO_CODEC_TYPES.union(VIDEO_CODEC_EXTS)