e92b1cd12a
A complete gutting of the internals of ytdl-sub to support functions in our variable syntax, in addition to being able to access a yt-dlp entry's .info.json fields using functions. Functionally, ytdl-sub should still look and behave the same from a user-perspective. With so many lines of code changed (+8927, -2708), no doubt there will be new issues. Please make a GH issue or reach out on Discord if your config/subscriptions break in any way/shape/form. Details on how to use function support will come soon in the form of proper documentation in our readthedocs.
102 lines
2.7 KiB
Python
102 lines
2.7 KiB
Python
import os
|
|
from typing import Any
|
|
from typing import Dict
|
|
|
|
from ytdl_sub.config.config_validator import ConfigValidator
|
|
from ytdl_sub.config.preset import Preset
|
|
from ytdl_sub.utils.exceptions import FileNotFoundException
|
|
from ytdl_sub.utils.ffmpeg import FFMPEG
|
|
from ytdl_sub.utils.file_path import FilePathTruncater
|
|
from ytdl_sub.utils.yaml import load_yaml
|
|
|
|
|
|
class ConfigFile(ConfigValidator):
|
|
def __init__(self, name: str, value: Any):
|
|
super().__init__(name, value)
|
|
|
|
for preset_name, preset_dict in self.presets.dict.items():
|
|
Preset.preset_partial_validate(config=self, name=preset_name, value=preset_dict)
|
|
|
|
# After validation, perform initialization
|
|
self._initialize()
|
|
|
|
def _initialize(self):
|
|
"""
|
|
Configures things (umask, pgid, etc) prior to any downloading
|
|
|
|
Returns
|
|
-------
|
|
self
|
|
"""
|
|
if self.config_options.umask:
|
|
os.umask(int(self.config_options.umask, 8))
|
|
|
|
FFMPEG.set_paths(
|
|
ffmpeg_path=self.config_options.ffmpeg_path,
|
|
ffprobe_path=self.config_options.ffprobe_path,
|
|
)
|
|
|
|
FilePathTruncater.set_max_file_name_bytes(
|
|
max_file_name_bytes=self.config_options.file_name_max_bytes
|
|
)
|
|
|
|
return self
|
|
|
|
@classmethod
|
|
def from_dict(cls, config_dict: dict, name: str = "") -> "ConfigFile":
|
|
"""
|
|
Parameters
|
|
----------
|
|
config_dict:
|
|
The config in dictionary format
|
|
name:
|
|
Name of the config
|
|
Returns
|
|
-------
|
|
Config file validator
|
|
"""
|
|
return ConfigFile(name=name, value=config_dict)
|
|
|
|
@classmethod
|
|
def from_file_path(cls, config_path: str) -> "ConfigFile":
|
|
"""
|
|
Parameters
|
|
----------
|
|
config_path:
|
|
Path to the config yaml
|
|
|
|
Returns
|
|
-------
|
|
Config file validator
|
|
|
|
Raises
|
|
------
|
|
FileNotFoundException
|
|
Not found
|
|
"""
|
|
try:
|
|
config_dict = load_yaml(file_path=config_path)
|
|
except FileNotFoundException as exc:
|
|
raise FileNotFoundException(
|
|
f"The config file '{config_path}' could not be found. "
|
|
f"Did you set --config correctly?"
|
|
) from exc
|
|
|
|
return ConfigFile.from_dict(name=config_path, config_dict=config_dict)
|
|
|
|
@classmethod
|
|
def default(cls) -> "ConfigFile":
|
|
"""
|
|
Returns
|
|
-------
|
|
Config initialized with all defaults
|
|
"""
|
|
return ConfigFile(name="default_config", value={})
|
|
|
|
def as_dict(self) -> Dict[str, Any]:
|
|
"""
|
|
Returns
|
|
-------
|
|
The config in its dict form
|
|
"""
|
|
return self._value
|