From e1f2e37858b58b8ad8446753ad454c89a726646d Mon Sep 17 00:00:00 2001 From: Benjamin Hsieh Date: Fri, 31 Mar 2023 20:47:44 -0700 Subject: [PATCH] [FEATURE] `creator_sanitized` source variable (#574) --- src/ytdl_sub/entries/variables/entry_variables.py | 10 ++++++++++ tests/unit/entries/conftest.py | 1 + 2 files changed, 11 insertions(+) diff --git a/src/ytdl_sub/entries/variables/entry_variables.py b/src/ytdl_sub/entries/variables/entry_variables.py index 5e915477..cbbb39de 100644 --- a/src/ytdl_sub/entries/variables/entry_variables.py +++ b/src/ytdl_sub/entries/variables/entry_variables.py @@ -361,6 +361,16 @@ class EntryVariables(BaseEntryVariables): """ return self.kwargs_get(CREATOR, self.channel) + @property + def creator_sanitized(self: Self) -> str: + """ + Returns + ------- + str + The creator name, sanitized + """ + return sanitize_filename(self.creator) + @property def channel(self: Self) -> str: """ diff --git a/tests/unit/entries/conftest.py b/tests/unit/entries/conftest.py index 096793af..1619cf49 100644 --- a/tests/unit/entries/conftest.py +++ b/tests/unit/entries/conftest.py @@ -70,6 +70,7 @@ def mock_entry_to_dict( "ext": ext, "description": "", "creator": "abc123", + "creator_sanitized": "abc123", "channel": "abc123", "channel_sanitized": "abc123", "extractor": extractor,