diff --git a/src/ytdl_sub/cli/parsers/dl.py b/src/ytdl_sub/cli/parsers/dl.py index 157a156f..39c28626 100644 --- a/src/ytdl_sub/cli/parsers/dl.py +++ b/src/ytdl_sub/cli/parsers/dl.py @@ -134,11 +134,14 @@ class DownloadArgsParser: next_dict[arg_name_split[-1]] = arg_value - # TODO: handle ints/floats if arg_value == "True": next_dict[arg_name_split[-1]] = True elif arg_value == "False": next_dict[arg_name_split[-1]] = False + elif arg_value.isdigit(): + next_dict[arg_name_split[-1]] = int(arg_value) + elif arg_value.replace(".", "", 1).isdigit(): + next_dict[arg_name_split[-1]] = float(arg_value) return argument_dict diff --git a/tests/e2e/conftest.py b/tests/e2e/conftest.py index 250d439b..d2957a27 100644 --- a/tests/e2e/conftest.py +++ b/tests/e2e/conftest.py @@ -9,7 +9,6 @@ import pytest from ytdl_sub.cli.entrypoint import main from ytdl_sub.subscriptions.subscription import Subscription from ytdl_sub.utils.file_handler import FileHandler -from ytdl_sub.utils.system import IS_WINDOWS @pytest.fixture