73 lines
2.2 KiB
Python
73 lines
2.2 KiB
Python
from typing import List
|
|
from typing import Optional
|
|
from typing import Tuple
|
|
|
|
from yt_dlp.utils import datetime_from_str
|
|
|
|
from ytdl_sub.config.plugin import Plugin
|
|
from ytdl_sub.config.preset_options import OptionsDictValidator
|
|
from ytdl_sub.validators.string_datetime import StringDatetimeValidator
|
|
|
|
|
|
class DateRangeOptions(OptionsDictValidator):
|
|
"""
|
|
Only download files uploaded within the specified date range.
|
|
|
|
Usage:
|
|
|
|
.. code-block:: yaml
|
|
|
|
presets:
|
|
my_example_preset:
|
|
date_range:
|
|
before: "now"
|
|
after: "today-2weeks"
|
|
"""
|
|
|
|
_optional_keys = {"before", "after"}
|
|
|
|
def __init__(self, name, value):
|
|
super().__init__(name, value)
|
|
self._before = self._validate_key_if_present("before", StringDatetimeValidator)
|
|
self._after = self._validate_key_if_present("after", StringDatetimeValidator)
|
|
|
|
@property
|
|
def before(self) -> Optional[StringDatetimeValidator]:
|
|
"""
|
|
Optional. Only download videos before this datetime.
|
|
"""
|
|
return self._before
|
|
|
|
@property
|
|
def after(self) -> Optional[StringDatetimeValidator]:
|
|
"""
|
|
Optional. Only download videos after this datetime.
|
|
"""
|
|
return self._after
|
|
|
|
|
|
class DateRangePlugin(Plugin[DateRangeOptions]):
|
|
plugin_options_type = DateRangeOptions
|
|
|
|
def _to_date(self, date_validator: StringDatetimeValidator) -> str:
|
|
date_str = self.overrides.apply_formatter(formatter=date_validator)
|
|
return str(datetime_from_str(date_str).date())
|
|
|
|
def ytdl_options_match_filters(self) -> Tuple[List[str], List[str]]:
|
|
"""
|
|
Returns
|
|
-------
|
|
match filters for before (non-breaking) and after (breaking)
|
|
"""
|
|
match_filters: List[str] = []
|
|
breaking_match_filters: List[str] = []
|
|
|
|
if self.plugin_options.before:
|
|
match_filters.append(f"upload_date < {self._to_date(self.plugin_options.before)}")
|
|
|
|
if self.plugin_options.after:
|
|
breaking_match_filters.append(
|
|
f"upload_date >= {self._to_date(self.plugin_options.after)}"
|
|
)
|
|
|
|
return match_filters, breaking_match_filters
|