From c443d2e29552f23b1ec1517bd52de67c46e7c58a Mon Sep 17 00:00:00 2001 From: Jesse Bannon Date: Thu, 18 Aug 2022 17:57:27 -0700 Subject: [PATCH] [BUGFIX] Fix truncated year source variable (#185) --- src/ytdl_sub/entries/variables/entry_variables.py | 2 +- tests/unit/entries/conftest.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/ytdl_sub/entries/variables/entry_variables.py b/src/ytdl_sub/entries/variables/entry_variables.py index dfb1b1ca..2f423a86 100644 --- a/src/ytdl_sub/entries/variables/entry_variables.py +++ b/src/ytdl_sub/entries/variables/entry_variables.py @@ -140,7 +140,7 @@ class EntryVariables(SourceVariables): int The last two digits of the upload year, i.e. 22 in 2022 """ - return int(self.upload_date[:2]) + return int(str(self.upload_year)[-2:]) @property def upload_month_padded(self) -> str: diff --git a/tests/unit/entries/conftest.py b/tests/unit/entries/conftest.py index 2e625eeb..e78d17d4 100644 --- a/tests/unit/entries/conftest.py +++ b/tests/unit/entries/conftest.py @@ -86,7 +86,7 @@ def mock_entry_to_dict( "upload_date": upload_date, "upload_date_standardized": f"{upload_year}-{_pad(upload_month)}-{_pad(upload_day)}", "upload_year": upload_year, - "upload_year_truncated": int(str(upload_year)[:2]), + "upload_year_truncated": 21, "upload_month": upload_month, "upload_month_padded": _pad(upload_month), "upload_day": upload_day,