Compare commits
13 commits
2026.03.19
...
master
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
658f5ef3e1 | ||
|
|
1c4633b815 | ||
|
|
19b0e621ca | ||
|
|
4421c4eede | ||
|
|
b195f37412 | ||
|
|
108c7cfa14 | ||
|
|
772f01a734 | ||
|
|
bb96f00ca5 | ||
|
|
915d445e21 | ||
|
|
970c74ba45 | ||
|
|
e34b9b6295 | ||
|
|
49e3b5f887 | ||
|
|
5cb9bbea9c |
30 changed files with 431 additions and 63 deletions
|
|
@ -242,3 +242,22 @@ Be sure to tell ytdl-sub to use your config by using the argument ``--config
|
||||||
|
|
||||||
If you run ytdl-sub in the same directory, and the config file is named ``config.yaml``,
|
If you run ytdl-sub in the same directory, and the config file is named ``config.yaml``,
|
||||||
it will use it by default.
|
it will use it by default.
|
||||||
|
|
||||||
|
Visualizing a subscription in Preset form
|
||||||
|
-----------------------------------------
|
||||||
|
|
||||||
|
Subscription file syntax is designed to minimize boiler-plate when authoring new subscriptions.
|
||||||
|
You can unpack any subscription using the ``inspect`` sub-command to see its boiler-plate *preset format*.
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
ytdl-sub inspect --match "BBC News" /path/to/subscriptions.yaml
|
||||||
|
|
||||||
|
This can be utilized for numerous purposes including:
|
||||||
|
|
||||||
|
* Ensuring your custom preset is getting applied correctly.
|
||||||
|
* Figuring out which variables set things like file names, metadata, etc.
|
||||||
|
* Understanding how subscription syntax translates to preset representation.
|
||||||
|
|
||||||
|
The default ``--level`` of inspect will fill in defined variables. Using ``--level original`` will
|
||||||
|
present the subscription's raw layout with no fill.
|
||||||
|
|
|
||||||
|
|
@ -112,3 +112,35 @@ Convert yt-dlp cli arguments to ytdl-sub `ytdl_options` arguments.
|
||||||
.. code-block::
|
.. code-block::
|
||||||
|
|
||||||
ytdl-sub cli-to-sub [YT-DLP ARGS]
|
ytdl-sub cli-to-sub [YT-DLP ARGS]
|
||||||
|
|
||||||
|
Inspect
|
||||||
|
-------
|
||||||
|
Inspect a single subscription's underlying preset representation.
|
||||||
|
This can be utilized for numerous purposes including:
|
||||||
|
|
||||||
|
* Ensuring your custom preset is getting applied correctly.
|
||||||
|
* Figuring out which variables set things like file names, metadata, etc.
|
||||||
|
* Understanding how subscription syntax translates to preset representation.
|
||||||
|
|
||||||
|
Usage:
|
||||||
|
|
||||||
|
.. code-block:: bash
|
||||||
|
|
||||||
|
ytdl-sub inspect --match "Game Chops" --mock 'title=Lets Play' examples/music_subscriptions.yaml
|
||||||
|
|
||||||
|
.. code-block:: text
|
||||||
|
:caption: Additional Options
|
||||||
|
|
||||||
|
-l 0,1,2,3, --level 0,1,2,3
|
||||||
|
level of inspection to perform:
|
||||||
|
0 - original present the subscription as-is
|
||||||
|
1 - fill fill in defined values
|
||||||
|
2 - resolve resolve all possible variables (default)
|
||||||
|
3 - internal resolve all variables to their internal representation
|
||||||
|
|
||||||
|
-m MATCH [MATCH ...], --match MATCH [MATCH ...]
|
||||||
|
match subscription names to one or more substrings, and only run those subscriptions
|
||||||
|
-o DL_OVERRIDE, --dl-override DL_OVERRIDE
|
||||||
|
override all subscription config values using `dl` syntax, i.e. --dl-override='--ytdl_options.max_downloads 3'
|
||||||
|
-k VAR=VALUE, --mock VAR=VALUE
|
||||||
|
ability to mock one or more variable values, i.e. --mock 'title=Lets Play'
|
||||||
|
|
|
||||||
|
|
@ -15,7 +15,7 @@ classifiers = [
|
||||||
"Programming Language :: Python :: 3.11",
|
"Programming Language :: Python :: 3.11",
|
||||||
]
|
]
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"yt-dlp[default]==2026.3.17",
|
"yt-dlp[default]==2026.6.9",
|
||||||
"colorama~=0.4",
|
"colorama~=0.4",
|
||||||
"mergedeep~=1.3",
|
"mergedeep~=1.3",
|
||||||
"mediafile~=0.12",
|
"mediafile~=0.12",
|
||||||
|
|
@ -48,7 +48,7 @@ test = [
|
||||||
]
|
]
|
||||||
lint = [
|
lint = [
|
||||||
"pylint==4.0.5",
|
"pylint==4.0.5",
|
||||||
"ruff==0.15.6",
|
"ruff==0.15.16",
|
||||||
]
|
]
|
||||||
docs = [
|
docs = [
|
||||||
"sphinx>=7,<10",
|
"sphinx>=7,<10",
|
||||||
|
|
|
||||||
|
|
@ -17,12 +17,15 @@ from ytdl_sub.cli.parsers.cli_to_sub import print_cli_to_sub
|
||||||
from ytdl_sub.cli.parsers.dl import DownloadArgsParser
|
from ytdl_sub.cli.parsers.dl import DownloadArgsParser
|
||||||
from ytdl_sub.cli.parsers.main import DEFAULT_CONFIG_FILE_NAME, parser
|
from ytdl_sub.cli.parsers.main import DEFAULT_CONFIG_FILE_NAME, parser
|
||||||
from ytdl_sub.config.config_file import ConfigFile
|
from ytdl_sub.config.config_file import ConfigFile
|
||||||
|
from ytdl_sub.config.validators.variable_validation import ResolutionLevel
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
from ytdl_sub.utils.exceptions import ExperimentalFeatureNotEnabled, ValidationException
|
from ytdl_sub.utils.exceptions import ExperimentalFeatureNotEnabled, ValidationException
|
||||||
from ytdl_sub.utils.file_handler import FileHandler
|
from ytdl_sub.utils.file_handler import FileHandler
|
||||||
from ytdl_sub.utils.file_lock import working_directory_lock
|
from ytdl_sub.utils.file_lock import working_directory_lock
|
||||||
from ytdl_sub.utils.logger import Logger
|
from ytdl_sub.utils.logger import Logger
|
||||||
|
|
||||||
|
# pylint: disable=too-many-branches
|
||||||
|
|
||||||
logger = Logger.get()
|
logger = Logger.get()
|
||||||
|
|
||||||
# View is a command to run a simple dry-run on a URL using the `_view` preset.
|
# View is a command to run a simple dry-run on a URL using the `_view` preset.
|
||||||
|
|
@ -202,6 +205,44 @@ def _view_url_from_cli(config: ConfigFile, url: str, split_chapters: bool) -> Su
|
||||||
return subscription
|
return subscription
|
||||||
|
|
||||||
|
|
||||||
|
def _parse_inspect_mocks(mocks: Optional[List[str]]) -> Dict[str, str]:
|
||||||
|
out: Dict[str, str] = {}
|
||||||
|
for mock in mocks or []:
|
||||||
|
spl = mock.split("=", 1)
|
||||||
|
if len(spl) == 1:
|
||||||
|
raise ValidationException("inspect mock must be in the form of VAR=VALUE")
|
||||||
|
out[spl[0].strip()] = spl[1]
|
||||||
|
|
||||||
|
return out
|
||||||
|
|
||||||
|
|
||||||
|
def _inspect(
|
||||||
|
config: ConfigFile,
|
||||||
|
subscription_paths: List[str],
|
||||||
|
subscription_matches: List[str],
|
||||||
|
subscription_override_dict: Dict,
|
||||||
|
inspection_level: int,
|
||||||
|
mocks: Dict[str, str],
|
||||||
|
) -> None:
|
||||||
|
|
||||||
|
subscriptions: List[Subscription] = []
|
||||||
|
for path in subscription_paths:
|
||||||
|
subscriptions += Subscription.from_file_path(
|
||||||
|
config=config,
|
||||||
|
subscription_path=path,
|
||||||
|
subscription_matches=subscription_matches,
|
||||||
|
subscription_override_dict=subscription_override_dict,
|
||||||
|
)
|
||||||
|
|
||||||
|
if len(subscriptions) > 1:
|
||||||
|
print(
|
||||||
|
"inspect can only inspect a single subscription. Use --match to filter for a single one"
|
||||||
|
)
|
||||||
|
return
|
||||||
|
|
||||||
|
print(subscriptions[0].resolved_yaml(resolution_level=inspection_level, mocks=mocks))
|
||||||
|
|
||||||
|
|
||||||
def main() -> List[Subscription]:
|
def main() -> List[Subscription]:
|
||||||
"""
|
"""
|
||||||
Entrypoint for ytdl-sub, without the error handling
|
Entrypoint for ytdl-sub, without the error handling
|
||||||
|
|
@ -228,6 +269,23 @@ def main() -> List[Subscription]:
|
||||||
|
|
||||||
subscriptions: List[Subscription] = []
|
subscriptions: List[Subscription] = []
|
||||||
|
|
||||||
|
if args.subparser == "inspect":
|
||||||
|
subscription_override_dict = {}
|
||||||
|
if args.dl_override:
|
||||||
|
subscription_override_dict = DownloadArgsParser.from_dl_override(
|
||||||
|
override=args.dl_override, config=config
|
||||||
|
).to_subscription_dict()
|
||||||
|
|
||||||
|
_inspect(
|
||||||
|
config=config,
|
||||||
|
subscription_paths=args.subscription_paths,
|
||||||
|
subscription_matches=args.match,
|
||||||
|
subscription_override_dict=subscription_override_dict,
|
||||||
|
inspection_level=ResolutionLevel.level_number(args.inspection_level),
|
||||||
|
mocks=_parse_inspect_mocks(args.mock),
|
||||||
|
)
|
||||||
|
return []
|
||||||
|
|
||||||
# If transaction log file is specified, make sure we can open it
|
# If transaction log file is specified, make sure we can open it
|
||||||
_maybe_validate_transaction_log_file(transaction_log_file_path=args.transaction_log)
|
_maybe_validate_transaction_log_file(transaction_log_file_path=args.transaction_log)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -1,6 +1,6 @@
|
||||||
import argparse
|
import argparse
|
||||||
import dataclasses
|
import dataclasses
|
||||||
from typing import List
|
from typing import Dict, List
|
||||||
|
|
||||||
from ytdl_sub import __local_version__
|
from ytdl_sub import __local_version__
|
||||||
from ytdl_sub.utils.logger import LoggerLevels
|
from ytdl_sub.utils.logger import LoggerLevels
|
||||||
|
|
@ -236,3 +236,77 @@ view_parser.add_argument("url", help="URL to view source variables for")
|
||||||
###################################################################################################
|
###################################################################################################
|
||||||
# CLI-TO-SUB PARSER
|
# CLI-TO-SUB PARSER
|
||||||
cli_to_sub_parser = subparsers.add_parser("cli-to-sub")
|
cli_to_sub_parser = subparsers.add_parser("cli-to-sub")
|
||||||
|
|
||||||
|
###################################################################################################
|
||||||
|
# INSPECT PARSER
|
||||||
|
|
||||||
|
|
||||||
|
class InspectArguments:
|
||||||
|
LEVEL = CLIArgument(
|
||||||
|
short="-l",
|
||||||
|
long="--level",
|
||||||
|
)
|
||||||
|
LevelChoices: Dict[str, str] = {
|
||||||
|
"0": "original",
|
||||||
|
"1": "fill",
|
||||||
|
"2": "resolve",
|
||||||
|
"3": "internal",
|
||||||
|
}
|
||||||
|
|
||||||
|
MOCK = CLIArgument(
|
||||||
|
short="-k",
|
||||||
|
long="--mock",
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
inspect_parser = subparsers.add_parser("inspect", formatter_class=argparse.RawTextHelpFormatter)
|
||||||
|
inspect_parser.add_argument(
|
||||||
|
InspectArguments.LEVEL.short,
|
||||||
|
InspectArguments.LEVEL.long,
|
||||||
|
metavar=",".join(str(i) for i in range(4)),
|
||||||
|
type=str,
|
||||||
|
help="""level of inspection to perform:
|
||||||
|
0 - original present the subscription as-is
|
||||||
|
1 - fill fill in defined values
|
||||||
|
2 - resolve resolve all possible variables (default)
|
||||||
|
3 - internal resolve all variables to their internal representation
|
||||||
|
""",
|
||||||
|
default="resolve",
|
||||||
|
choices=list(InspectArguments.LevelChoices.keys())
|
||||||
|
+ list(InspectArguments.LevelChoices.values()),
|
||||||
|
dest="inspection_level",
|
||||||
|
)
|
||||||
|
inspect_parser.add_argument(
|
||||||
|
MainArguments.MATCH.short,
|
||||||
|
MainArguments.MATCH.long,
|
||||||
|
dest="match",
|
||||||
|
nargs="+",
|
||||||
|
action="extend",
|
||||||
|
type=str,
|
||||||
|
help="match subscription names to one or more substrings, and only run those subscriptions",
|
||||||
|
default=[],
|
||||||
|
)
|
||||||
|
|
||||||
|
inspect_parser.add_argument(
|
||||||
|
"subscription_paths",
|
||||||
|
metavar="SUBPATH",
|
||||||
|
nargs="*",
|
||||||
|
help="path to subscription files, uses subscriptions.yaml if not provided",
|
||||||
|
default=["subscriptions.yaml"],
|
||||||
|
)
|
||||||
|
|
||||||
|
inspect_parser.add_argument(
|
||||||
|
SubArguments.OVERRIDE.short,
|
||||||
|
SubArguments.OVERRIDE.long,
|
||||||
|
type=str,
|
||||||
|
help="override all subscription config values using `dl` syntax, "
|
||||||
|
"i.e. --dl-override='--ytdl_options.max_downloads 3'",
|
||||||
|
)
|
||||||
|
|
||||||
|
inspect_parser.add_argument(
|
||||||
|
InspectArguments.MOCK.short,
|
||||||
|
InspectArguments.MOCK.long,
|
||||||
|
metavar="VAR=VALUE",
|
||||||
|
action="append",
|
||||||
|
help="ability to mock one or more variable values, i.e. --mock 'title=Lets Play'",
|
||||||
|
)
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,4 @@
|
||||||
from typing import Dict, List, Set
|
from typing import Dict, List, Optional, Set
|
||||||
|
|
||||||
from ytdl_sub.config.overrides import Overrides
|
from ytdl_sub.config.overrides import Overrides
|
||||||
from ytdl_sub.config.plugin.plugin_mapping import PluginMapping
|
from ytdl_sub.config.plugin.plugin_mapping import PluginMapping
|
||||||
|
|
@ -35,6 +35,21 @@ class ResolutionLevel:
|
||||||
return "internal"
|
return "internal"
|
||||||
raise ValueError("Invalid resolution level")
|
raise ValueError("Invalid resolution level")
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def level_number(cls, resolution_arg: str) -> int:
|
||||||
|
"""
|
||||||
|
Numeric resolution level
|
||||||
|
"""
|
||||||
|
if resolution_arg in ("0", "original"):
|
||||||
|
return 0
|
||||||
|
if resolution_arg in ("1", "fill"):
|
||||||
|
return 1
|
||||||
|
if resolution_arg in ("2", "resolve"):
|
||||||
|
return 2
|
||||||
|
if resolution_arg in ("3", "internal"):
|
||||||
|
return 3
|
||||||
|
raise ValueError("Invalid resolution level")
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def all(cls) -> List[int]:
|
def all(cls) -> List[int]:
|
||||||
"""
|
"""
|
||||||
|
|
@ -53,7 +68,7 @@ class VariableValidation:
|
||||||
if name not in self.unresolved_variables and not name.endswith("_sanitized")
|
if name not in self.unresolved_variables and not name.endswith("_sanitized")
|
||||||
}
|
}
|
||||||
|
|
||||||
def _apply_resolution_level(self) -> None:
|
def _apply_resolution_level(self, mocks: Optional[Dict[str, str]]) -> None:
|
||||||
if self._resolution_level == ResolutionLevel.FILL:
|
if self._resolution_level == ResolutionLevel.FILL:
|
||||||
self.unresolved_variables |= VARIABLES.variable_names(include_sanitized=True)
|
self.unresolved_variables |= VARIABLES.variable_names(include_sanitized=True)
|
||||||
# Only partial resolve definitions that are already resolved
|
# Only partial resolve definitions that are already resolved
|
||||||
|
|
@ -71,6 +86,16 @@ class VariableValidation:
|
||||||
else:
|
else:
|
||||||
raise ValueError("Invalid resolution level for validation")
|
raise ValueError("Invalid resolution level for validation")
|
||||||
|
|
||||||
|
if mocks is not None:
|
||||||
|
for mock_name in mocks.keys():
|
||||||
|
if mock_name in self.unresolved_variables:
|
||||||
|
self.unresolved_variables.remove(mock_name)
|
||||||
|
|
||||||
|
self.script.add(
|
||||||
|
variables=mocks,
|
||||||
|
unresolvable=self.unresolved_variables,
|
||||||
|
)
|
||||||
|
|
||||||
self.script = self.script.resolve_partial(
|
self.script = self.script.resolve_partial(
|
||||||
unresolvable=self.unresolved_variables,
|
unresolvable=self.unresolved_variables,
|
||||||
output_filter=self._get_resolve_partial_filter(),
|
output_filter=self._get_resolve_partial_filter(),
|
||||||
|
|
@ -83,6 +108,7 @@ class VariableValidation:
|
||||||
output_options: OutputOptions,
|
output_options: OutputOptions,
|
||||||
plugins: PresetPlugins,
|
plugins: PresetPlugins,
|
||||||
resolution_level: int = ResolutionLevel.RESOLVE,
|
resolution_level: int = ResolutionLevel.RESOLVE,
|
||||||
|
mocks: Optional[Dict[str, str]] = None,
|
||||||
):
|
):
|
||||||
self.overrides = overrides
|
self.overrides = overrides
|
||||||
self.downloader_options = downloader_options
|
self.downloader_options = downloader_options
|
||||||
|
|
@ -100,8 +126,7 @@ class VariableValidation:
|
||||||
additional_options=[self.output_options, self.downloader_options]
|
additional_options=[self.output_options, self.downloader_options]
|
||||||
)
|
)
|
||||||
self._resolution_level = resolution_level
|
self._resolution_level = resolution_level
|
||||||
|
self._apply_resolution_level(mocks=mocks)
|
||||||
self._apply_resolution_level()
|
|
||||||
|
|
||||||
def _add_runtime_variables(self, plugin_op: PluginOperation, options: OptionsValidator) -> None:
|
def _add_runtime_variables(self, plugin_op: PluginOperation, options: OptionsValidator) -> None:
|
||||||
"""
|
"""
|
||||||
|
|
@ -114,6 +139,20 @@ class VariableValidation:
|
||||||
|
|
||||||
self.unresolved_runtime_variables -= added_variables | modified_variables
|
self.unresolved_runtime_variables -= added_variables | modified_variables
|
||||||
|
|
||||||
|
def _output_override_variables(self) -> Dict:
|
||||||
|
output = {}
|
||||||
|
for name in self.overrides.keys:
|
||||||
|
value = self.script.definition_of(name)
|
||||||
|
if name in self.script.function_names:
|
||||||
|
# Keep custom functions as-is
|
||||||
|
output[name] = self.overrides.dict_with_format_strings[name]
|
||||||
|
elif resolved := value.maybe_resolvable:
|
||||||
|
output[name] = resolved.native
|
||||||
|
else:
|
||||||
|
output[name] = ScriptUtils.to_native_script(value)
|
||||||
|
|
||||||
|
return output
|
||||||
|
|
||||||
def ensure_proper_usage(self, partial_resolve_formatters: bool = False) -> Dict:
|
def ensure_proper_usage(self, partial_resolve_formatters: bool = False) -> Dict:
|
||||||
"""
|
"""
|
||||||
Validate variables resolve as plugins are executed, and return
|
Validate variables resolve as plugins are executed, and return
|
||||||
|
|
@ -174,19 +213,6 @@ class VariableValidation:
|
||||||
if url_output["url"]:
|
if url_output["url"]:
|
||||||
resolved_subscription["download"].append(url_output)
|
resolved_subscription["download"].append(url_output)
|
||||||
|
|
||||||
# TODO: make function
|
resolved_subscription["overrides"] = self._output_override_variables()
|
||||||
resolved_subscription["overrides"] = {}
|
|
||||||
for name in self.overrides.keys:
|
|
||||||
value = self.script.definition_of(name)
|
|
||||||
if name in self.script.function_names:
|
|
||||||
# Keep custom functions as-is
|
|
||||||
resolved_subscription["overrides"][name] = self.overrides.dict_with_format_strings[
|
|
||||||
name
|
|
||||||
]
|
|
||||||
elif resolved := value.maybe_resolvable:
|
|
||||||
resolved_subscription["overrides"][name] = resolved.native
|
|
||||||
else:
|
|
||||||
resolved_subscription["overrides"][name] = ScriptUtils.to_native_script(value)
|
|
||||||
|
|
||||||
assert not self.unresolved_runtime_variables
|
|
||||||
return resolved_subscription
|
return resolved_subscription
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,7 @@
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
from ytdl_sub.cli.parsers.main import parser
|
from ytdl_sub.cli.parsers.main import parser
|
||||||
from ytdl_sub.utils.logger import Logger
|
from ytdl_sub.utils.logger import Logger, LoggerLevels
|
||||||
|
|
||||||
|
|
||||||
def _main() -> int:
|
def _main() -> int:
|
||||||
|
|
@ -10,6 +10,11 @@ def _main() -> int:
|
||||||
args, _ = parser.parse_known_args()
|
args, _ = parser.parse_known_args()
|
||||||
Logger.set_log_level(log_level_name=args.ytdl_sub_log_level)
|
Logger.set_log_level(log_level_name=args.ytdl_sub_log_level)
|
||||||
|
|
||||||
|
# Suppress all logs during inspection since the output of the subcommand itself
|
||||||
|
# is all that is necessary
|
||||||
|
if args.subparser == "inspect":
|
||||||
|
Logger.set_log_level(log_level_name=LoggerLevels.QUIET.name)
|
||||||
|
|
||||||
# pylint: disable=import-outside-toplevel
|
# pylint: disable=import-outside-toplevel
|
||||||
import ytdl_sub.cli.entrypoint
|
import ytdl_sub.cli.entrypoint
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -7,8 +7,11 @@ from typing import Callable, List, Optional, Type, TypeVar, Union, get_origin
|
||||||
|
|
||||||
from ytdl_sub.script.types.resolvable import (
|
from ytdl_sub.script.types.resolvable import (
|
||||||
Argument,
|
Argument,
|
||||||
|
Boolean,
|
||||||
BuiltInFunctionType,
|
BuiltInFunctionType,
|
||||||
|
Float,
|
||||||
FutureResolvable,
|
FutureResolvable,
|
||||||
|
Integer,
|
||||||
Lambda,
|
Lambda,
|
||||||
LambdaReduce,
|
LambdaReduce,
|
||||||
LambdaThree,
|
LambdaThree,
|
||||||
|
|
@ -248,6 +251,17 @@ class FunctionSpec:
|
||||||
return l_type
|
return l_type
|
||||||
return None
|
return None
|
||||||
|
|
||||||
|
def has_sanitized_output(self) -> bool:
|
||||||
|
"""
|
||||||
|
Returns
|
||||||
|
-------
|
||||||
|
If this function were to be sanitized, whether it's redundant or not based on its
|
||||||
|
output
|
||||||
|
"""
|
||||||
|
return inspect.isclass(self.return_type) and issubclass(
|
||||||
|
self.return_type, (Integer, Float, Boolean)
|
||||||
|
)
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def _to_human_readable_name(cls, python_type: Type[NamedType] | Type[Union[NamedType]]) -> str:
|
def _to_human_readable_name(cls, python_type: Type[NamedType] | Type[Union[NamedType]]) -> str:
|
||||||
if is_optional(python_type):
|
if is_optional(python_type):
|
||||||
|
|
|
||||||
|
|
@ -1,6 +1,6 @@
|
||||||
from abc import ABC
|
from abc import ABC
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Optional
|
from typing import Dict, Optional
|
||||||
|
|
||||||
from ytdl_sub.config.config_validator import ConfigOptions
|
from ytdl_sub.config.config_validator import ConfigOptions
|
||||||
from ytdl_sub.config.overrides import Overrides
|
from ytdl_sub.config.overrides import Overrides
|
||||||
|
|
@ -254,7 +254,11 @@ class BaseSubscription(ABC):
|
||||||
"""
|
"""
|
||||||
return self._preset_options.yaml(subscription_only=False)
|
return self._preset_options.yaml(subscription_only=False)
|
||||||
|
|
||||||
def resolved_yaml(self, resolution_level: int = ResolutionLevel.RESOLVE) -> str:
|
def resolved_yaml(
|
||||||
|
self,
|
||||||
|
resolution_level: int = ResolutionLevel.RESOLVE,
|
||||||
|
mocks: Optional[Dict[str, str]] = None,
|
||||||
|
) -> str:
|
||||||
"""
|
"""
|
||||||
Returns
|
Returns
|
||||||
-------
|
-------
|
||||||
|
|
@ -269,5 +273,6 @@ class BaseSubscription(ABC):
|
||||||
output_options=self.output_options,
|
output_options=self.output_options,
|
||||||
plugins=self.plugins,
|
plugins=self.plugins,
|
||||||
resolution_level=resolution_level,
|
resolution_level=resolution_level,
|
||||||
|
mocks=mocks,
|
||||||
).ensure_proper_usage(partial_resolve_formatters=True)
|
).ensure_proper_usage(partial_resolve_formatters=True)
|
||||||
return dump_yaml(out)
|
return dump_yaml(out)
|
||||||
|
|
|
||||||
|
|
@ -51,6 +51,13 @@ class FilePathTruncater:
|
||||||
|
|
||||||
return f"{file_sub_name}{delimiter}{file_ext}"
|
return f"{file_sub_name}{delimiter}{file_ext}"
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def _truncate_directory_name(cls, directory_name: str) -> str:
|
||||||
|
while len(directory_name.encode("utf-8")) > cls._MAX_BASE_FILE_NAME_BYTES:
|
||||||
|
directory_name = directory_name[:-1]
|
||||||
|
|
||||||
|
return directory_name
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def maybe_truncate_file_path(cls, file_path: str) -> str:
|
def maybe_truncate_file_path(cls, file_path: str) -> str:
|
||||||
"""Turn into a Path, then a string, to get correct directory separators"""
|
"""Turn into a Path, then a string, to get correct directory separators"""
|
||||||
|
|
@ -61,6 +68,30 @@ class FilePathTruncater:
|
||||||
|
|
||||||
return str(file_path)
|
return str(file_path)
|
||||||
|
|
||||||
|
@classmethod
|
||||||
|
def maybe_truncate_file_name_path(cls, file_name_path: str) -> str:
|
||||||
|
"""
|
||||||
|
Truncates each component of a relative output file name path. Both the
|
||||||
|
subdirectories (which can be derived from metadata such as the title) and the
|
||||||
|
final file name are truncated if they exceed the OS limit. The file name's
|
||||||
|
extension is always preserved.
|
||||||
|
"""
|
||||||
|
parts = Path(file_name_path).parts
|
||||||
|
if not parts:
|
||||||
|
return file_name_path
|
||||||
|
|
||||||
|
*directory_parts, file_name = parts
|
||||||
|
|
||||||
|
truncated_parts = [
|
||||||
|
cls._truncate_directory_name(part) if cls._is_file_name_too_long(part) else part
|
||||||
|
for part in directory_parts
|
||||||
|
]
|
||||||
|
|
||||||
|
if cls._is_file_name_too_long(file_name):
|
||||||
|
file_name = cls._truncate_file_name(file_name)
|
||||||
|
|
||||||
|
return str(Path(*truncated_parts, file_name))
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def to_native_filepath(cls, file_path: str) -> str:
|
def to_native_filepath(cls, file_path: str) -> str:
|
||||||
"""Ensures file paths use the correct separator"""
|
"""Ensures file paths use the correct separator"""
|
||||||
|
|
|
||||||
|
|
@ -146,8 +146,7 @@ class ScriptUtils:
|
||||||
elif isinstance(sub_arg, String):
|
elif isinstance(sub_arg, String):
|
||||||
output += CustomFunctions.sanitize(sub_arg).native
|
output += CustomFunctions.sanitize(sub_arg).native
|
||||||
elif isinstance(sub_arg, BuiltInFunction) and (
|
elif isinstance(sub_arg, BuiltInFunction) and (
|
||||||
issubclass(sub_arg.function_spec.return_type, (Integer, Float, Boolean))
|
sub_arg.function_spec.has_sanitized_output() or sub_arg.name == "pad_zero"
|
||||||
or sub_arg.name == "pad_zero"
|
|
||||||
):
|
):
|
||||||
# If we know the function's output is sanitized, let's not wrap it
|
# If we know the function's output is sanitized, let's not wrap it
|
||||||
output += cls._to_script_code(sub_arg, top_level=True)
|
output += cls._to_script_code(sub_arg, top_level=True)
|
||||||
|
|
@ -180,7 +179,6 @@ class ScriptUtils:
|
||||||
return "" if top_level else "''"
|
return "" if top_level else "''"
|
||||||
|
|
||||||
quote = cls._get_quote_char(arg.native)
|
quote = cls._get_quote_char(arg.native)
|
||||||
|
|
||||||
return arg.native if top_level else f"{quote}{arg.native}{quote}"
|
return arg.native if top_level else f"{quote}{arg.native}{quote}"
|
||||||
|
|
||||||
arg = cls._maybe_to_optimized_sanitize(arg)
|
arg = cls._maybe_to_optimized_sanitize(arg)
|
||||||
|
|
|
||||||
|
|
@ -48,7 +48,7 @@ class StringFormatterFileNameValidator(StringFormatterValidator):
|
||||||
|
|
||||||
def post_process(self, resolved: str) -> str:
|
def post_process(self, resolved: str) -> str:
|
||||||
return FilePathTruncater.to_native_filepath(
|
return FilePathTruncater.to_native_filepath(
|
||||||
FilePathTruncater.maybe_truncate_file_path(resolved)
|
FilePathTruncater.maybe_truncate_file_name_path(resolved)
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -293,7 +293,7 @@ def _validate_formatter(
|
||||||
)
|
)
|
||||||
|
|
||||||
if maybe_resolved := parsed.maybe_resolvable:
|
if maybe_resolved := parsed.maybe_resolvable:
|
||||||
return formatter_validator.post_process(maybe_resolved)
|
return formatter_validator.post_process(maybe_resolved.native)
|
||||||
|
|
||||||
return ScriptUtils.to_native_script(parsed)
|
return ScriptUtils.to_native_script(parsed)
|
||||||
except RuntimeException as exc:
|
except RuntimeException as exc:
|
||||||
|
|
|
||||||
|
|
@ -79,7 +79,7 @@ def working_directory() -> str:
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
def output_directory() -> str:
|
def output_directory() -> str:
|
||||||
with tempfile.TemporaryDirectory() as temp_dir:
|
with tempfile.TemporaryDirectory() as temp_dir:
|
||||||
yield temp_dir
|
yield os.path.normpath(temp_dir)
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
|
|
@ -165,10 +165,11 @@ def subscription_yaml_file_generator() -> Callable:
|
||||||
with tempfile.NamedTemporaryFile(suffix=".yaml", delete=False) as tmp_file:
|
with tempfile.NamedTemporaryFile(suffix=".yaml", delete=False) as tmp_file:
|
||||||
tmp_file.write(json.dumps(yaml_dict).encode("utf-8"))
|
tmp_file.write(json.dumps(yaml_dict).encode("utf-8"))
|
||||||
|
|
||||||
|
file_path = os.path.normpath(tmp_file.name)
|
||||||
try:
|
try:
|
||||||
yield tmp_file.name
|
yield file_path
|
||||||
finally:
|
finally:
|
||||||
FileHandler.delete(tmp_file.name)
|
FileHandler.delete(file_path)
|
||||||
|
|
||||||
return _subscription_yaml_file_generator
|
return _subscription_yaml_file_generator
|
||||||
|
|
||||||
|
|
@ -269,10 +270,11 @@ def default_config_path(default_config) -> str:
|
||||||
with tempfile.NamedTemporaryFile(suffix=".yaml", delete=False) as tmp_file:
|
with tempfile.NamedTemporaryFile(suffix=".yaml", delete=False) as tmp_file:
|
||||||
tmp_file.write(json.dumps(default_config._value).encode("utf-8"))
|
tmp_file.write(json.dumps(default_config._value).encode("utf-8"))
|
||||||
|
|
||||||
|
file_path = os.path.normpath(tmp_file.name)
|
||||||
try:
|
try:
|
||||||
yield tmp_file.name
|
yield file_path
|
||||||
finally:
|
finally:
|
||||||
FileHandler.delete(tmp_file.name)
|
FileHandler.delete(file_path)
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture()
|
@pytest.fixture()
|
||||||
|
|
|
||||||
|
|
@ -2,12 +2,12 @@ from typing import Optional
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from conftest import mock_run_from_cli
|
from conftest import mock_run_from_cli
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.utils.file_handler import FileMetadata
|
from ytdl_sub.utils.file_handler import FileMetadata
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestView:
|
class TestView:
|
||||||
@pytest.mark.parametrize("split_chapters", [True, False])
|
@pytest.mark.parametrize("split_chapters", [True, False])
|
||||||
def test_view_from_cli(
|
def test_view_from_cli(
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,7 @@
|
||||||
import pytest
|
import pytest
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
||||||
|
|
@ -47,7 +47,7 @@ def youtube_release_preset_dict(output_directory):
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestAudioExtract:
|
class TestAudioExtract:
|
||||||
@pytest.mark.parametrize("dry_run", [False])
|
@pytest.mark.parametrize("dry_run", [False])
|
||||||
def test_audio_extract_single_song(
|
def test_audio_extract_single_song(
|
||||||
|
|
|
||||||
|
|
@ -3,7 +3,7 @@ from typing import Dict
|
||||||
import pytest
|
import pytest
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
||||||
|
|
@ -54,7 +54,7 @@ def chapters_from_comments_preset_dict(sponsorblock_and_subs_preset_dict: Dict)
|
||||||
return sponsorblock_and_subs_preset_dict
|
return sponsorblock_and_subs_preset_dict
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestChapters:
|
class TestChapters:
|
||||||
@pytest.mark.parametrize("dry_run", [True, False])
|
@pytest.mark.parametrize("dry_run", [True, False])
|
||||||
def test_chapters_sponsorblock_and_removal_with_subs(
|
def test_chapters_sponsorblock_and_removal_with_subs(
|
||||||
|
|
|
||||||
|
|
@ -5,7 +5,7 @@ import pytest
|
||||||
from conftest import assert_logs
|
from conftest import assert_logs
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.downloaders.ytdlp import YTDLP
|
from ytdl_sub.downloaders.ytdlp import YTDLP
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
@ -39,7 +39,7 @@ def rolling_recent_channel_preset_dict(recent_preset_dict):
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestDateRange:
|
class TestDateRange:
|
||||||
@pytest.mark.parametrize("dry_run", [True, False])
|
@pytest.mark.parametrize("dry_run", [True, False])
|
||||||
@pytest.mark.parametrize("date_range_breaks", [True, False])
|
@pytest.mark.parametrize("date_range_breaks", [True, False])
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,7 @@
|
||||||
import pytest
|
import pytest
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
||||||
|
|
@ -20,7 +20,7 @@ def preset_dict(output_directory):
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestFileConvert:
|
class TestFileConvert:
|
||||||
@pytest.mark.parametrize("dry_run", [True, False])
|
@pytest.mark.parametrize("dry_run", [True, False])
|
||||||
def test_file_convert(
|
def test_file_convert(
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,7 @@
|
||||||
import pytest
|
import pytest
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
||||||
|
|
@ -46,7 +46,7 @@ def livestream_preset_dict(output_directory):
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestFileConvert:
|
class TestFileConvert:
|
||||||
def test_livestreams_download_filtered(
|
def test_livestreams_download_filtered(
|
||||||
self,
|
self,
|
||||||
|
|
|
||||||
|
|
@ -4,7 +4,7 @@ import mergedeep
|
||||||
import pytest
|
import pytest
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
from ytdl_sub.utils.exceptions import ValidationException
|
from ytdl_sub.utils.exceptions import ValidationException
|
||||||
|
|
@ -58,7 +58,7 @@ def yt_album_as_chapters_with_regex_preset_dict(yt_album_as_chapters_preset_dict
|
||||||
return yt_album_as_chapters_preset_dict
|
return yt_album_as_chapters_preset_dict
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestSplitByChapters:
|
class TestSplitByChapters:
|
||||||
@pytest.mark.parametrize("dry_run", [True, False])
|
@pytest.mark.parametrize("dry_run", [True, False])
|
||||||
def test_video_with_chapters(
|
def test_video_with_chapters(
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,7 @@
|
||||||
import pytest
|
import pytest
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.config.config_file import ConfigFile
|
from ytdl_sub.config.config_file import ConfigFile
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
@ -32,7 +32,7 @@ def test_single_video_subs_embed_and_file_preset_dict(single_video_subs_embed_pr
|
||||||
return single_video_subs_embed_preset_dict
|
return single_video_subs_embed_preset_dict
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestSubtitles:
|
class TestSubtitles:
|
||||||
def test_subtitle_lang_variable_partial_validates(self, default_config):
|
def test_subtitle_lang_variable_partial_validates(self, default_config):
|
||||||
default_config_dict = default_config.as_dict()
|
default_config_dict = default_config.as_dict()
|
||||||
|
|
|
||||||
|
|
@ -2,7 +2,7 @@ from typing import Dict
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.config.config_file import ConfigFile
|
from ytdl_sub.config.config_file import ConfigFile
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
@ -51,7 +51,7 @@ def tv_show_collection_bilateral_dict(output_directory):
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestBilateral:
|
class TestBilateral:
|
||||||
def test_tv_show_by_date_downloads_bilateral(
|
def test_tv_show_by_date_downloads_bilateral(
|
||||||
self,
|
self,
|
||||||
|
|
|
||||||
|
|
@ -4,6 +4,7 @@ import pytest
|
||||||
from conftest import assert_logs
|
from conftest import assert_logs
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.downloaders.ytdlp import YTDLP
|
from ytdl_sub.downloaders.ytdlp import YTDLP
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
@ -23,6 +24,7 @@ def subscription_dict(output_directory):
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="Soundcloud tests cannot run in GH")
|
||||||
class TestSoundcloudDiscography:
|
class TestSoundcloudDiscography:
|
||||||
"""
|
"""
|
||||||
Downloads my (bad) SC recordings I made. Ensure the above files exist and have the
|
Downloads my (bad) SC recordings I made. Ensure the above files exist and have the
|
||||||
|
|
|
||||||
|
|
@ -3,7 +3,7 @@ from typing import Callable, Dict
|
||||||
import pytest
|
import pytest
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.config.config_file import ConfigFile
|
from ytdl_sub.config.config_file import ConfigFile
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
@ -42,7 +42,7 @@ def channel_preset_dict(output_directory):
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestChannel:
|
class TestChannel:
|
||||||
"""
|
"""
|
||||||
Downloads my old minecraft youtube channel. Ensure the above files exist and have the
|
Downloads my old minecraft youtube channel. Ensure the above files exist and have the
|
||||||
|
|
|
||||||
|
|
@ -1,7 +1,7 @@
|
||||||
import pytest
|
import pytest
|
||||||
from expected_download import assert_expected_downloads
|
from expected_download import assert_expected_downloads
|
||||||
from expected_transaction_log import assert_transaction_log_matches
|
from expected_transaction_log import assert_transaction_log_matches
|
||||||
from resources import DISABLE_YOUTUBE_TESTS
|
from resources import DISABLE_E2E_TESTS
|
||||||
|
|
||||||
from ytdl_sub.subscriptions.subscription import Subscription
|
from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
|
|
||||||
|
|
@ -30,7 +30,7 @@ def single_video_preset_dict(output_directory):
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(DISABLE_YOUTUBE_TESTS, reason="YouTube tests cannot run in GH")
|
@pytest.mark.skipif(DISABLE_E2E_TESTS, reason="YouTube tests cannot run in GH")
|
||||||
class TestYoutubeVideo:
|
class TestYoutubeVideo:
|
||||||
@pytest.mark.parametrize("dry_run", [True, False])
|
@pytest.mark.parametrize("dry_run", [True, False])
|
||||||
def test_single_video_download(
|
def test_single_video_download(
|
||||||
|
|
|
||||||
45
tests/integration/cli/test_inspect.py
Normal file
45
tests/integration/cli/test_inspect.py
Normal file
|
|
@ -0,0 +1,45 @@
|
||||||
|
from pathlib import Path
|
||||||
|
|
||||||
|
import pytest
|
||||||
|
import yaml
|
||||||
|
from conftest import mock_run_from_cli
|
||||||
|
from unit.config.test_subscription_resolution import compare_resolved_yaml
|
||||||
|
|
||||||
|
from ytdl_sub.config.validators.variable_validation import ResolutionLevel
|
||||||
|
from ytdl_sub.utils.system import IS_WINDOWS
|
||||||
|
|
||||||
|
|
||||||
|
class TestInspect:
|
||||||
|
@pytest.mark.parametrize("config_provided", [True, False])
|
||||||
|
@pytest.mark.parametrize(
|
||||||
|
"inspect_level", ["0", "original", "1", "fill", "2", "resolve", "3", "internal"]
|
||||||
|
)
|
||||||
|
def test_inspect_command(
|
||||||
|
self,
|
||||||
|
capsys,
|
||||||
|
default_config_path: str,
|
||||||
|
tv_show_subscriptions_path: Path,
|
||||||
|
output_directory: str,
|
||||||
|
config_provided: bool,
|
||||||
|
inspect_level: str,
|
||||||
|
):
|
||||||
|
# TODO: fix mock_run_from_cli in windows to handle file paths correctly
|
||||||
|
if IS_WINDOWS:
|
||||||
|
return
|
||||||
|
|
||||||
|
# Shares same test fixture as `test_subscription_resolution.py`
|
||||||
|
args = f"--config {default_config_path} " if config_provided else ""
|
||||||
|
args += f"inspect {tv_show_subscriptions_path} --match 'NOVA PBS' "
|
||||||
|
args += f"--level {inspect_level} "
|
||||||
|
|
||||||
|
subscriptions = mock_run_from_cli(args=args)
|
||||||
|
assert len(subscriptions) == 0
|
||||||
|
|
||||||
|
out = yaml.safe_load(capsys.readouterr().out)
|
||||||
|
compare_resolved_yaml(
|
||||||
|
out=out,
|
||||||
|
output_directory=output_directory,
|
||||||
|
subscription_name="NOVA PBS",
|
||||||
|
preset_type="tv_show",
|
||||||
|
resolution_level=ResolutionLevel.level_number(resolution_arg=inspect_level),
|
||||||
|
)
|
||||||
|
|
@ -4,7 +4,7 @@ import shutil
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Dict
|
from typing import Dict
|
||||||
|
|
||||||
DISABLE_YOUTUBE_TESTS: bool = True
|
DISABLE_E2E_TESTS: bool = True
|
||||||
REGENERATE_FIXTURES: bool = False
|
REGENERATE_FIXTURES: bool = False
|
||||||
|
|
||||||
RESOURCE_PATH: Path = Path("tests") / "resources"
|
RESOURCE_PATH: Path = Path("tests") / "resources"
|
||||||
|
|
|
||||||
|
|
@ -1,4 +1,5 @@
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
from typing import Dict
|
||||||
|
|
||||||
import pytest
|
import pytest
|
||||||
import yaml
|
import yaml
|
||||||
|
|
@ -10,12 +11,13 @@ from ytdl_sub.subscriptions.subscription import Subscription
|
||||||
from ytdl_sub.utils.file_path import FilePathTruncater
|
from ytdl_sub.utils.file_path import FilePathTruncater
|
||||||
|
|
||||||
|
|
||||||
def _ensure_resolved_yaml(
|
def compare_resolved_yaml(
|
||||||
sub: Subscription, output_directory: str, preset_type: str, resolution_level: int
|
out: Dict,
|
||||||
|
output_directory: str,
|
||||||
|
subscription_name: str,
|
||||||
|
preset_type: str,
|
||||||
|
resolution_level: int,
|
||||||
) -> None:
|
) -> None:
|
||||||
output_yaml = sub.resolved_yaml(resolution_level=resolution_level)
|
|
||||||
out = yaml.safe_load(output_yaml)
|
|
||||||
|
|
||||||
expected_out_filename = (
|
expected_out_filename = (
|
||||||
f"{preset_type}/inspect_sub_{ResolutionLevel.name_of(resolution_level)}.json"
|
f"{preset_type}/inspect_sub_{ResolutionLevel.name_of(resolution_level)}.json"
|
||||||
)
|
)
|
||||||
|
|
@ -24,7 +26,7 @@ def _ensure_resolved_yaml(
|
||||||
if resolution_level > ResolutionLevel.ORIGINAL:
|
if resolution_level > ResolutionLevel.ORIGINAL:
|
||||||
output_path = Path(output_directory)
|
output_path = Path(output_directory)
|
||||||
if "tv_show_directory" in expected_out["overrides"]:
|
if "tv_show_directory" in expected_out["overrides"]:
|
||||||
output_path = output_path / sub.name
|
output_path = output_path / subscription_name
|
||||||
|
|
||||||
expected_out["output_options"]["output_directory"] = FilePathTruncater.to_native_filepath(
|
expected_out["output_options"]["output_directory"] = FilePathTruncater.to_native_filepath(
|
||||||
str(output_path)
|
str(output_path)
|
||||||
|
|
@ -40,6 +42,21 @@ def _ensure_resolved_yaml(
|
||||||
assert out == expected_out
|
assert out == expected_out
|
||||||
|
|
||||||
|
|
||||||
|
def _ensure_resolved_yaml(
|
||||||
|
sub: Subscription, output_directory: str, preset_type: str, resolution_level: int
|
||||||
|
) -> None:
|
||||||
|
output_yaml = sub.resolved_yaml(resolution_level=resolution_level)
|
||||||
|
out = yaml.safe_load(output_yaml)
|
||||||
|
|
||||||
|
compare_resolved_yaml(
|
||||||
|
out=out,
|
||||||
|
output_directory=output_directory,
|
||||||
|
subscription_name=sub.name,
|
||||||
|
preset_type=preset_type,
|
||||||
|
resolution_level=resolution_level,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.parametrize("resolution_level", ResolutionLevel.all())
|
@pytest.mark.parametrize("resolution_level", ResolutionLevel.all())
|
||||||
class TestResolution:
|
class TestResolution:
|
||||||
def test_resolution_tv_show(
|
def test_resolution_tv_show(
|
||||||
|
|
|
||||||
|
|
@ -77,6 +77,46 @@ class TestStringFormatterFilePathValidator:
|
||||||
assert len(dir_paths) == 1
|
assert len(dir_paths) == 1
|
||||||
assert Path(truncated_file_path) == dir_paths[0]
|
assert Path(truncated_file_path) == dir_paths[0]
|
||||||
|
|
||||||
|
@pytest.mark.parametrize("ext", ["mp4", "-thumb.jpg"])
|
||||||
|
def test_truncates_subdirectory_in_file_name_path(self, ext: str):
|
||||||
|
if "thumb" not in ext: # do not put . in front of -thumb
|
||||||
|
ext = f".{ext}"
|
||||||
|
|
||||||
|
# Mathematical bold unicode characters are 4 bytes each in UTF-8, so a title used
|
||||||
|
# as a subdirectory can easily exceed the OS file name limit (typically 255 bytes).
|
||||||
|
long_directory = (
|
||||||
|
"[2026] 𝗿𝘁𝗶𝘀𝘁 𝗢𝗻𝗲 - 𝗔 𝗩𝗲𝗿𝘆 𝗟𝗼𝗻𝗴 𝗦𝗼𝗻𝗴 𝗧𝗶𝘁𝗹𝗲 "
|
||||||
|
"𝗧𝗵𝗮𝘁 𝗘𝘅𝗰𝗲𝗲𝗱𝘀 𝗧𝗵𝗲 𝗟𝗶𝗺𝗶𝘁 (𝗘𝘅𝘁𝗲𝗻𝗱𝗲𝗱 𝗠𝗶𝘅 𝘅 "
|
||||||
|
"𝗦𝗲𝗰𝗼𝗻𝗱 𝗔𝗿𝘁𝗶𝘀𝘁 & 𝗧𝗵𝗶𝗿𝗱 𝗔𝗿𝘁𝗶𝘀𝘁)"
|
||||||
|
)
|
||||||
|
# Sanity check that the directory really does exceed the limit
|
||||||
|
assert len(long_directory.encode("utf-8")) > FilePathTruncater._MAX_BASE_FILE_NAME_BYTES
|
||||||
|
|
||||||
|
with tempfile.TemporaryDirectory() as temp_dir:
|
||||||
|
file_path = str(Path(temp_dir) / long_directory / f"video{ext}")
|
||||||
|
|
||||||
|
formatter = StringFormatterFileNameValidator(name="test", value="")
|
||||||
|
truncated_file_path = formatter.post_process(file_path)
|
||||||
|
|
||||||
|
truncated_directory, truncated_file_name = os.path.split(truncated_file_path)
|
||||||
|
_, truncated_directory_name = os.path.split(truncated_directory)
|
||||||
|
|
||||||
|
# Both the subdirectory and the file name must fit within the OS limit
|
||||||
|
assert (
|
||||||
|
len(truncated_directory_name.encode("utf-8"))
|
||||||
|
<= FilePathTruncater._MAX_BASE_FILE_NAME_BYTES
|
||||||
|
)
|
||||||
|
assert (
|
||||||
|
len(truncated_file_name.encode("utf-8"))
|
||||||
|
<= FilePathTruncater._MAX_BASE_FILE_NAME_BYTES
|
||||||
|
)
|
||||||
|
assert truncated_file_name.endswith(ext)
|
||||||
|
|
||||||
|
# The original bug raised "OSError: [Errno 36] Filename too long" here because
|
||||||
|
# the subdirectory component was never truncated.
|
||||||
|
os.makedirs(truncated_directory, exist_ok=True)
|
||||||
|
assert os.path.isdir(truncated_directory)
|
||||||
|
|
||||||
@pytest.mark.parametrize(
|
@pytest.mark.parametrize(
|
||||||
"file_name_max_bytes, expected_max",
|
"file_name_max_bytes, expected_max",
|
||||||
[
|
[
|
||||||
|
|
|
||||||
Loading…
Reference in a new issue