refactory

This commit is contained in:
Jesse Bannon 2023-09-20 08:19:16 -07:00
parent 526ef90298
commit 566f550a32
2 changed files with 19 additions and 9 deletions

View file

@ -2,10 +2,9 @@ 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.utils.datetime import to_date_str
from ytdl_sub.validators.string_datetime import StringDatetimeValidator
@ -49,10 +48,6 @@ class DateRangeOptions(OptionsDictValidator):
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
@ -63,11 +58,15 @@ class DateRangePlugin(Plugin[DateRangeOptions]):
breaking_match_filters: List[str] = []
if self.plugin_options.before:
match_filters.append(f"upload_date < {self._to_date(self.plugin_options.before)}")
before_str = to_date_str(
date_validator=self.plugin_options.before, overrides=self.overrides
)
match_filters.append(f"upload_date < {before_str}")
if self.plugin_options.after:
breaking_match_filters.append(
f"upload_date >= {self._to_date(self.plugin_options.after)}"
after_str = to_date_str(
date_validator=self.plugin_options.after, overrides=self.overrides
)
breaking_match_filters.append(f"upload_date >= {after_str}")
return match_filters, breaking_match_filters

View file

@ -1,6 +1,7 @@
from typing import Optional
from yt_dlp import DateRange
from yt_dlp.utils import datetime_from_str
from ytdl_sub.config.preset_options import Overrides
from ytdl_sub.validators.string_datetime import StringDatetimeValidator
@ -29,3 +30,13 @@ def to_date_range(
return DateRange(start=start, end=end)
return None
def to_date_str(date_validator: StringDatetimeValidator, overrides: Overrides) -> str:
"""
Returns
-------
Date in the form of YYYYMMDD as a string
"""
date_str = overrides.apply_formatter(formatter=date_validator)
return datetime_from_str(date_str).date().strftime("%Y%m%d")